ENH: BQ Job Id in verbose output

googleapis · Jul 10, 2017 · f8c6a24 · f8c6a24
1 parent dd53b5c
commit f8c6a24
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pandas_gbq/gbq.py b/pandas_gbq/gbq.py
@@ -514,7 +514,7 @@ def run_query(self, query, **kwargs):
             self._print('Requesting query... ', end="")
             query_reply = job_collection.insert(
                 projectId=self.project_id, body=job_data).execute()
-            self._print('ok.\nQuery running...')
+            self._print('ok.')
         except (RefreshError, ValueError):
             if self.private_key:
                 raise AccessDenied(
@@ -527,13 +527,15 @@ def run_query(self, query, **kwargs):
             self.process_http_error(ex)
 
         job_reference = query_reply['jobReference']
+        job_id = job_reference['jobId']
+        self._print('Job ID: %s\nQuery running...' % job_id)
 
         while not query_reply.get('jobComplete', False):
             self.print_elapsed_seconds('  Elapsed', 's. Waiting...')
             try:
                 query_reply = job_collection.getQueryResults(
                     projectId=job_reference['projectId'],
-                    jobId=job_reference['jobId']).execute()
+                    jobId=job_id).execute()
             except HttpError as ex:
                 self.process_http_error(ex)
 
@@ -584,7 +586,7 @@ def run_query(self, query, **kwargs):
             try:
                 query_reply = job_collection.getQueryResults(
                     projectId=job_reference['projectId'],
-                    jobId=job_reference['jobId'],
+                    jobId=job_id,
                     pageToken=page_token).execute()
             except HttpError as ex:
                 self.process_http_error(ex)