Fix select * query xcom push for BigQueryGetDataOperator (#22936)

pankajastro · web-flow · commit aa317d92ea4d · 2022-04-13T11:43:38.000+02:00
Use in instead of get for conditinal check
diff --git a/airflow/providers/google/cloud/operators/bigquery.py b/airflow/providers/google/cloud/operators/bigquery.py
@@ -467,6 +467,14 @@ def execute(self, context: 'Context') -> list:
             impersonation_chain=self.impersonation_chain,
         )
 
+        if not self.selected_fields:
+            schema: Dict[str, list] = hook.get_schema(
+                dataset_id=self.dataset_id,
+                table_id=self.table_id,
+            )
+            if "fields" in schema:
+                self.selected_fields = ','.join([field["name"] for field in schema["fields"]])
+
         rows = hook.list_rows(
             dataset_id=self.dataset_id,
             table_id=self.table_id,
