astronomer.providers.google.cloud.operators.dataproc

This module contains Google Dataproc operators.

Module Contents

Classes

DataprocSubmitJobOperatorAsync

Submits a job to a cluster and wait until is completely finished or any error occurs.
class astronomer.providers.google.cloud.operators.dataproc.DataprocSubmitJobOperatorAsync(*, job, project_id=None, region=None, location=None, request_id=None, retry=None, timeout=None, metadata=(), gcp_conn_id='google_cloud_default', impersonation_chain=None, asynchronous=False, cancel_on_kill=True, wait_timeout=None, **kwargs)[source]

Bases: airflow.providers.google.cloud.operators.dataproc.DataprocSubmitJobOperator

Submits a job to a cluster and wait until is completely finished or any error occurs.

Parameters

  • project_id (Optional[str]) – Optional. The ID of the Google Cloud project that the job belongs to.

  • region (Optional[str]) – Required. The Cloud Dataproc region in which to handle the request.

  • location (Optional[str]) – (To be deprecated). The Cloud Dataproc region in which to handle the request.

  • job (Dict) – Required. The job resource. If a dict is provided, it must be of the same form as the protobuf message class:~google.cloud.dataproc_v1.types.Job

  • request_id (Optional[str]) – Optional. A unique id used to identify the request. If the server receives two SubmitJobRequest requests with the same id, then the second request will be ignored and the first Job created and stored in the backend is returned. It is recommended to always set this value to a UUID.

  • retry (Optional[google.api_core.retry.Retry]) – A retry object used to retry requests. If None is specified, requests will not be retried.

  • timeout (Optional[float]) – The amount of time, in seconds, to wait for the request to complete. Note that if retry is specified, the timeout applies to each individual attempt.

  • metadata (Sequence[Tuple[str, str]]) – Additional metadata that is provided to the method.

  • gcp_conn_id (str) – The connection ID to use connecting to Google Cloud Platform.

  • impersonation_chain (Optional[Union[str, Sequence[str]]]) – Optional service account to impersonate using short-term credentials, or chained list of accounts required to get the access_token of the last account in the list, which will be impersonated in the request. If set as a string, the account must grant the originating account the Service Account Token Creator IAM role. If set as a sequence, the identities from the list must grant Service Account Token Creator IAM role to the directly preceding identity, with first account from the list granting this role to the originating account (templated).

  • cancel_on_kill (bool) – Flag which indicates whether cancel the hook’s job or not, when on_kill is called

execute(self, context)[source]

Airflow runs this method on the worker and defers using the trigger. Submit the job and get the job_id using which we defer and poll in trigger

execute_complete(self, context, event=None)[source]

Callback for when the trigger fires - returns immediately. Relies on trigger to throw an exception, otherwise it assumes execution was successful.