airflow_dag_docker_operator_hello_world_quickstart.py

1from datetime import datetime, timedelta
2
3from airflow import DAG
4from airflow.providers.docker.operators.docker import DockerOperator
5
6default_args = {
7    'owner': 'airflow',
8    'depends_on_past': False,
9    'email_on_failure': False,
10    'email_on_retry': False,
11    'retries': 1,
12    'retry_delay': timedelta(minutes=5),
13}
14
15with DAG(
16    'docker_operator_demo',
17    default_args=default_args,
18    schedule_interval=timedelta(days=1),
19    start_date=datetime(2021, 1, 1),
20    catchup=False,
21) as dag:
22
23    t1 = DockerOperator(
24        task_id='docker_command_hello_world',
25        image='hello-world',
26        container_name='task_container_hello',
27        auto_remove=True,
28        command="/bin/sleep 30",
29        docker_url="unix://var/run/docker.sock",
30        network_mode="bridge"
31    )
32
33    t1