109 lines
2.8 KiB
Python
109 lines
2.8 KiB
Python
|
"""
|
||
|
|
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__)
|