command

daemons/management/commands/email_sender.py

@@ -1,20 +0,0 @@
-from django.core.mail import send_mail
-
-from Sprint.settings import EMAIL_HOST_USER
-from SprintLib.queue import MessagingSupport
-
-
-class Command(MessagingSupport):
-    help = "starts file email sender"
-    queue_name = "email"
-
-    def process(self, payload: dict):
-        subject = payload['subject']
-        message = payload['message']
-        email = payload['email']
-        send_mail(
-            subject,
-            message,
-            EMAIL_HOST_USER,
-            [email]
-        )

daemons/management/commands/notification_manager.py

@@ -1,13 +1,36 @@
 from django.contrib.auth.models import User
+from django.core.mail import send_mail
 
 from Main.models import Solution, Progress
+from Sprint.settings import EMAIL_HOST_USER
 from SprintLib.queue import MessagingSupport, send_to_queue
+from daemons.management.commands.bot import bot
 
 
 class Command(MessagingSupport):
     help = "starts notification manager"
     queue_name = "notification"
 
+    def send_email(self, payload):
+        subject = payload['subject']
+        message = payload['message']
+        email = payload['email']
+        send_mail(
+            subject,
+            message,
+            EMAIL_HOST_USER,
+            [email]
+        )
+
+    def send_telegram(self, payload):
+        chat_id = payload['chat_id']
+        text = payload['text']
+        bot.send_message(
+            int(chat_id),
+            text,
+            parse_mode="html",
+        )
+
     def handle_solution(self, payload):
         solution = Solution.objects.get(id=payload["solution_id"])
         user = solution.user
@@ -50,4 +73,9 @@ class Command(MessagingSupport):
         if handler is None:
             raise ValueError(f"Unknown type: {notification_type}")
         for queue, payload in handler(payload):
-            send_to_queue(queue, payload)
+            if queue == 'telegram':
+                self.send_telegram(payload)
+            elif queue == 'email':
+                self.send_email(payload)
+            else:
+                raise ValueError(f"Unknown type {queue}")

daemons/management/commands/telegram_sender.py

@@ -1,16 +0,0 @@
-from SprintLib.queue import MessagingSupport
-from daemons.management.commands.bot import bot
-
-
-class Command(MessagingSupport):
-    help = "starts file telegram sender"
-    queue_name = "telegram"
-
-    def process(self, payload: dict):
-        chat_id = payload['chat_id']
-        text = payload['text']
-        bot.send_message(
-            int(chat_id),
-            text,
-            parse_mode="html",
-        )