python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2025-12-17 04:25:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Refactor filter checker.

Browse source
This commit is contained in:
Alex Root Junior 2017-08-23 23:27:03 +03:00
parent 62417da6a7
commit e9a3ca6c8e
1 changed files with 4 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

13
aiogram/dispatcher/filters.py
View file

@ -5,18 +5,13 @@ from ..utils.helper import Helper, HelperMode, Item
async def check_filter(filter_, args, kwargs):
# TODO: Refactor that shit.
if not callable(filter_):
raise TypeError('Filter must be callable and/or awaitable!')
if any((inspect.isasyncgen(filter_),
inspect.iscoroutine(filter_),
inspect.isawaitable(filter_),
inspect.isasyncgenfunction(filter_),
inspect.iscoroutinefunction(filter_))):
if inspect.isawaitable(filter_) or inspect.iscoroutinefunction(filter_):
return await filter_(*args, **kwargs)
elif callable(filter_):
return filter_(*args, **kwargs)
else:
return True
return filter_(*args, **kwargs)
async def check_filters(filters, args, kwargs):