1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44
|
from typing import Any, Callable, Optional, Type, TypeVar
T = TypeVar("T")
def ignore_exceptions_optional(
_tp: Type[T], default: Optional[T], *exceptions: Type[BaseException]
) -> Callable[[Callable[..., Optional[T]]], Callable[..., Optional[T]]]:
"""
Decorator, that wraps around a function preventing it from raising exceptions
and instead returning the configured default value.
:param Type[T] _tp: Return type of the function. Essentialy only a template argument for type-checking
:param T default: The value to return as a default
:param List[Type[BaseException]] exceptions: The list of exceptions to catch
:return: value of the decorated function, or default if exception raised
:rtype: T
"""
def decorator(func: Callable[..., Optional[T]]) -> Callable[..., Optional[T]]:
def f(*nargs: Any, **nkwargs: Any) -> Optional[T]:
try:
return func(*nargs, **nkwargs)
except BaseException as e:
if isinstance(e, exceptions):
return default
raise e
return f
return decorator
def ignore_exceptions(
default: T, *exceptions: Type[BaseException]
) -> Callable[[Callable[..., Optional[T]]], Callable[..., Optional[T]]]:
return ignore_exceptions_optional(type(default), default, *exceptions)
def phantom_use(var: Any) -> None: # pylint: disable=unused-argument
"""
Function, which consumes its argument doing absolutely nothing with it. Useful
for convincing pylint, that we need the variable even when its unused.
"""
|