diff --git a/airflow/providers/google/cloud/operators/dataproc.py b/airflow/providers/google/cloud/operators/dataproc.py index ad47a9e151602..6faea746e6fa6 100644 --- 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