FOSS/Bot-Template
Template
1
0
Fork 0
mirror of https://github.com/djimboy/djimbo_template_aio3.git synced 2026-08-26 15:07:43 +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
+3 -7
View File
@@ -20,16 +20,14 @@ 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
@@ -55,5 +53,3 @@ class ThrottlingMiddleware(BaseMiddleware):
await event.reply("<b>❗ Бот не будет отвечать до прекращения спама.</b>") await event.reply("<b>❗ Бот не будет отвечать до прекращения спама.</b>")
self.last_throttled = int(time.time()) self.last_throttled = int(time.time())
else:
return await handler(event, data)