prevent templated field logic checks in operators __init__ (#36489)

apache · Dec 29, 2023 · 918552a · 918552a
1 parent 718efc0
commit 918552a
Showing 1 changed file with 90 additions and 6 deletions.
diff --git a/airflow/providers/google/cloud/operators/dataproc.py b/airflow/providers/google/cloud/operators/dataproc.py
@@ -1297,6 +1297,12 @@ def __init__(
         query: str | None = None,
         query_uri: str | None = None,
         variables: dict | None = None,
+        impersonation_chain: str | Sequence[str] | None = None,
+        region: str,
+        job_name: str = "{{task.task_id}}_{{ds_nodash}}",
+        cluster_name: str = "cluster-1",
+        dataproc_properties: dict | None = None,
+        dataproc_jars: list[str] | None = None,
         **kwargs,
     ) -> None:
         # TODO: Remove one day
@@ -1308,7 +1314,15 @@ def __init__(
             stacklevel=1,
         )
 
-        super().__init__(**kwargs)
+        super().__init__(
+            impersonation_chain=impersonation_chain,
+            region=region,
+            job_name=job_name,
+            cluster_name=cluster_name,
+            dataproc_properties=dataproc_properties,
+            dataproc_jars=dataproc_jars,
+            **kwargs,
+        )
         self.query = query
         self.query_uri = query_uri
         self.variables = variables
@@ -1375,6 +1389,12 @@ def __init__(
         query: str | None = None,
         query_uri: str | None = None,
         variables: dict | None = None,
+        impersonation_chain: str | Sequence[str] | None = None,
+        region: str,
+        job_name: str = "{{task.task_id}}_{{ds_nodash}}",
+        cluster_name: str = "cluster-1",
+        dataproc_properties: dict | None = None,
+        dataproc_jars: list[str] | None = None,
         **kwargs,
     ) -> None:
         # TODO: Remove one day
@@ -1386,7 +1406,15 @@ def __init__(
             stacklevel=1,
         )
 
-        super().__init__(**kwargs)
+        super().__init__(
+            impersonation_chain=impersonation_chain,
+            region=region,
+            job_name=job_name,
+            cluster_name=cluster_name,
+            dataproc_properties=dataproc_properties,
+            dataproc_jars=dataproc_jars,
+            **kwargs,
+        )
         self.query = query
         self.query_uri = query_uri
         self.variables = variables
@@ -1454,6 +1482,12 @@ def __init__(
         query: str | None = None,
         query_uri: str | None = None,
         variables: dict | None = None,
+        impersonation_chain: str | Sequence[str] | None = None,
+        region: str,
+        job_name: str = "{{task.task_id}}_{{ds_nodash}}",
+        cluster_name: str = "cluster-1",
+        dataproc_properties: dict | None = None,
+        dataproc_jars: list[str] | None = None,
         **kwargs,
     ) -> None:
         # TODO: Remove one day
@@ -1465,7 +1499,15 @@ def __init__(
             stacklevel=1,
         )
 
-        super().__init__(**kwargs)
+        super().__init__(
+            impersonation_chain=impersonation_chain,
+            region=region,
+            job_name=job_name,
+            cluster_name=cluster_name,
+            dataproc_properties=dataproc_properties,
+            dataproc_jars=dataproc_jars,
+            **kwargs,
+        )
         self.query = query
         self.query_uri = query_uri
         self.variables = variables
@@ -1535,6 +1577,12 @@ def __init__(
         arguments: list | None = None,
         archives: list | None = None,
         files: list | None = None,
+        impersonation_chain: str | Sequence[str] | None = None,
+        region: str,
+        job_name: str = "{{task.task_id}}_{{ds_nodash}}",
+        cluster_name: str = "cluster-1",
+        dataproc_properties: dict | None = None,
+        dataproc_jars: list[str] | None = None,
         **kwargs,
     ) -> None:
         # TODO: Remove one day
@@ -1546,7 +1594,15 @@ def __init__(
             stacklevel=1,
         )
 
-        super().__init__(**kwargs)
+        super().__init__(
+            impersonation_chain=impersonation_chain,
+            region=region,
+            job_name=job_name,
+            cluster_name=cluster_name,
+            dataproc_properties=dataproc_properties,
+            dataproc_jars=dataproc_jars,
+            **kwargs,
+        )
         self.main_jar = main_jar
         self.main_class = main_class
         self.arguments = arguments
@@ -1612,6 +1668,12 @@ def __init__(
         arguments: list | None = None,
         archives: list | None = None,
         files: list | None = None,
+        impersonation_chain: str | Sequence[str] | None = None,
+        region: str,
+        job_name: str = "{{task.task_id}}_{{ds_nodash}}",
+        cluster_name: str = "cluster-1",
+        dataproc_properties: dict | None = None,
+        dataproc_jars: list[str] | None = None,
         **kwargs,
     ) -> None:
         # TODO: Remove one day
@@ -1623,7 +1685,15 @@ def __init__(
             stacklevel=1,
         )
 
-        super().__init__(**kwargs)
+        super().__init__(
+            impersonation_chain=impersonation_chain,
+            region=region,
+            job_name=job_name,
+            cluster_name=cluster_name,
+            dataproc_properties=dataproc_properties,
+            dataproc_jars=dataproc_jars,
+            **kwargs,
+        )
         self.main_jar = main_jar
         self.main_class = main_class
         self.arguments = arguments
@@ -1712,6 +1782,12 @@ def __init__(
         archives: list | None = None,
         pyfiles: list | None = None,
         files: list | None = None,
+        impersonation_chain: str | Sequence[str] | None = None,
+        region: str,
+        job_name: str = "{{task.task_id}}_{{ds_nodash}}",
+        cluster_name: str = "cluster-1",
+        dataproc_properties: dict | None = None,
+        dataproc_jars: list[str] | None = None,
         **kwargs,
     ) -> None:
         # TODO: Remove one day
@@ -1723,7 +1799,15 @@ def __init__(
             stacklevel=1,
         )
 
-        super().__init__(**kwargs)
+        super().__init__(
+            impersonation_chain=impersonation_chain,
+            region=region,
+            job_name=job_name,
+            cluster_name=cluster_name,
+            dataproc_properties=dataproc_properties,
+            dataproc_jars=dataproc_jars,
+            **kwargs,
+        )
         self.main = main
         self.arguments = arguments
         self.archives = archives