Function handle_api_exception

beanhub_cli/api_helpers.py:8–33

(logger: logging.Logger | None = None)

def handle_api_exception(logger: logging.Logger | None = None):
    def decorator(func: typing.Callable):
        @functools.wraps(func)
        def callee(*args, **kwargs):
            from .internal_api.errors import UnexpectedStatus
            from .internal_api.models import GenericError

            try:
                return func(*args, **kwargs)
            except UnexpectedStatus as exp:
                log = logger or logging.getLogger(__name__)
                error = None
                try:
                    error = GenericError.from_dict(json.loads(exp.content))
                except ValueError:
                    pass
                log.error(
                    "Failed to make BeanHub API call with code %s and error: %s",
                    exp.status_code,
                    error.detail if error is not None else exp.content,
                )
                sys.exit(-1)

        return callee

    return decorator