Update

tgbot/middlewares/middleware_throttling.py

@@ -0,0 +1,63 @@
+# - *- coding: utf- 8 - *-
+import time
+from typing import Any, Awaitable, Callable, Dict, Union
+
+from aiogram import BaseMiddleware
+from aiogram.dispatcher.flags import get_flag
+from aiogram.types import Message, User
+from cachetools import TTLCache
+
+
+# Антиспам
+class ThrottlingMiddleware(BaseMiddleware):
+    def __init__(self, default_rate: Union[int, float] = 1) -> None:
+        self.default_rate = default_rate
+
+        self.users = TTLCache(maxsize=10_000, ttl=600)
+
+    async def __call__(self, handler: Callable[[Message, Dict[str, Any]], Awaitable[Any]], event: Message, data):
+        this_user: User = data.get("event_from_user")
+
+        if get_flag(data, "rate") is not None:
+            self.default_rate = get_flag(data, "rate")
+
+        if self.default_rate == 0:
+            return await handler(event, data)
+
+        if this_user.id not in self.users:
+            self.users[this_user.id] = {
+                'last_throttled': int(time.time()),
+                'count_throttled': 0,
+                'now_rate': self.default_rate,
+            }
+
+            return await handler(event, data)
+        else:
+            if int(time.time()) - self.users[this_user.id]['last_throttled'] >= self.users[this_user.id]['now_rate']:
+                self.users.pop(this_user.id)
+
+                return await handler(event, data)
+            else:
+                self.users[this_user.id]['last_throttled'] = int(time.time())
+
+                if self.users[this_user.id]['count_throttled'] == 0:
+                    self.users[this_user.id]['count_throttled'] = 1
+                    self.users[this_user.id]['now_rate'] = self.default_rate + 2
+
+                    return await handler(event, data)
+                elif self.users[this_user.id]['count_throttled'] == 1:
+                    self.users[this_user.id]['count_throttled'] = 2
+                    self.users[this_user.id]['now_rate'] = self.default_rate + 3
+
+                    await event.reply(
+                        "<b>❗ Пожалуйста, не спамьте.\n"
+                        "❗ Please, do not spam.</b>",
+                    )
+                elif self.users[this_user.id]['count_throttled'] == 2:
+                    self.users[this_user.id]['count_throttled'] = 3
+                    self.users[this_user.id]['now_rate'] = self.default_rate + 5
+
+                    await event.reply(
+                        "<b>❗ Бот не будет отвечать до прекращения спама.\n"
+                        "❗ The bot will not respond until the spam stops.</b>",
+                    )