mirror of
https://github.com/civsocit/olgram.git
synced 2023-07-22 01:29:12 +03:00
55 lines
2.2 KiB
Python
55 lines
2.2 KiB
Python
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
|