Beta 2.0

1 files changed, 21 insertions, 0 deletions

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,
+                    )
+                )