Skip to content

Commit cc2a29a

Browse files
🐛 fix: import DockerOperator for Airflow 3
1 parent d3b53aa commit cc2a29a

1 file changed

Lines changed: 20 additions & 17 deletions

File tree

examples/dags/sample_docker_operator_dag.py

Lines changed: 20 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -1,30 +1,33 @@
1-
from airflow import DAG
21
from datetime import datetime, timedelta
3-
from airflow.operators.bash_operator import BashOperator
4-
from airflow.operators.docker_operator import DockerOperator
52

3+
from airflow import DAG
4+
from airflow.providers.docker.operators.docker import DockerOperator
65

76
default_args = {
8-
'owner' : 'Mostafa Ghadimi',
9-
'description' : 'Use of the DockerOperator',
10-
'depend_on_past' : False,
11-
'start_date' : datetime(2023, 4, 23),
12-
'email_on_failure' : False,
13-
'email_on_retry' : False,
14-
'retries' : 1,
15-
'retry_delay' : timedelta(minutes=5)
7+
"owner": "Mostafa Ghadimi",
8+
"description": "Use of the DockerOperator",
9+
"depend_on_past": False,
10+
"start_date": datetime(2023, 4, 23),
11+
"email_on_failure": False,
12+
"email_on_retry": False,
13+
"retries": 1,
14+
"retry_delay": timedelta(minutes=5),
1615
}
1716

18-
with DAG('docker_dag_v2', default_args=default_args, schedule_interval="5 10 * * *", catchup=False) as dag:
17+
with DAG(
18+
"docker_dag_v2",
19+
default_args=default_args,
20+
schedule="5 10 * * *",
21+
catchup=False,
22+
) as dag:
1923
t1 = DockerOperator(
20-
task_id='docker_command',
21-
image='alpine:latest',
22-
api_version='auto',
24+
task_id="docker_command",
25+
image="alpine:latest",
26+
api_version="auto",
2327
auto_remove=True,
2428
command="/bin/sleep 30",
2529
docker_url="unix://var/run/docker.sock",
26-
network_mode="bridge"
30+
network_mode="bridge",
2731
)
2832

2933
t1
30-

0 commit comments

Comments
 (0)