From f8c6a24f0601315a50f58e958aba44c7745b6046 Mon Sep 17 00:00:00 2001 From: Piotr Chromiec Date: Mon, 10 Jul 2017 11:09:09 +0000 Subject: [PATCH] ENH: BQ Job Id in verbose output --- pandas_gbq/gbq.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/pandas_gbq/gbq.py b/pandas_gbq/gbq.py index b9bb9498..5fa78153 100644 --- 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)