|
3 | 3 | from airflow.decorators import dag, task |
4 | 4 | from airflow.operators.dummy import DummyOperator |
5 | 5 | from airflow.operators.python import BranchPythonOperator, PythonOperator |
6 | | -from airflow.utils.trigger_rule import TriggerRule |
7 | 6 | from airflow.timetables.trigger import CronTriggerTimetable |
| 7 | +from airflow.utils.trigger_rule import TriggerRule |
8 | 8 |
|
9 | 9 | from dags import DEFAULT_DAG_ARGUMENTS |
10 | 10 | from dags.dags_utils import get_dag_param, push_dag_downstream, pull_dag_upstream |
11 | 11 | from dags.operators.DagBatchPipelineOperator import NOTICE_IDS_KEY, TriggerNoticeBatchPipelineOperator |
12 | 12 | from dags.pipelines.notice_fetcher_pipelines import notice_fetcher_by_date_pipeline |
| 13 | +from ted_sws import config |
13 | 14 | from ted_sws.event_manager.adapters.event_log_decorator import event_log |
14 | 15 | from ted_sws.event_manager.model.event_message import TechnicalEventMessage, EventMessageMetadata, \ |
15 | 16 | EventMessageProcessType |
|
27 | 28 |
|
28 | 29 | DAG_FETCH_DEFAULT_TIMEZONE = "UTC" |
29 | 30 |
|
| 31 | + |
30 | 32 | @dag(default_args=DEFAULT_DAG_ARGUMENTS, |
| 33 | + dag_id=FETCHER_DAG_NAME, |
31 | 34 | catchup=False, |
32 | | - timetable=CronTriggerTimetable('0 1 * * *', timezone='UTC'), |
| 35 | + timetable=CronTriggerTimetable( |
| 36 | + cron=config.SCHEDULE_DAG_FETCH, |
| 37 | + timezone=DAG_FETCH_DEFAULT_TIMEZONE), |
33 | 38 | tags=['selector', 'daily-fetch']) |
34 | 39 | def fetch_notices_by_date(): |
35 | 40 | @task |
|
0 commit comments