diff options
Diffstat (limited to 'websocket/_exceptions.py')
| -rw-r--r-- | websocket/_exceptions.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/websocket/_exceptions.py b/websocket/_exceptions.py index 811d594..15bbdea 100644 --- a/websocket/_exceptions.py +++ b/websocket/_exceptions.py @@ -66,11 +66,12 @@ class WebSocketBadStatusException(WebSocketException): WebSocketBadStatusException will be raised when we get bad handshake status code. """ - def __init__(self, message, status_code, status_message=None, resp_headers=None): + def __init__(self, message, status_code, status_message=None, resp_headers=None, resp_body=None): msg = message % (status_code, status_message) super().__init__(msg) self.status_code = status_code self.resp_headers = resp_headers + self.resp_body = resp_body class WebSocketAddressException(WebSocketException): |