import datetime

from airflow import DAG
from airflow.operators.python_operator import PythonOperator

default_args = {
    'owner': 'airflow',
    'execution_timeout': datetime.timedelta(seconds=300),
    'provide_context': True,
}

with DAG(
        'dummy_dag',
        default_args=default_args,
        catchup=False,
        start_date=datetime.datetime(2016, 12, 2),
) as dag_test_different_queues:
    test_task = PythonOperator(
        task_id='test_task',
        python_callable=lambda *args, **kwargs: print('Hello, world!'),
    )

    test_task