Allow passing error type to `unpack_error()`

contexts
Tyler Goodlet 2019-01-12 13:26:02 -05:00
parent 3092dc6df6
commit a9932e6c01
1 changed files with 2 additions and 2 deletions

View File

@ -55,12 +55,12 @@ def pack_error(exc):
} }
def unpack_error(msg, chan=None): def unpack_error(msg, chan=None, err_type=RemoteActorError):
"""Unpack an 'error' message from the wire """Unpack an 'error' message from the wire
into a local ``RemoteActorError``. into a local ``RemoteActorError``.
""" """
tb_str = msg['error'].get('tb_str', '') tb_str = msg['error'].get('tb_str', '')
return RemoteActorError( return err_type(
f"{chan.uid}\n" + tb_str, f"{chan.uid}\n" + tb_str,
**msg['error'], **msg['error'],
) )