Skip to content

Commit 7b39eac

Browse files
Update selector_retransform_process_orchestrator.py
1 parent 961d7e6 commit 7b39eac

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

dags/selector_retransform_process_orchestrator.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,10 @@
88
from ted_sws.core.model.notice import NoticeStatus
99
from ted_sws.data_manager.adapters.notice_repository import NoticeRepository
1010
from ted_sws.event_manager.adapters.event_log_decorator import event_log
11+
from ted_sws.event_manager.adapters.event_logger import EventLogger
1112
from ted_sws.event_manager.model.event_message import TechnicalEventMessage, EventMessageMetadata, \
12-
EventMessageProcessType
13+
EventMessageProcessType, EventMessage
14+
from ted_sws.event_manager.services.logger_from_context import get_logger_from_dag_context
1315

1416
DAG_NAME = "selector_re_transform_process_orchestrator"
1517

@@ -32,10 +34,12 @@ def selector_re_transform_process_orchestrator():
3234
process_type=EventMessageProcessType.DAG, process_name=DAG_NAME
3335
))
3436
)
35-
def select_notices_for_re_transform_and_reset_status():
37+
def select_notices_for_re_transform_and_reset_status(**context_args):
38+
event_logger: EventLogger = get_logger_from_dag_context(context_args)
3639
mongodb_client = MongoClient(config.MONGO_DB_AUTH_URL)
3740
notice_repository = NoticeRepository(mongodb_client=mongodb_client)
3841
for target_notice_state in RE_TRANSFORM_TARGET_NOTICE_STATES:
42+
event_logger.info(event_message=EventMessage(message=f"select notices with status : {target_notice_state}"))
3943
notices = notice_repository.get_notice_by_status(notice_status=target_notice_state)
4044
for notice in notices:
4145
notice.update_status_to(new_status=NoticeStatus.NORMALISED_METADATA)

0 commit comments

Comments
 (0)