Add template fields renderers to Biguery and Dataproc operators (#12067)

turbaszek · web-flow · commit 5f5244b74df9 · 2020-11-04T11:30:03.000+01:00
diff --git a/airflow/providers/google/cloud/operators/bigquery.py b/airflow/providers/google/cloud/operators/bigquery.py
@@ -879,7 +879,7 @@ class BigQueryCreateEmptyTableOperator(BaseOperator):
         'view',
         'impersonation_chain',
     )
-
+    template_fields_renderers = {"table_resource": "json"}
     ui_color = BigQueryUIColors.TABLE.value
 
     # pylint: disable=too-many-arguments
@@ -1074,6 +1074,7 @@ class BigQueryCreateExternalTableOperator(BaseOperator):
         'table_resource',
         'impersonation_chain',
     )
+    template_fields_renderers = {"table_resource": "json"}
     ui_color = BigQueryUIColors.TABLE.value
 
     # pylint: disable=too-many-arguments,too-many-locals
@@ -1369,6 +1370,7 @@ class BigQueryCreateEmptyDatasetOperator(BaseOperator):
         'dataset_reference',
         'impersonation_chain',
     )
+    template_fields_renderers = {"dataset_reference": "json"}
     ui_color = BigQueryUIColors.DATASET.value
 
     @apply_defaults
@@ -1615,6 +1617,7 @@ class BigQueryPatchDatasetOperator(BaseOperator):
         'project_id',
         'impersonation_chain',
     )
+    template_fields_renderers = {"dataset_resource": "json"}
     ui_color = BigQueryUIColors.DATASET.value
 
     @apply_defaults
@@ -1705,6 +1708,7 @@ class BigQueryUpdateDatasetOperator(BaseOperator):
         'project_id',
         'impersonation_chain',
     )
+    template_fields_renderers = {"dataset_resource": "json"}
     ui_color = BigQueryUIColors.DATASET.value
 
     @apply_defaults
@@ -1876,6 +1880,7 @@ class BigQueryUpsertTableOperator(BaseOperator):
         'table_resource',
         'impersonation_chain',
     )
+    template_fields_renderers = {"table_resource": "json"}
     ui_color = BigQueryUIColors.TABLE.value
 
     @apply_defaults
@@ -1993,6 +1998,7 @@ class BigQueryInsertJobOperator(BaseOperator):
         "impersonation_chain",
     )
     template_ext = (".json",)
+    template_fields_renderers = {"configuration": "json"}
     ui_color = BigQueryUIColors.QUERY.value
 
     def __init__(
diff --git a/airflow/providers/google/cloud/operators/dataproc.py b/airflow/providers/google/cloud/operators/dataproc.py
@@ -1699,6 +1699,7 @@ class DataprocInstantiateInlineWorkflowTemplateOperator(BaseOperator):
     """
 
     template_fields = ['template', 'impersonation_chain']
+    template_fields_renderers = {"template": "json"}
 
     @apply_defaults
     def __init__(
