Airflow -告诉DAG跳过每个月2号的处理

from datetime import datetime

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

def process_data():
    # 处理数据的逻辑代码

def check_date():
    if datetime.now().day == 2:
        return 'skip_task'
    else:
        return 'process_data'

dag = DAG(
    'my_dag',
    schedule_interval='@monthly',
    start_date=datetime(2022, 1, 1)
)

skip_task = PythonOperator(
    task_id='skip_task',
    python_callable=lambda: None,
    dag=dag
)

process_data_task = PythonOperator(
    task_id='process_data',
    python_callable=process_data,
    dag=dag
)

check_date_task = PythonOperator(
    task_id='check_date',
    python_callable=check_date,
    dag=dag
)

check_date_task >> [skip_task, process_data_task]