Какие-то правки кода

Уже не помню даже, почему

Author: MasterGroosha

code/05_filters_and_middlewares/bot.py

@@ -4,6 +4,7 @@
 
 from config_reader import config
 from handlers import group_games, checkin, usernames
+from middlewares.slow import ChatActionMiddleware
 from middlewares.weekend import WeekendCallbackMiddleware
 
 
@@ -15,6 +16,7 @@ async def main():
     dp.include_router(checkin.router)
     dp.include_router(usernames.router)
 
+    dp.message.middleware(ChatActionMiddleware())
     dp.callback_query.outer_middleware(WeekendCallbackMiddleware())
 
     # Запускаем бота и пропускаем все накопленные входящие

code/05_filters_and_middlewares/handlers/checkin.py

@@ -1,3 +1,5 @@
+import asyncio
+
 from aiogram import Router, F
 from aiogram.filters import Command
 from aiogram.types import Message, CallbackQuery
@@ -10,8 +12,9 @@
 router.message.middleware(WeekendMessageMiddleware())
 
 
-@router.message(Command("checkin"))
+@router.message(Command("checkin"), flags={"long_operation": "upload_video_note"})
 async def cmd_checkin(message: Message):
+    await asyncio.sleep(5)
     await message.answer(
         "Пожалуйста, нажмите на кнопку ниже:",
         reply_markup=get_checkin_kb()

code/05_filters_and_middlewares/middlewares/slow.py

@@ -0,0 +1,29 @@
+from statistics import median
+from typing import Callable, Dict, Any, Awaitable
+
+from aiogram import BaseMiddleware
+from aiogram.dispatcher.flags import get_flag
+from aiogram.types import Message
+from aiogram.utils.chat_action import ChatActionSender
+
+
+class ChatActionMiddleware(BaseMiddleware):
+    async def __call__(
+        self,
+        handler: Callable[[Message, Dict[str, Any]], Awaitable[Any]],
+        event: Message,
+        data: Dict[str, Any],
+    ) -> Any:
+        long_operation_type = get_flag(data, "long_operation")
+
+        # Если такого флага на хэндлере нет
+        if not long_operation_type:
+            return await handler(event, data)
+
+        # Если флаг есть
+        async with ChatActionSender(
+                action=long_operation_type,
+                chat_id=event.chat.id,
+                bot=data["bot"],
+        ):
+            return await handler(event, data)