Template
mirror of
https://github.com/djimboy/djimbo_template_aio3.git
synced 2026-08-24 06:07:42 +00:00
djimbo
56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
# - *- coding: utf- 8 - *-
|
|
import time
|
|
from typing import Any, Awaitable, Callable, Dict
|
|
|
|
from aiogram import BaseMiddleware
|
|
from aiogram.dispatcher.event.handler import HandlerObject
|
|
from aiogram.types import Message
|
|
|
|
from tgbot.services.api_sqlite import get_userx
|
|
|
|
|
|
# Антиспам
|
|
class ThrottlingMiddleware(BaseMiddleware):
|
|
def __init__(self, default_rate: int = 0.6) -> None:
|
|
self.default_rate = default_rate
|
|
self.now_rate = default_rate
|
|
|
|
self.last_throttled = int(time.time())
|
|
self.count_throttled = 0
|
|
|
|
async def __call__(self, handler: Callable[[Message, Dict[str, Any]], Awaitable[Any]], event: Message, data):
|
|
real_handler: HandlerObject = data['handler']
|
|
this_user = data.get("event_from_user")
|
|
|
|
if not this_user.is_bot:
|
|
data['get_user'] = get_userx(user_id=this_user.id)
|
|
|
|
if real_handler.flags.get("rate") is not None:
|
|
self.now_rate = real_handler.flags.get("rate")
|
|
|
|
if int(time.time()) - self.last_throttled >= self.now_rate:
|
|
self.last_throttled = int(time.time())
|
|
self.now_rate = self.default_rate
|
|
self.count_throttled = 0
|
|
|
|
return await handler(event, data)
|
|
else:
|
|
if self.count_throttled == 0:
|
|
self.count_throttled += 1
|
|
self.now_rate = self.default_rate * 2
|
|
|
|
return await handler(event, data)
|
|
elif self.count_throttled == 1:
|
|
self.count_throttled += 1
|
|
self.now_rate = self.default_rate * 2
|
|
|
|
await event.reply("<b>❗ Пожалуйста, не спамьте.</b>")
|
|
else:
|
|
self.now_rate = 3
|
|
|
|
if self.count_throttled == 2:
|
|
self.count_throttled = 3
|
|
await event.reply("<b>❗ Бот не будет отвечать до прекращения спама.</b>")
|
|
|
|
self.last_throttled = int(time.time())
|