|
1 | | -from airflow import DAG |
2 | 1 | from datetime import datetime, timedelta |
3 | | -from airflow.operators.bash_operator import BashOperator |
4 | | -from airflow.operators.docker_operator import DockerOperator |
5 | 2 |
|
| 3 | +from airflow import DAG |
| 4 | +from airflow.providers.docker.operators.docker import DockerOperator |
6 | 5 |
|
7 | 6 | 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), |
16 | 15 | } |
17 | 16 |
|
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: |
19 | 23 | 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", |
23 | 27 | auto_remove=True, |
24 | 28 | command="/bin/sleep 30", |
25 | 29 | docker_url="unix://var/run/docker.sock", |
26 | | - network_mode="bridge" |
| 30 | + network_mode="bridge", |
27 | 31 | ) |
28 | 32 |
|
29 | 33 | t1 |
30 | | - |
|
0 commit comments