2580b66de5
Makes this pass without errors: mypy --strict hwilib/base58.py hwilib/errors.py hwilib/serializations.py hwilib/hwwclient.py
127 lines
3.2 KiB
Python
127 lines
3.2 KiB
Python
# Defines errors and error codes
|
|
|
|
from typing import Any, Dict, Iterator, Optional
|
|
from contextlib import contextmanager
|
|
|
|
# Error codes
|
|
NO_DEVICE_TYPE = -1
|
|
MISSING_ARGUMENTS = -2
|
|
DEVICE_CONN_ERROR = -3
|
|
UNKNWON_DEVICE_TYPE = -4
|
|
INVALID_TX = -5
|
|
NO_PASSWORD = -6
|
|
BAD_ARGUMENT = -7
|
|
NOT_IMPLEMENTED = -8
|
|
UNAVAILABLE_ACTION = -9
|
|
DEVICE_ALREADY_INIT = -10
|
|
DEVICE_ALREADY_UNLOCKED = -11
|
|
DEVICE_NOT_READY = -12
|
|
UNKNOWN_ERROR = -13
|
|
ACTION_CANCELED = -14
|
|
DEVICE_BUSY = -15
|
|
NEED_TO_BE_ROOT = -16
|
|
HELP_TEXT = -17
|
|
DEVICE_NOT_INITIALIZED = -18
|
|
|
|
# Exceptions
|
|
class HWWError(Exception):
|
|
def __init__(self, msg: str, code: int):
|
|
Exception.__init__(self)
|
|
self.code = code
|
|
self.msg = msg
|
|
|
|
def get_code(self) -> int:
|
|
return self.code
|
|
|
|
def get_msg(self) -> str:
|
|
return self.msg
|
|
|
|
def __str__(self) -> str:
|
|
return self.msg
|
|
|
|
class NoPasswordError(HWWError):
|
|
def __init__(self, msg: str):
|
|
HWWError.__init__(self, msg, NO_PASSWORD)
|
|
|
|
class UnavailableActionError(HWWError):
|
|
def __init__(self, msg: str):
|
|
HWWError.__init__(self, msg, UNAVAILABLE_ACTION)
|
|
|
|
class DeviceAlreadyInitError(HWWError):
|
|
def __init__(self, msg: str):
|
|
HWWError.__init__(self, msg, DEVICE_ALREADY_INIT)
|
|
|
|
class DeviceNotReadyError(HWWError):
|
|
def __init__(self, msg: str):
|
|
HWWError.__init__(self, msg, DEVICE_NOT_READY)
|
|
|
|
class DeviceAlreadyUnlockedError(HWWError):
|
|
def __init__(self, msg: str):
|
|
HWWError.__init__(self, msg, DEVICE_ALREADY_UNLOCKED)
|
|
|
|
class UnknownDeviceError(HWWError):
|
|
def __init__(self, msg: str):
|
|
HWWError.__init__(self, msg, UNKNWON_DEVICE_TYPE)
|
|
|
|
class NotImplementedError(HWWError):
|
|
def __init__(self, msg: str):
|
|
HWWError.__init__(self, msg, NOT_IMPLEMENTED)
|
|
|
|
class PSBTSerializationError(HWWError):
|
|
def __init__(self, msg: str):
|
|
HWWError.__init__(self, msg, INVALID_TX)
|
|
|
|
class BadArgumentError(HWWError):
|
|
def __init__(self, msg: str):
|
|
HWWError.__init__(self, msg, BAD_ARGUMENT)
|
|
|
|
class DeviceFailureError(HWWError):
|
|
def __init__(self, msg: str):
|
|
HWWError.__init__(self, msg, UNKNOWN_ERROR)
|
|
|
|
class ActionCanceledError(HWWError):
|
|
def __init__(self, msg: str):
|
|
HWWError.__init__(self, msg, ACTION_CANCELED)
|
|
|
|
class DeviceConnectionError(HWWError):
|
|
def __init__(self, msg: str):
|
|
HWWError.__init__(self, msg, DEVICE_CONN_ERROR)
|
|
|
|
class DeviceBusyError(HWWError):
|
|
def __init__(self, msg: str):
|
|
HWWError.__init__(self, msg, DEVICE_BUSY)
|
|
|
|
@contextmanager
|
|
def handle_errors(
|
|
msg: Optional[str] = None,
|
|
result: Optional[Dict[str, Any]] = None,
|
|
code: int = UNKNOWN_ERROR,
|
|
debug: bool = False,
|
|
) -> Iterator[None]:
|
|
if result is None:
|
|
result = {}
|
|
|
|
if msg is None:
|
|
msg = ""
|
|
else:
|
|
msg = msg + " "
|
|
|
|
try:
|
|
yield
|
|
|
|
except HWWError as e:
|
|
result['error'] = msg + e.get_msg()
|
|
result['code'] = e.get_code()
|
|
except Exception as e:
|
|
result['error'] = msg + str(e)
|
|
result['code'] = code
|
|
if debug:
|
|
import traceback
|
|
traceback.print_exc()
|
|
return result
|
|
|
|
|
|
common_err_msgs = {
|
|
"enumerate": "Could not open client or get fingerprint information:"
|
|
}
|