Update supra_notice_repository.py

Author: CaptainOfHacks

ted_sws/data_manager/adapters/supra_notice_repository.py

@@ -8,8 +8,9 @@
 from ted_sws.data_manager.adapters import inject_date_string_fields, remove_date_string_fields
 from ted_sws.data_manager.adapters.repository_abc import DailySupraNoticeRepositoryABC
 
-DAILY_SUPRA_NOTICE_ID = "notice_fetched_date"
+DAILY_SUPRA_NOTICE_FETCHED_DATE = "notice_fetched_date"
 DAILY_SUPRA_NOTICE_CREATED_AT = "created_at"
+DAILY_SUPRA_NOTICE_ID = "_id"
 
 
 class DailySupraNoticeRepository(DailySupraNoticeRepositoryABC):
@@ -25,7 +26,7 @@ def __init__(self, mongodb_client: MongoClient, database_name: str = None):
         daily_supra_notice_db = mongodb_client[self._database_name]
         self.collection = daily_supra_notice_db[self._collection_name]
         self.collection.create_index(
-            [(DAILY_SUPRA_NOTICE_ID, ASCENDING)])  # TODO: index creation may bring race condition error.
+            [(DAILY_SUPRA_NOTICE_FETCHED_DATE, ASCENDING)])  # TODO: index creation may bring race condition error.
 
     def _create_dict_from_daily_supra_notice(self, daily_supra_notice: DailySupraNotice) -> dict:
         """
@@ -34,9 +35,10 @@ def _create_dict_from_daily_supra_notice(self, daily_supra_notice: DailySupraNot
         :return:
         """
         daily_supra_notice_dict = daily_supra_notice.dict()
-        daily_supra_notice_dict[DAILY_SUPRA_NOTICE_ID] = datetime.combine(
-            daily_supra_notice_dict[DAILY_SUPRA_NOTICE_ID], time())
-        inject_date_string_fields(data=daily_supra_notice_dict, date_field_name=DAILY_SUPRA_NOTICE_ID)
+        daily_supra_notice_dict[DAILY_SUPRA_NOTICE_FETCHED_DATE] = datetime.combine(
+            daily_supra_notice_dict[DAILY_SUPRA_NOTICE_FETCHED_DATE], time())
+        daily_supra_notice_dict[DAILY_SUPRA_NOTICE_ID] = daily_supra_notice_dict[DAILY_SUPRA_NOTICE_FETCHED_DATE]
+        inject_date_string_fields(data=daily_supra_notice_dict, date_field_name=DAILY_SUPRA_NOTICE_FETCHED_DATE)
         inject_date_string_fields(data=daily_supra_notice_dict, date_field_name=DAILY_SUPRA_NOTICE_CREATED_AT)
         return daily_supra_notice_dict
 
@@ -47,8 +49,10 @@ def _create_daily_supra_notice_from_dict(self, daily_supra_notice_dict: dict) ->
         :return:
         """
         if daily_supra_notice_dict is not None:
-            daily_supra_notice_dict[DAILY_SUPRA_NOTICE_ID] = daily_supra_notice_dict[DAILY_SUPRA_NOTICE_ID].date()
-            remove_date_string_fields(data=daily_supra_notice_dict, date_field_name=DAILY_SUPRA_NOTICE_ID)
+            daily_supra_notice_dict.pop(DAILY_SUPRA_NOTICE_ID, None)
+            daily_supra_notice_dict[DAILY_SUPRA_NOTICE_FETCHED_DATE] = daily_supra_notice_dict[
+                DAILY_SUPRA_NOTICE_FETCHED_DATE].date()
+            remove_date_string_fields(data=daily_supra_notice_dict, date_field_name=DAILY_SUPRA_NOTICE_FETCHED_DATE)
             remove_date_string_fields(data=daily_supra_notice_dict, date_field_name=DAILY_SUPRA_NOTICE_CREATED_AT)
             daily_supra_notice = DailySupraNotice.parse_obj(daily_supra_notice_dict)
             return daily_supra_notice