LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-07-27 08:19:47 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix async filter call in combinators

Browse Source
This commit is contained in:
Jahongir Qurbonov 2024-08-19 12:21:50 +05:00
parent eb7ed5dd31
commit 6626fbcd0c
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
client/src/telethon/_impl/client/events/filters/combinators.py
View File

@ -1,6 +1,7 @@
import abc
import typing
from collections.abc import Callable
from inspect import isawaitable
from typing import Awaitable, TypeAlias
from ..event import Event
@ -41,7 +42,7 @@ class Combinable(abc.ABC):
return self.filter if isinstance(self, Not) else Not(self) # type: ignore [return-value]
@abc.abstractmethod
def __call__(self, event: Event) -> bool:
async def __call__(self, event: Event) -> bool:
pass
@ -81,8 +82,10 @@ class Any(Combinable):
"""
return self._filters
def __call__(self, event: Event) -> bool:
return any(f(event) for f in self._filters)
async def __call__(self, event: Event) -> bool:
return any(
[await r if isawaitable(r := f(event)) else r for f in self._filters]
)
class All(Combinable):
@ -121,8 +124,10 @@ class All(Combinable):
"""
return self._filters
def __call__(self, event: Event) -> bool:
return all(f(event) for f in self._filters)
async def __call__(self, event: Event) -> bool:
return all(
[await r if isawaitable(r := f(event)) else r for f in self._filters]
)
class Not(Combinable):
@ -159,5 +164,5 @@ class Not(Combinable):
"""
return self._filter
def __call__(self, event: Event) -> bool:
return not self._filter(event)
async def __call__(self, event: Event) -> bool:
return not (await r if isawaitable(r := self._filter(event)) else r)