def exit_on_entrypoint_error()

in chz/blueprint/_entrypoint.py [0:0]

• 15 lines of code
• 6 McCabe index (conditional complexity)

def exit_on_entrypoint_error(fn: _F) -> _F:
    @functools.wraps(fn)
    def inner(*args, **kwargs):
        try:
            return fn(*args, **kwargs)
        except EntrypointException as e:
            if isinstance(e, EntrypointHelpException):
                print(e, end="" if e.args[0][-1] == "\n" else "\n")
            else:
                print("Error:", file=sys.stderr)
                print(e, end="" if e.args[0][-1] == "\n" else "\n", file=sys.stderr)
            if "PYTEST_VERSION" in os.environ:
                raise
            sys.exit(1)

    return inner  # type: ignore[return-value]