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 (Iterable[str] | None) – the target statuses, sensor waits until the task reaches any of these states

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

Get DmsHook.

hook()[source]
poke(context)[source]

Function defined by the sensors 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 sensor, take a look at the guide: Wait for a replication task to complete

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?