Skip to content

Commit 5f5244b

Browse files
authored
Add template fields renderers to Biguery and Dataproc operators (#12067)
1 parent d971c1c commit 5f5244b

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

β€Žairflow/providers/google/cloud/operators/bigquery.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -879,7 +879,7 @@ class BigQueryCreateEmptyTableOperator(BaseOperator):
879879
'view',
880880
'impersonation_chain',
881881
)
882-
882+
template_fields_renderers = {"table_resource": "json"}
883883
ui_color = BigQueryUIColors.TABLE.value
884884

885885
# pylint: disable=too-many-arguments
@@ -1074,6 +1074,7 @@ class BigQueryCreateExternalTableOperator(BaseOperator):
10741074
'table_resource',
10751075
'impersonation_chain',
10761076
)
1077+
template_fields_renderers = {"table_resource": "json"}
10771078
ui_color = BigQueryUIColors.TABLE.value
10781079

10791080
# pylint: disable=too-many-arguments,too-many-locals
@@ -1369,6 +1370,7 @@ class BigQueryCreateEmptyDatasetOperator(BaseOperator):
13691370
'dataset_reference',
13701371
'impersonation_chain',
13711372
)
1373+
template_fields_renderers = {"dataset_reference": "json"}
13721374
ui_color = BigQueryUIColors.DATASET.value
13731375

13741376
@apply_defaults
@@ -1615,6 +1617,7 @@ class BigQueryPatchDatasetOperator(BaseOperator):
16151617
'project_id',
16161618
'impersonation_chain',
16171619
)
1620+
template_fields_renderers = {"dataset_resource": "json"}
16181621
ui_color = BigQueryUIColors.DATASET.value
16191622

16201623
@apply_defaults
@@ -1705,6 +1708,7 @@ class BigQueryUpdateDatasetOperator(BaseOperator):
17051708
'project_id',
17061709
'impersonation_chain',
17071710
)
1711+
template_fields_renderers = {"dataset_resource": "json"}
17081712
ui_color = BigQueryUIColors.DATASET.value
17091713

17101714
@apply_defaults
@@ -1876,6 +1880,7 @@ class BigQueryUpsertTableOperator(BaseOperator):
18761880
'table_resource',
18771881
'impersonation_chain',
18781882
)
1883+
template_fields_renderers = {"table_resource": "json"}
18791884
ui_color = BigQueryUIColors.TABLE.value
18801885

18811886
@apply_defaults
@@ -1993,6 +1998,7 @@ class BigQueryInsertJobOperator(BaseOperator):
19931998
"impersonation_chain",
19941999
)
19952000
template_ext = (".json",)
2001+
template_fields_renderers = {"configuration": "json"}
19962002
ui_color = BigQueryUIColors.QUERY.value
19972003

19982004
def __init__(

β€Žairflow/providers/google/cloud/operators/dataproc.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1699,6 +1699,7 @@ class DataprocInstantiateInlineWorkflowTemplateOperator(BaseOperator):
16991699
"""
17001700

17011701
template_fields = ['template', 'impersonation_chain']
1702+
template_fields_renderers = {"template": "json"}
17021703

17031704
@apply_defaults
17041705
def __init__(

0 commit comments

Comments
 (0)