initial

daemons/base.py

@@ -0,0 +1,3 @@
+class Base:
+    def execute(self):
+        raise NotImplemented

daemons/mailbox.py

@@ -0,0 +1,28 @@
+from telebot import apihelper
+
+from daemons import base
+from utils import platform
+from utils import queues
+
+
+class Daemon(base.Daemon, queues.TasksHandlerMixin):
+    def execute(self):
+        self.poll()
+
+    @property
+    def queue_name(self):
+        return 'botalka_mailbox'
+    
+    def process(self, payload):
+        bot = platform.platform_client.get_config('bots')[payload['project']][payload['name']]
+        if not bot['mailbox_enabled']:
+            return
+        if bot['type'] == 'telegram':
+            token = bot['secrets']['telegram_token']
+            self.process_telegram(token, payload['body'])
+
+    def process_telegram(self, token, payload):
+        try:
+            apihelper.send_message(token, **payload)
+        except Exception as exc:
+            print('Error', str(exc))

daemons/poll.py

@@ -0,0 +1,42 @@
+import telebot
+import multiprocessing
+import time
+
+from daemons import base
+from utils import platform
+from utils import queues
+
+
+class Daemon(base.Daemon):
+    def __init__(self):
+        self.telegram_pollers: dict[str, dict[str, multiprocessing.Process|None]] = {}
+
+    def execute(self):
+        bots = platform.platform_client.get_config('bots')
+        for project_name, project in bots.items():
+            if project_name not in self.telegram_pollers:
+                self.telegram_pollers[project_name] = {}
+            for bot_name, bot_info in project.items():
+                if bot_name not in self.telegram_pollers[project_name]:
+                    self.telegram_pollers[project_name][bot_name] = None
+                process = self.telegram_pollers[project_name][bot_name]
+                if bot_info.get('poll_enabled'):
+                    if process is not None and process.is_alive:
+                        continue
+                    new_process = multiprocessing.Process(target=self.start_polling, args=[bot_info['secrets']['telegram_token'], bot_info['queue']])
+                    new_process.start()
+                    self.telegram_pollers[project_name][bot_name] = new_process
+                else:
+                    if process is None:
+                        continue
+                    if process.is_alive:
+                        process.terminate()
+                        self.telegram_pollers[project_name][bot_name] = None
+        time.sleep(10)
+
+    def start_polling(telegram_token, queue):
+        bot = telebot.TeleBot(telegram_token)
+        @bot.message_handler()
+        def do_action(message):
+            queues.set_task(queue, message.json, 1)
+        bot.polling()