Add PROJECT_ID (GoogleCloudPlatform#1953)

engelke · web-flow · commit 633d03c7f56d · 2019-01-24T14:32:29.000-08:00
* Update requirements.txt

* Adding some rate limiting

* Add PROJECT_ID

* Update main.py

* Update main.py

* Update main.py

* Update main.py

* Use standard replacement project ID in sample

* Remove unneeded import

* Back to using environment for PROJECT ID

* Change how PROJECT_ID is injected

* Had used wrong object

* Update main.py

* Update main.py

* Update main.py
diff --git a/trace/main.py b/trace/main.py
@@ -12,19 +12,28 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import argparse
 import random
 import time
 
 from flask import Flask, redirect, url_for
 
+
 # [START trace_setup_python_configure]
 from opencensus.trace.exporters import stackdriver_exporter
 import opencensus.trace.tracer
 
-exporter = stackdriver_exporter.StackdriverExporter()
-tracer = opencensus.trace.tracer.Tracer(exporter=exporter)
+
+def initialize_tracer(project_id):
+    exporter = stackdriver_exporter.StackdriverExporter(
+        project_id=project_id
+    )
+    tracer = opencensus.trace.tracer.Tracer(exporter=exporter)
+
+    return tracer
 # [END trace_setup_python_configure]
 
+
 app = Flask(__name__)
 
 
@@ -36,6 +45,7 @@ def root():
 # [START trace_setup_python_quickstart]
 @app.route('/index.html', methods=['GET'])
 def index():
+    tracer = app.config['TRACER']
     tracer.start_span(name='index')
 
     # Add up to 1 sec delay, weighted toward zero
@@ -48,4 +58,15 @@ def index():
 
 
 if __name__ == '__main__':
+    parser = argparse.ArgumentParser(
+        description=__doc__,
+        formatter_class=argparse.RawDescriptionHelpFormatter
+    )
+    parser.add_argument(
+        '--project_id', help='Project ID you want to access.', required=True)
+    args = parser.parse_args()
+
+    tracer = initialize_tracer(args.project_id)
+    app.config['TRACER'] = tracer
+
     app.run()
diff --git a/trace/main_test.py b/trace/main_test.py
@@ -12,11 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import os
 import main
 
 
 def test_index():
+    project_id = os.environ['GCLOUD_PROJECT']
     main.app.testing = True
+    main.app.config['TRACER'] = main.initialize_tracer(project_id)
     client = main.app.test_client()
 
     resp = client.get('/index.html')
@@ -25,7 +28,9 @@ def test_index():
 
 
 def test_redirect():
+    project_id = os.environ['GCLOUD_PROJECT']
     main.app.testing = True
+    main.app.config['TRACER'] = main.initialize_tracer(project_id)
     client = main.app.test_client()
 
     resp = client.get('/')
