feat: send BigQuery cancel request when canceling bigframes process (#103)

Co-authored-by: Henry J Solberg <[email protected]>

Author: milkshakeiii

bigframes/formatting_helpers.py

@@ -153,6 +153,11 @@ def wait_for_query_job(
     except api_core_exceptions.GoogleAPICallError as exc:
         add_feedback_link(exc)
         raise
+    except KeyboardInterrupt:
+        query_job.cancel()
+        print(f"Requested cancelation for {query_job.job_type} {query_job.job_id}...")
+        # begin the cancel request before immediately rethrowing
+        raise
 
 
 def wait_for_job(job: GenericJob, progress_bar: Optional[str] = None):
@@ -190,6 +195,11 @@ def wait_for_job(job: GenericJob, progress_bar: Optional[str] = None):
     except api_core_exceptions.GoogleAPICallError as exc:
         add_feedback_link(exc)
         raise
+    except KeyboardInterrupt:
+        job.cancel()
+        print(f"Requested cancelation for {job.job_type} {job.job_id}...")
+        # begin the cancel request before immediately rethrowing
+        raise
 
 
 def get_job_url(query_job: GenericJob):