OwlCyberSecurity - MANAGER

Edit File: _result.py

"""
Hook wrapper "result" utilities.
"""
from __future__ import annotations

from types import TracebackType
from typing import Callable
from typing import cast
from typing import Generator
from typing import Generic
from typing import Optional
from typing import Tuple
from typing import Type
from typing import TYPE_CHECKING
from typing import TypeVar

if TYPE_CHECKING:
    from typing import NoReturn

_ExcInfo = Tuple[Type[BaseException], BaseException, Optional[TracebackType]]
_T = TypeVar("_T")

def _raise_wrapfail(
    wrap_controller: (
        Generator[None, _Result[_T], None] | Generator[None, object, object]
    ),
    msg: str,
) -> NoReturn:
    co = wrap_controller.gi_code
    raise RuntimeError(
        "wrap_controller at %r %s:%d %s"
        % (co.co_name, co.co_filename, co.co_firstlineno, msg)
    )

class HookCallError(Exception):
    """Hook was called incorrectly."""

class _Result(Generic[_T]):
    __slots__ = ("_result", "_exception")

def __init__(
        self,
        result: _T | None,
        exception: BaseException | None,
    ) -> None:
        self._result = result
        self._exception = exception

@property
    def excinfo(self) -> _ExcInfo | None:
        exc = self._exception
        if exc is None:
            return None
        else:
            return (type(exc), exc, exc.__traceback__)

@property
    def exception(self) -> BaseException | None:
        return self._exception

@classmethod
    def from_call(cls, func: Callable[[], _T]) -> _Result[_T]:
        __tracebackhide__ = True
        result = exception = None
        try:
            result = func()
        except BaseException as exc:
            exception = exc
        return cls(result, exception)

def force_result(self, result: _T) -> None:
        """Force the result(s) to ``result``.

If the hook was marked as a ``firstresult`` a single value should
        be set, otherwise set a (modified) list of results. Any exceptions
        found during invocation will be deleted.

This overrides any previous result or exception.
        """
        self._result = result
        self._exception = None

def force_exception(self, exception: BaseException) -> None:
        """Force the result to fail with ``exception``.

This overrides any previous result or exception.

.. versionadded:: 1.1.0
        """
        self._result = None
        self._exception = exception

def get_result(self) -> _T:
        """Get the result(s) for this hook call.

If the hook was marked as a ``firstresult`` only a single value
        will be returned, otherwise a list of results.
        """
        __tracebackhide__ = True
        exc = self._exception
        if exc is None:
            return cast(_T, self._result)
        else:
            raise exc.with_traceback(exc.__traceback__)