FOSS/Bot-Template
1
0
Fork 0
mirror of https://github.com/djimboy/djimbo_template_aio3.git synced 2026-07-25 09:44:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update

Browse Source
This commit is contained in:
djimbo
2022-09-11 09:11:36 +03:00
parent 0242744eec
commit 57e02355ed
1 changed files with 23 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tgbot/middlewares/throttling.py
+23 -27
View File
@@ -20,40 +20,36 @@ class ThrottlingMiddleware(BaseMiddleware):
async def __call__(self, handler: Callable[[Message, Dict[str, Any]], Awaitable[Any]], event: Message, data): async def __call__(self, handler: Callable[[Message, Dict[str, Any]], Awaitable[Any]], event: Message, data):
real_handler: HandlerObject = data['handler'] real_handler: HandlerObject = data['handler']
skip_pass = True
this_user = data.get("event_from_user") this_user = data.get("event_from_user")
if not this_user.is_bot: if not this_user.is_bot:
data['get_user'] = get_userx(user_id=this_user.id) data['get_user'] = get_userx(user_id=this_user.id)
if real_handler.flags.get("skip_pass") is not None: if real_handler.flags.get("rate") is not None:
skip_pass = real_handler.flags.get("skip_pass") self.now_rate = real_handler.flags.get("rate")
if skip_pass: if int(time.time()) - self.last_throttled >= self.now_rate:
if int(time.time()) - self.last_throttled >= self.now_rate: self.last_throttled = int(time.time())
self.last_throttled = int(time.time()) self.now_rate = self.default_rate
self.now_rate = self.default_rate self.count_throttled = 0
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) 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: else:
if self.count_throttled == 0: self.now_rate = 3
self.count_throttled += 1
self.now_rate = self.default_rate * 2
return await handler(event, data) if self.count_throttled == 2:
elif self.count_throttled == 1: self.count_throttled = 3
self.count_throttled += 1 await event.reply("<b>❗ Бот не будет отвечать до прекращения спама.</b>")
self.now_rate = self.default_rate * 2
await event.reply("<b>❗ Пожалуйста, не спамьте.</b>") self.last_throttled = int(time.time())
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())
else:
return await handler(event, data)