from __future__ import annotations
import typing
if typing.TYPE_CHECKING:
import wsproto
from .._models import Response
class HTTPXWSException(Exception):
class WebSocketUpgradeError(HTTPXWSException):
def __init__(self, response: Response) -> None:
self.response = response
class WebSocketDisconnect(HTTPXWSException):
def __init__(self, code: int = 1000, reason: str | None = None) -> None:
self.code = code
self.reason = reason or ""
class WebSocketInvalidTypeReceived(HTTPXWSException):
def __init__(self, event: wsproto.events.Event) -> None:
self.event = event
class WebSocketNetworkError(HTTPXWSException):