Remove sub directories of metrics scripts (GoogleCloudPlatform#159)

liyanhui1228 · web-flow · commit 509cb5ce2768 · 2017-10-24T16:32:44.000-07:00
diff --git a/perf_dashboard/__init__.py b/perf_dashboard/__init__.py
diff --git a/perf_dashboard/posts_stats.py b/perf_dashboard/posts_stats.py
@@ -25,9 +25,7 @@
 
 from google.cloud import bigquery
 
-# Need this to import the local helper function
-sys.path.insert(0, os.path.abspath(__file__+"/../../.."))
-from perf_dashboard import bq_utils
+import bq_utils
 
 GCLOUD_PROJECT_ENV = 'GCLOUD_PROJECT'
 DATASET_NAME = 'stackoverflow'
diff --git a/perf_dashboard/python_clientlibs_download.py b/perf_dashboard/python_clientlibs_download.py
@@ -20,9 +20,7 @@
 
 from google.cloud import bigquery
 
-sys.path.insert(0, os.path.abspath(__file__+"/../../.."))
-from perf_dashboard import bq_utils
-
+import bq_utils
 
 GCLOUD_PROJECT_ENV = 'GCLOUD_PROJECT'
 
@@ -136,7 +134,7 @@ def main():
             clientlibs_table_name=table_name,
             date_str=datetime.datetime.now().strftime("%Y%m%d"))
         bq_utils.insert_rows(
-            project=project,
+            project=os.environ.get(GCLOUD_PROJECT_ENV),
             dataset_name=DATASET_NAME,
             table_name=table_name,
             rows=rows)
