Skip to content

Commit c615c10

Browse files
Update selector_daily_fetch_orchestrator.py
1 parent 8fca710 commit c615c10

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

dags/selector_daily_fetch_orchestrator.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
import datetime
1+
from datetime import datetime, timedelta
22
from typing import List
33

44
from airflow.decorators import dag, task
@@ -40,13 +40,15 @@ def fetch_notice_from_ted():
4040
if WILD_CARD_PARAM in dag_params.keys():
4141
current_datetime_wildcard = dag_params[WILD_CARD_PARAM]
4242
else:
43-
current_datetime_wildcard = (datetime.datetime.now() - datetime.timedelta(days=1)).strftime("%Y%m%d*")
43+
current_datetime_wildcard = (datetime.now() - timedelta(days=1)).strftime("%Y%m%d*")
44+
notice_publication_date = datetime.strptime(current_datetime_wildcard, "%Y%m%d*").date()
4445
mongodb_client = MongoClient(config.MONGO_DB_AUTH_URL)
4546
notice_ids = NoticeFetcher(notice_repository=NoticeRepository(mongodb_client=mongodb_client),
4647
ted_api_adapter=TedAPIAdapter(
4748
request_api=TedRequestAPI())).fetch_notices_by_date_wild_card(
4849
wildcard_date=current_datetime_wildcard)
49-
create_and_store_in_mongo_db_daily_supra_notice(notice_ids=notice_ids, mongodb_client=mongodb_client)
50+
create_and_store_in_mongo_db_daily_supra_notice(notice_ids=notice_ids, mongodb_client=mongodb_client,
51+
notice_publication_date=notice_publication_date)
5052
return notice_ids
5153

5254
@task

0 commit comments

Comments
 (0)