SIENTIAPDE-994

Update requirements.txt with new dependencies and refactor activity methods for improved functionality and error handling

laborious/activities/opc.py

@@ -0,0 +1,76 @@
+import traceback
+from pandas import DataFrame
+from temporalio import activity, workflow
+
+
+with workflow.unsafe.imports_passed_through():
+    from logging import Logger
+    from sientia_do.notifications.handlers import NotificationHandler
+    from laborious.activities.base import BaseActivity
+    from laborious.utils.repository.opc_repository import OpcRepository
+    from typing import Any
+    from sientia_do.notifications.models import NotificationLevel
+
+
+class OPC(BaseActivity):
+    def __init__(self,
+                 name: str, url: str, server_uri: str,
+                 cert_path: str, private_key_path: str, server_cert_path: str,
+                 logger: Logger, notification_handler: NotificationHandler):
+
+        self.logger = logger
+        self.notification_handler = notification_handler
+        self.name = name
+        self.url = url
+        self.server_uri = server_uri
+        self.cert_path = cert_path
+        self.private_key_path = private_key_path
+        self.server_cert_path = server_cert_path
+
+        self.opc_repository = OpcRepository(
+            name=self.name,
+            url=self.url,
+            logger=self.logger,
+            server_uri=self.server_uri,
+            cert_path=self.cert_path,
+            private_key_path=self.private_key_path,
+            server_cert_path=self.server_cert_path
+        )
+
+        self.opc_repository.connect()
+
+    @activity.defn(name='write_opc_data')
+    async def write_opc_data(self, input_data: dict[str, Any]):
+        data = DataFrame(input_data['data'])
+        _opc_servers = input_data['opc_servers']
+        opc_output_config = input_data['opc_output_config']
+
+        for tag, config in opc_output_config['prediction_tags'].items():
+            try:
+                self.opc_repository.write_data(
+                    tag, data.head(1)['prediction'].values[0], config['data_type'])
+            except Exception as e:
+                trace = traceback.format_exc()
+                self.notification_handler.build_and_send_notification(
+                    notification_id="WRITE_OPC_PREDICTION_ERROR",
+                    message=f"Error writing data to OPC server: {e}",
+                    block="write_opc_data",
+                    level=NotificationLevel.ERROR,
+                    attachment_content=trace
+                )
+                self.logger.error(trace)
+
+        for tag, config in opc_output_config['confidence_tags'].items():
+            try:
+                self.opc_repository.write_data(
+                    tag, data.head(1)['prediction_confidence'].values[0], config['data_type'])
+            except Exception as e:
+                trace = traceback.format_exc()
+                self.notification_handler.build_and_send_notification(
+                    notification_id="WRITE_OPC_CONFIDENCE_ERROR",
+                    message=f"Error writing data to OPC server: {e}",
+                    block="write_opc_data",
+                    level=NotificationLevel.ERROR,
+                    attachment_content=trace
+                )
+                self.logger.error(trace)