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

import pendulum

local_tz = pendulum.timezone("Asia/Shanghai")

args = {
    'owner': 'bi-DPB',
    'start_date': datetime(2024, 8, 29, tzinfo=local_tz),
    'email': 'xxxxx@lenztechretail.com',
    'email_on_failure': False,
    'email_on_retry': False,
    'retries': 3,
    'retry_delay': timedelta(minutes=1),
}

dag = DAG(
    dag_id='DPB_hsm',
    default_args=args,
    schedule_interval='00 02 * * *',
    dagrun_timeout=timedelta(minutes=200))

t1 = BashOperator(task_id="pg_w_new",
                  bash_command='cd /usr/local/airflow/dags/bi_etl/bi_project_airflow/project_common/;nohup python3 pg_hsm/pg_hsm_w_new4.py',
                  dag=dag)