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, aws_conn_id='aws_default', target_statuses=None, termination_statuses=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 – aws connection to uses

  • target_statuses (Optional[Iterable[str]]) – the target statuses, sensor waits until the task reaches any of these states

  • termination_statuses (Optional[Iterable[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)[source]

Get DmsHook

poke(self, 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 – AWS DMS replication task ARN

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

Was this entry helpful?