airflow.providers.amazon.aws.hooks.dms

Module Contents

Classes

DmsTaskWaiterStatus

Available AWS DMS Task Waiter statuses.

DmsHook

Interact with AWS Database Migration Service.
class airflow.providers.amazon.aws.hooks.dms.DmsTaskWaiterStatus[source]

Bases: str, enum.Enum

Available AWS DMS Task Waiter statuses.

DELETED = deleted[source]
READY = ready[source]
RUNNING = running[source]
STOPPED = stopped[source]
class airflow.providers.amazon.aws.hooks.dms.DmsHook(*args, **kwargs)[source]

Bases: airflow.providers.amazon.aws.hooks.base_aws.AwsBaseHook

Interact with AWS Database Migration Service.

describe_replication_tasks(**kwargs)[source]

Describe replication tasks

Returns

Marker and list of replication tasks

Return type

tuple[str | None, list]

find_replication_tasks_by_arn(replication_task_arn, without_settings=False)[source]

Find and describe replication tasks by task ARN :param replication_task_arn: Replication task arn :param without_settings: Indicates whether to return task information with settings.

Returns

list of replication tasks that match the ARN

get_task_status(replication_task_arn)[source]

Retrieve task status.

Parameters

replication_task_arn (str) – Replication task ARN

Returns

Current task status

Return type

str | None

create_replication_task(replication_task_id, source_endpoint_arn, target_endpoint_arn, replication_instance_arn, migration_type, table_mappings, **kwargs)[source]

Create DMS replication task

Parameters

  • replication_task_id (str) – Replication task id

  • source_endpoint_arn (str) – Source endpoint ARN

  • target_endpoint_arn (str) – Target endpoint ARN

  • replication_instance_arn (str) – Replication instance ARN

  • table_mappings (dict) – Table mappings

  • migration_type (str) – Migration type (‘full-load’|’cdc’|’full-load-and-cdc’), full-load by default.

Returns

Replication task ARN

Return type

str

start_replication_task(replication_task_arn, start_replication_task_type, **kwargs)[source]

Starts replication task.

Parameters

  • replication_task_arn (str) – Replication task ARN

  • start_replication_task_type (str) – Replication task start type (default=’start-replication’) (‘start-replication’|’resume-processing’|’reload-target’)

stop_replication_task(replication_task_arn)[source]

Stops replication task.

Parameters

replication_task_arn – Replication task ARN

delete_replication_task(replication_task_arn)[source]

Starts replication task deletion and waits for it to be deleted

Parameters

replication_task_arn – Replication task ARN

wait_for_task_status(replication_task_arn, status)[source]

Waits for replication task to reach status. Supported statuses: deleted, ready, running, stopped.

Parameters

Was this entry helpful?