python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2025-12-11 01:54:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor dp.throttled

Browse source
This commit is contained in:
birdi 2019-08-05 11:32:28 +03:00
parent 288c7099b0
commit 4d00b7c502
1 changed files with 3 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

11
aiogram/dispatcher/dispatcher.py
View file

@ -1054,16 +1054,12 @@ class Dispatcher(DataMixin, ContextInstanceMixin):
:param chat_id: chat id
:return: decorator
"""
def decorator(func):
nonlocal on_throttled, key, rate, user_id, chat_id
if key is None:
key = func.__name__
@functools.wraps(func)
async def wrapped(*args, **kwargs):
is_not_throttled = await self.throttle(key, rate=rate,
user_id=user_id, chat_id=chat_id,
is_not_throttled = await self.throttle(key if key is not None else func.__name__,
rate=rate,
user=user_id, chat=chat_id,
no_error=True)
if is_not_throttled:
return await func(*args, **kwargs)
@ -1090,7 +1086,6 @@ class Dispatcher(DataMixin, ContextInstanceMixin):
asyncio.get_running_loop().run_in_executor(None,
partial_func
)
return wrapped
return decorator