import pendulum
from airflow.api_fastapi.execution_api.datamodels.taskinstance import DagRun
from airflow.sdk import dag, task, task_group
from airflow.providers.common.sql.operators.sql import SQLExecuteQueryOperator


@dag(
    schedule=None,
    start_date=pendulum.datetime(2025, 7, 14, tz="UTC"),
    catchup=True,
    tags=["example"],
)
def cs():
    """
    测试
    """

    @task.bash()
    def cs_bash(dag_run: DagRun=None):
        d = dag_run.data_interval_start
        d = d.strftime("%Y-%m-%d")
        return f"echo {d}"

    create_pet_table = SQLExecuteQueryOperator(
        task_id="create_pet_table",
        conn_id="cs",
        sql="""
                delete from t1
              """,
    )


    cs_bash() >> create_pet_table

cs()