diff --git a/pybit/exceptions.py b/pybit/exceptions.py index abcb94c..486aeec 100644 --- a/pybit/exceptions.py +++ b/pybit/exceptions.py @@ -18,6 +18,10 @@ def __init__(self, request, message, status_code, time): f'.\nRequest → {request}.' ) + def __reduce__(self): + return self.__class__, (self.request, self.message, self.status_code, + self.time) + class InvalidRequestError(Exception): """ @@ -38,3 +42,7 @@ def __init__(self, request, message, status_code, time): f'{message.capitalize()} (ErrCode: {status_code}) (ErrTime: {time})' f'.\nRequest → {request}.' ) + + def __reduce__(self): + return self.__class__, (self.request, self.message, self.status_code, + self.time)