| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108 |
- """
- 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__)
|
