fz0x1/HPI
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

core: improve types for warn_if_empty

Browse source 
ok, works with this advice https://github.com/python/mypy/issues/1927 + overloads
This commit is contained in:
Dima Gerasimov 2020-05-25 01:23:30 +01:00
parent 216944b3cd
commit 248e48dc30
2 changed files with 11 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

13
my/core/common.py
View file

@ -314,14 +314,21 @@ def _warn_iterable(it, f=None):
return _warn_iterator(it, f=f)
# ok, this seems to work...
# https://github.com/python/mypy/issues/1927#issue-167100413
FL = TypeVar('FL', bound=Callable[..., List])
FI = TypeVar('FI', bound=Callable[..., Iterable])
@overload
def warn_if_empty(f: Callable[[], List[X]] ) -> Callable[[], List[X]] : ...
def warn_if_empty(f: FL) -> FL: ...
@overload
def warn_if_empty(f: Callable[[], Iterable[X]]) -> Callable[[], Iterable[X]]: ...
def warn_if_empty(f: FI) -> FI: ...
def warn_if_empty(f):
from functools import wraps
@wraps(f)
def wrapped(*args, **kwargs):
res = f(*args, **kwargs)
return _warn_iterable(res, f=f)
return wrapped
return wrapped # type: ignore