SIENTIAPDE-1163

Refactor notification handler imports across multiple files to use CoreNotificationHandler instead of NotificationHandler for consistency and clarity.

scouter/activities/activities.py

@@ -2,7 +2,7 @@ from temporalio import workflow
 
 with workflow.unsafe.imports_passed_through():
     from sientia_do.temporal.activities.postgres import Postgres
-    from sientia_do.notifications.handlers import NotificationHandler
+    from sientia_do.notifications.handlers import CoreNotificationHandler as NotificationHandler
     from sientia_do.temporal.utils.logger import Logger
     from scouter.activities.redis import Redis
     from scouter.activities.gates import Gates

scouter/activities/faker.py

@@ -5,7 +5,7 @@ import json
 from kafka import KafkaProducer
 from temporalio import activity
 
-from sientia_do.notifications.handlers import NotificationHandler
+from sientia_do.notifications.handlers import CoreNotificationHandler as NotificationHandler
 from sientia_do.temporal.activities.base import BaseActivity
 from sientia_do.temporal.utils.logger import Logger
 

scouter/activities/mongodb.py

@@ -6,7 +6,7 @@ with workflow.unsafe.imports_passed_through():
     from datetime import datetime
     from pymongo import MongoClient
     from pandas import DataFrame
-    from sientia_do.notifications.handlers import NotificationHandler
+    from sientia_do.notifications.handlers import CoreNotificationHandler as NotificationHandler
     from sientia_do.notifications.models import NotificationLevel
     from sientia_do.temporal.activities.base import BaseActivity
     from sientia_do.temporal.utils.logger import Logger

scouter/activities/redis.py

@@ -3,7 +3,7 @@ from temporalio import workflow, activity
 
 with workflow.unsafe.imports_passed_through():
     from logging import Logger
-    from sientia_do.notifications.handlers import NotificationHandler
+    from sientia_do.notifications.handlers import CoreNotificationHandler as NotificationHandler
     from sientia_do.notifications.models import NotificationLevel
     from sientia_do.temporal.activities.redis_base import Redis as RedisBase
     from sientia_do.temporal.utils.logger import Logger

scouter/worker/worker.py

@@ -4,7 +4,7 @@ from temporalio.worker import Worker
 with workflow.unsafe.imports_passed_through():
     import sys
     import os
-    from sientia_do.notifications.handlers import NotificationHandler
+    from sientia_do.notifications.handlers import CoreNotificationHandler as NotificationHandler
     from sientia_do.temporal.utils.logger import get_logger
     from scouter.activities.activities import Activities
     from scouter.workflow.scouter import Scouter
@@ -28,8 +28,10 @@ async def main():
 
     logger.info('Starting Notification Handler...')
 
+    mongo_config = build_mongodb_config()
     notification_handler = NotificationHandler(
-        servers=os.getenv('KAFKA_BOOTSTRAP_SERVERS', 'http://localhost:9092'),
+        connection_string=mongo_config['connection_string'],
+        database=mongo_config['database_name'],
         logger=logger,
         project_name=os.getenv('PROJECT_NAME', 'scouter')
     )

tests/activities/test_faker.py

@@ -1,6 +1,6 @@
 from unittest.mock import MagicMock, patch, call
 import pytest
-from sientia_do.notifications.handlers import NotificationHandler
+from sientia_do.notifications.handlers import CoreNotificationHandler as NotificationHandler
 from scouter.activities.faker import Faker
 
 

tests/activities/test_redis.py

@@ -3,7 +3,7 @@ from datetime import datetime
 import pytest
 import numpy as np
 from pandas import DataFrame
-from sientia_do.notifications.handlers import NotificationHandler
+from sientia_do.notifications.handlers import CoreNotificationHandler as NotificationHandler
 from sientia_do.notifications.models import NotificationLevel
 from scouter.activities.redis import Redis