code refactoring

processors/status_send.py

@@ -0,0 +1,38 @@
+import os
+from datetime import datetime, timedelta, timezone
+from dotenv import load_dotenv
+from connectors.mattermost import MattermostConnector
+
+load_dotenv()
+
+
+class StatusSend:
+    def __init__(self):
+        conf = {
+            "url": os.getenv("MATTERMOST_SERVER_URL"),
+            "token": os.getenv("MATTERMOST_ACCESS_TOKEN"),
+        }
+        self.connector = MattermostConnector(conf)
+
+    def set_status(self, text, emoji, duration=None, **kwargs):
+        if duration is not None:
+            expires_at = datetime.now(timezone.utc) + timedelta(seconds=duration)
+        if isinstance(emoji, dict):
+            emoji_name = emoji.get("name")
+        else:
+            emoji_name = emoji
+        data = {
+            "emoji": emoji_name,
+            "text": text,
+            "expires_at": expires_at.isoformat() if expires_at else None,
+        }
+        try:
+            self.connector.send(data=data)
+        except Exception as e:
+            print(e)
+
+    def clear_status(self):
+        try:
+            self.connector.send(data={"emoji": "", "text": "", "expires_at": ""})
+        except Exception as e:
+            print(e)