daemons/mailbox.py

@@ -1,5 +1,3 @@
-import pydantic
-
 from telebot import apihelper
 
 from daemons import base
@@ -7,14 +5,6 @@ from utils import platform
 from utils import queues
 
 
-class Message(pydantic.BaseModel):
-    project: str
-    name: str
-    type: str
-    body: dict
-    method: str = 'send_message'
-
-
 class Daemon(base.Daemon, queues.TasksHandlerMixin):
     def execute(self):
         self.poll()
@@ -23,19 +13,18 @@ class Daemon(base.Daemon, queues.TasksHandlerMixin):
     def queue_name(self):
         return 'botalka_mailbox'
     
-    def process(self, payload: dict):
+    def process(self, payload):
-        message = Message.model_validate(payload)
+        bot = platform.platform_client.get_config('bots')[payload['project']][payload['name']]
-        bot = platform.platform_client.get_config('bots')[message.project][message.name]
         if not bot['mailbox_enabled']:
             return
         if bot['type'] == 'telegram':
             token = bot['secrets']['telegram_token']
-            self.process_telegram(token, message.method, message.body)
+            self.process_telegram(token, payload['body'])
         else:
             print('Unknown bot type:', bot['type'])
 
-    def process_telegram(self, token, method, payload):
+    def process_telegram(self, token, payload):
         try:
-            getattr(apihelper, method)(token, **payload)
+            apihelper.send_message(token, **payload)
         except Exception as exc:
             print('Error', str(exc))

requirements.txt

@@ -1,10 +1,6 @@
-annotated-types==0.7.0
 certifi==2024.8.30
 charset-normalizer==3.4.0
 idna==3.10
-pydantic==2.10.2
-pydantic_core==2.27.1
 pyTelegramBotAPI==4.1.1
 requests==2.32.3
-typing_extensions==4.12.2
 urllib3==2.2.3