Update worker_single_notice_process_orchestrator.py

Author: CaptainOfHacks

dags/worker_single_notice_process_orchestrator.py

@@ -75,7 +75,11 @@ def _normalise_notice_metadata(**context_args):
 
         notice: Notice = pull_dag_upstream(NOTICE_OBJECT)
         normalised_notice = normalise_notice(notice=notice)
-        push_dag_downstream(NOTICE_OBJECT, normalised_notice)
+
+        mongodb_client = MongoClient(config.MONGO_DB_AUTH_URL)
+        notice_repository = NoticeRepository(mongodb_client=mongodb_client)
+        notice_repository.update(notice=normalised_notice)
+        push_dag_downstream(NOTICE_ID, notice.ted_id)
 
         context = get_current_context()
 
@@ -92,14 +96,14 @@ def _check_eligibility_for_transformation(**context_args):
         event_message: NoticeEventMessage = NoticeEventMessage()
         event_message.start_record()
 
-        notice = pull_dag_upstream(NOTICE_OBJECT)
+        notice_id = pull_dag_upstream(NOTICE_ID)
         mongodb_client = MongoClient(config.MONGO_DB_AUTH_URL)
         notice_repository = NoticeRepository(mongodb_client=mongodb_client)
+        notice = notice_repository.get(reference=notice_id)
         mapping_suite_repository = MappingSuiteRepositoryMongoDB(mongodb_client=mongodb_client)
         result = notice_eligibility_checker(notice=notice, mapping_suite_repository=mapping_suite_repository)
         notice_repository.update(notice=notice)
         mapping_suite_id = None
-        notice_id = notice.ted_id
         if result:
             notice_id, mapping_suite_id = result
             push_dag_downstream(MAPPING_SUITE_ID, mapping_suite_id)