test

server/custom.py

@@ -0,0 +1,54 @@
+from aiogram import Bot as AioBot, Dispatcher
+from aiogram.dispatcher.webhook import WebhookRequestHandler
+from aiogram.dispatcher.webhook import SendMessage
+from aiogram import types
+from olgram.models.models import Bot
+
+
+async def message_handler(message, *args, **kwargs):
+    if message.text and message.text.startswith("/start"):
+        # На команду start нужно ответить, не пересылая сообщение никуда
+        return SendMessage(chat_id=message.chat.id, text=f'Hi from webhook {args} {kwargs}')
+
+
+class CustomRequestHandler(WebhookRequestHandler):
+
+    def __init__(self, *args, **kwargs):
+        self._dispatcher = None
+        super(CustomRequestHandler, self).__init__(*args, **kwargs)
+
+    async def _create_dispatcher(self):
+        key = self.request.url.path[1:]
+
+        bot = await Bot.filter(code=key).first()
+        if not bot:
+            return None
+
+        dp = Dispatcher(AioBot(bot.token))
+
+        dp.register_message_handler(message_handler, content_types=[types.ContentType.TEXT,
+                                                                    types.ContentType.CONTACT,
+                                                                    types.ContentType.ANIMATION,
+                                                                    types.ContentType.AUDIO,
+                                                                    types.ContentType.DOCUMENT,
+                                                                    types.ContentType.PHOTO,
+                                                                    types.ContentType.STICKER,
+                                                                    types.ContentType.VIDEO,
+                                                                    types.ContentType.VOICE])
+
+        return dp
+
+    async def post(self):
+        # TODO: refactor
+        self._dispatcher = await self._create_dispatcher()
+        res = await super(CustomRequestHandler, self).post()
+        self._dispatcher = None
+        return res
+
+    def get_dispatcher(self):
+        """
+        Get Dispatcher instance from environment
+
+        :return: :class:`aiogram.Dispatcher`
+        """
+        return self._dispatcher