Replacing non-attribute template_fields for BigQueryToMsSqlOperator (#…

…19052) * Replacing non-attribute template_fields for BigQueryToMsSqlOperator * Updating source_project_dataset_table arg in example DAG
apache · Oct 22, 2021 · d9192a5 · d9192a5
1 parent 4d1f14a
commit d9192a5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 7 deletions.
diff --git a/airflow/providers/google/cloud/example_dags/example_bigquery_to_mssql.py b/airflow/providers/google/cloud/example_dags/example_bigquery_to_mssql.py
@@ -44,7 +44,7 @@
 ) as dag:
  bigquery_to_mssql = BigQueryToMsSqlOperator(
  task_id="bigquery_to_mssql",
- source_project_dataset_table=f'{DATASET_NAME}.{TABLE}',
+ source_project_dataset_table=f'{PROJECT_ID}.{DATASET_NAME}.{TABLE}',
  mssql_table=destination_table,
  replace=False,
  )

diff --git a/airflow/providers/google/cloud/transfers/bigquery_to_mssql.py b/airflow/providers/google/cloud/transfers/bigquery_to_mssql.py
@@ -82,12 +82,7 @@ class BigQueryToMsSqlOperator(BaseOperator):
  :type impersonation_chain: Union[str, Sequence[str]]
  """
 
- template_fields = (
- 'dataset_id',
- 'table_id',
- 'mssql_table',
- 'impersonation_chain',
- )
+ template_fields = ('source_project_dataset_table', 'mssql_table', 'impersonation_chain')
 
  def __init__(
  self,