airflow.providers.amazon.aws.sensors.dms

Module Contents

Classes

DmsTaskBaseSensor

Contains general sensor behavior for DMS task.

DmsTaskCompletedSensor

Pokes DMS task until it is completed.
class airflow.providers.amazon.aws.sensors.dms.DmsTaskBaseSensor(replication_task_arn: str, aws_conn_id='aws_default', target_statuses: Optional[Iterable[str]] = None, termination_statuses: Optional[Iterable[str]] = None, *args, **kwargs)[source]

Bases: airflow.sensors.base.BaseSensorOperator

Contains general sensor behavior for DMS task.

Subclasses should set target_statuses and termination_statuses fields.

Parameters

  • replication_task_arn (str) -- AWS DMS replication task ARN

  • aws_conn_id (str) -- aws connection to uses

  • target_statuses -- the target statuses, sensor waits until the task reaches any of these states

  • termination_statuses (list[str]) -- the termination statuses, sensor fails when the task reaches any of these states

template_fields :Sequence[str] = ['replication_task_arn'][source]
template_ext :Sequence[str] = [][source]
get_hook(self) → airflow.providers.amazon.aws.hooks.dms.DmsHook[source]

Get DmsHook

poke(self, context: airflow.utils.context.Context)[source]

Function that the sensors defined while deriving this class should override.

class airflow.providers.amazon.aws.sensors.dms.DmsTaskCompletedSensor(*args, **kwargs)[source]

Bases: DmsTaskBaseSensor

Pokes DMS task until it is completed.

See also

For more information on how to use this operator, take a look at the guide: Create replication task, wait for it completion and delete it.

Parameters

replication_task_arn (str) -- AWS DMS replication task ARN

template_fields :Sequence[str] = ['replication_task_arn'][source]
template_ext :Sequence[str] = [][source]

Was this entry helpful?