21 lines
652 B
Python
21 lines
652 B
Python
from wsproto.frame_protocol import CloseReason
|
|
|
|
|
|
class SimpleWebsocketError(RuntimeError):
|
|
pass
|
|
|
|
|
|
class ConnectionError(SimpleWebsocketError):
|
|
"""Connection error exception class."""
|
|
def __init__(self, status_code=None): # pragma: no cover
|
|
self.status_code = status_code
|
|
super().__init__(f'Connection error: {status_code}')
|
|
|
|
|
|
class ConnectionClosed(SimpleWebsocketError):
|
|
"""Connection closed exception class."""
|
|
def __init__(self, reason=CloseReason.NO_STATUS_RCVD, message=None):
|
|
self.reason = reason
|
|
self.message = message
|
|
super().__init__(f'Connection closed: {reason} {message or ""}')
|