Module arti.internal

None

None

View Source

from __future__ import annotations

from collections.abc import Iterator

from contextlib import contextmanager

@contextmanager

def wrap_exc(error_type: type[Exception], *, prefix: str) -> Iterator[None]:

    """Wrap exceptions of `error_type` and add a message prefix.

    `error_type` must be initializable with a single string message argument.

    NOTE: When used inside a generator, any exceptions raised by the *caller of the generator* will **not** be wrapped.

    """

    try:

        yield

    except error_type as e:

        msg = str(e)

        if getattr(e, "wrapped", False) and e.__cause__ is not None:

            src = e.__cause__  # Shorten exception chains to the root and last wrapped only

        else:

            msg = f" - {msg}"

            src = e

        error = error_type(f"{prefix}{msg}")

        error.wrapped = True  # type: ignore[attr-defined]

        raise error from src

Sub-modules

• arti.internal.dispatch
• arti.internal.models
• arti.internal.patches
• arti.internal.type_hints
• arti.internal.utils

Functions

wrap_exc

def wrap_exc(
    error_type: 'type[Exception]',
    *,
    prefix: 'str'
) -> 'Iterator[None]'

Wrap exceptions of error_type and add a message prefix.

error_type must be initializable with a single string message argument.

NOTE: When used inside a generator, any exceptions raised by the caller of the generator will not be wrapped.

View Source

@contextmanager

def wrap_exc(error_type: type[Exception], *, prefix: str) -> Iterator[None]:

    """Wrap exceptions of `error_type` and add a message prefix.

    `error_type` must be initializable with a single string message argument.

    NOTE: When used inside a generator, any exceptions raised by the *caller of the generator* will **not** be wrapped.

    """

    try:

        yield

    except error_type as e:

        msg = str(e)

        if getattr(e, "wrapped", False) and e.__cause__ is not None:

            src = e.__cause__  # Shorten exception chains to the root and last wrapped only

        else:

            msg = f" - {msg}"

            src = e

        error = error_type(f"{prefix}{msg}")

        error.wrapped = True  # type: ignore[attr-defined]

        raise error from src