from airflow.operators.bash_operator import BashOperator
from airflow import DAG
from datetime import datetime, timedelta
import pendulum

# 设置时区
local_tz = pendulum.timezone("Asia/Shanghai")

# 定义DAG，设置为每10分钟执行一次，不限制并行任务数
dag = DAG(
    dag_id='DPB_hsm_audit_link',
    schedule_interval=timedelta(minutes=10),  # 每10分钟执行一次
    dagrun_timeout=timedelta(minutes=10),
    start_date=datetime(2025, 1, 20, tzinfo=local_tz),
)

# 使用BashOperator来执行Python脚本
t1 = BashOperator(
    task_id="pg_audit_link",
    bash_command='cd /usr/local/airflow/dags/bi_etl/bi_project_airflow/project_common/; nohup python3 pg_hsm/pg_hsm_audit_link.py',
    dag=dag
)