Fix exception typing

transport_hardening
Tyler Goodlet 2021-06-13 20:19:52 -04:00
parent a146034cb7
commit 3b48f89056
1 changed files with 3 additions and 3 deletions

View File

@ -1,7 +1,7 @@
""" """
Our classy exception set. Our classy exception set.
""" """
from typing import Dict, Any, Optional from typing import Dict, Any, Optional, Type
import importlib import importlib
import builtins import builtins
import traceback import traceback
@ -18,7 +18,7 @@ class RemoteActorError(Exception):
def __init__( def __init__(
self, self,
message: str, message: str,
suberror_type: Optional[Exception] = None, suberror_type: Optional[Type[BaseException]] = None,
**msgdata **msgdata
) -> None: ) -> None:
@ -85,7 +85,7 @@ def unpack_error(
tb_str = error.get('tb_str', '') tb_str = error.get('tb_str', '')
message = f"{chan.uid}\n" + tb_str message = f"{chan.uid}\n" + tb_str
type_name = error['type_str'] type_name = error['type_str']
suberror_type = Exception suberror_type: Type[BaseException] = Exception
if type_name == 'ContextCancelled': if type_name == 'ContextCancelled':
err_type = ContextCancelled err_type = ContextCancelled