mirror of
https://github.com/Garmelon/PFERD.git
synced 2023-12-21 10:23:01 +01:00
40 lines
935 B
Python
40 lines
935 B
Python
"""
|
|
An error logging decorator.
|
|
"""
|
|
|
|
import logging
|
|
from typing import Any, Callable, TypeVar, cast
|
|
|
|
from rich.console import Console
|
|
|
|
from .logging import PrettyLogger
|
|
|
|
LOGGER = logging.getLogger(__name__)
|
|
PRETTY = PrettyLogger(LOGGER)
|
|
|
|
|
|
class FatalException(Exception):
|
|
"""
|
|
A fatal exception occurred. Recovery is not possible.
|
|
"""
|
|
|
|
|
|
TFun = TypeVar('TFun', bound=Callable[..., Any])
|
|
|
|
|
|
def swallow_and_print_errors(function: TFun) -> TFun:
|
|
"""
|
|
Decorates a function, swallows all errors, logs them and returns none if one occurred.
|
|
"""
|
|
def inner(*args: Any, **kwargs: Any) -> Any:
|
|
# pylint: disable=broad-except
|
|
try:
|
|
return function(*args, **kwargs)
|
|
except FatalException as error:
|
|
PRETTY.error(str(error))
|
|
return None
|
|
except Exception as error:
|
|
Console().print_exception()
|
|
return None
|
|
return cast(TFun, inner)
|