forked from goodboy/tractor
Add type annots to exceptions mod
Also add a `is_multi_cancelled()` predicate to test for `trio.MultiError`s that contain entirely cancel signals. Resolves #125denoise_logging
parent
f4f39c29f3
commit
8522f90000
|
@ -1,6 +1,7 @@
|
||||||
"""
|
"""
|
||||||
Our classy exception set.
|
Our classy exception set.
|
||||||
"""
|
"""
|
||||||
|
from typing import Dict, Any
|
||||||
import importlib
|
import importlib
|
||||||
import builtins
|
import builtins
|
||||||
import traceback
|
import traceback
|
||||||
|
@ -14,7 +15,7 @@ _this_mod = importlib.import_module(__name__)
|
||||||
class RemoteActorError(Exception):
|
class RemoteActorError(Exception):
|
||||||
# TODO: local recontruction of remote exception deats
|
# TODO: local recontruction of remote exception deats
|
||||||
"Remote actor exception bundled locally"
|
"Remote actor exception bundled locally"
|
||||||
def __init__(self, message, type_str, **msgdata):
|
def __init__(self, message, type_str, **msgdata) -> None:
|
||||||
super().__init__(message)
|
super().__init__(message)
|
||||||
for ns in [builtins, _this_mod, trio]:
|
for ns in [builtins, _this_mod, trio]:
|
||||||
try:
|
try:
|
||||||
|
@ -45,7 +46,7 @@ class ModuleNotExposed(ModuleNotFoundError):
|
||||||
"The requested module is not exposed for RPC"
|
"The requested module is not exposed for RPC"
|
||||||
|
|
||||||
|
|
||||||
def pack_error(exc):
|
def pack_error(exc: BaseException) -> Dict[str, Any]:
|
||||||
"""Create an "error message" for tranmission over
|
"""Create an "error message" for tranmission over
|
||||||
a channel (aka the wire).
|
a channel (aka the wire).
|
||||||
"""
|
"""
|
||||||
|
@ -57,7 +58,11 @@ def pack_error(exc):
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
def unpack_error(msg, chan=None, err_type=RemoteActorError):
|
def unpack_error(
|
||||||
|
msg: Dict[str, Any],
|
||||||
|
chan=None,
|
||||||
|
err_type=RemoteActorError
|
||||||
|
) -> Exception:
|
||||||
"""Unpack an 'error' message from the wire
|
"""Unpack an 'error' message from the wire
|
||||||
into a local ``RemoteActorError``.
|
into a local ``RemoteActorError``.
|
||||||
"""
|
"""
|
||||||
|
@ -66,3 +71,15 @@ def unpack_error(msg, chan=None, err_type=RemoteActorError):
|
||||||
f"{chan.uid}\n" + tb_str,
|
f"{chan.uid}\n" + tb_str,
|
||||||
**msg['error'],
|
**msg['error'],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def is_multi_cancelled(exc: BaseException) -> bool:
|
||||||
|
"""Predicate to determine if a ``trio.MultiError`` contains only
|
||||||
|
``trio.Cancelled`` sub-exceptions (and is likely the result of
|
||||||
|
cancelling a collection of subtasks.
|
||||||
|
|
||||||
|
"""
|
||||||
|
return not trio.MultiError.filter(
|
||||||
|
lambda exc: exc if not isinstance(exc, trio.Cancelled) else None,
|
||||||
|
exc,
|
||||||
|
)
|
||||||
|
|
Loading…
Reference in New Issue