(providers_google) add a location check in bigquery (#19571)

apache · Feb 13, 2022 · 6bb0857 · 6bb0857
1 parent 6579648
commit 6bb0857
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/airflow/providers/google/cloud/hooks/bigquery.py b/airflow/providers/google/cloud/hooks/bigquery.py
@@ -189,6 +189,11 @@ def get_sqlalchemy_engine(self, engine_kwargs=None):
                 "and extra__google_cloud_platform__keyfile_dict"
             )
 
+    def get_records(self, sql, parameters=None):
+        if self.location is None:
+            raise AirflowException("Need to specify 'location' to use BigQueryHook.get_records()")
+        return super().get_records(sql, parameters=parameters)
+
     @staticmethod
     def _resolve_table_reference(
         table_resource: Dict[str, Any],