Skip to content

Commit 306cafe

Browse files
Update daily_supra_notice_manager.py
1 parent 0a5a462 commit 306cafe

1 file changed

Lines changed: 6 additions & 4 deletions

File tree

ted_sws/supra_notice_manager/services/daily_supra_notice_manager.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,21 @@
11
from typing import List
22

33
from pymongo import MongoClient
4-
4+
from datetime import date, datetime, time
55
from ted_sws.core.model.supra_notice import DailySupraNotice
66
from ted_sws.data_manager.adapters.supra_notice_repository import DailySupraNoticeRepository
77

88

9-
def create_and_store_in_mongo_db_daily_supra_notice(notice_ids: List[str], mongodb_client: MongoClient):
9+
def create_and_store_in_mongo_db_daily_supra_notice(notice_ids: List[str], mongodb_client: MongoClient,
10+
notice_publication_date: date = date.today()):
1011
"""
1112
This function creates and stores a DailySupraNotice in MongoDB.
1213
:param notice_ids:
1314
:param mongodb_client:
15+
:param notice_publication_date:
1416
:return:
1517
"""
1618
daily_supra_notice_repository = DailySupraNoticeRepository(mongodb_client=mongodb_client)
17-
daily_supra_notice = DailySupraNotice(notice_ids=notice_ids)
19+
daily_supra_notice = DailySupraNotice(notice_publication_date=notice_publication_date,
20+
notice_ids=notice_ids)
1821
daily_supra_notice_repository.add(daily_supra_notice=daily_supra_notice)
19-

0 commit comments

Comments
 (0)