From 467e3e51d8d61e4f1c5530ee98efb8bafe520672 Mon Sep 17 00:00:00 2001 From: Anonymous Date: Wed, 8 Sep 2021 12:54:03 +0300 Subject: [PATCH] Update __reduce__ of exceptions to support pickling --- pybit/exceptions.py | 8 ++++++++ 1 file changed, 8 insertions(+) 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)