From 70becfc32ec98084067686bd456971a4cb76082c Mon Sep 17 00:00:00 2001 From: Igor Tolmachov Date: Sat, 3 Dec 2022 02:03:05 +0900 Subject: Beta 2.0 --- handlers/middleware.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 handlers/middleware.py (limited to 'handlers/middleware.py') diff --git a/handlers/middleware.py b/handlers/middleware.py new file mode 100644 index 0000000..6cf39b6 --- /dev/null +++ b/handlers/middleware.py @@ -0,0 +1,21 @@ +from logging import info + +from aiogram import types as t +from aiogram.dispatcher.middlewares import BaseMiddleware + +from shared.database import Message +from shared.instances import session + + +class MessageMiddleware(BaseMiddleware): + async def on_pre_process_message(self, msg: t.Message, data: dict) -> None: + text = msg.text or msg.caption + if text is not None and not text.startswith("/"): + with session.begin() as s: + s.add( + Message( + chat_id=msg.chat.id, + message_id=msg.message_id, + message=text, + ) + ) -- cgit v1.2.3