WebSocket class

Importing the WebSocket from ravyn is as simple as:

from ravyn import WebSocket

ravyn.WebSocket

WebSocket(scope, receive, send)

Bases: WebsocketMixin

Source code in lilya/websockets.py

def __init__(self, scope: Scope, receive: Receive, send: Send) -> None:
    super().__init__(scope)
    assert scope["type"] == ScopeType.WEBSOCKET
    self._receive = receive
    self._send = send
    self._accepted = False
    self.client_state = WebSocketState.CONNECTING
    self.application_state = WebSocketState.CONNECTING
    self._cleanup_callbacks: list[Callable[[], Any]] = []

scope instance-attribute

scope = scope

app property

app

url property

url

base_url property

base_url

headers property

headers

state property

state

query_params property

query_params

path_params property

path_params

cookies property

cookies

This addresses multiple cookies and duplicates as per RFC 7540. https://datatracker.ietf.org/doc/html/rfc7540

client property

client

server property

server

auth property

auth

user property

user

session property

session

is_server_push property

is_server_push

is_server_pull property

is_server_pull

client_state instance-attribute

client_state = CONNECTING

application_state instance-attribute

application_state = CONNECTING

set_session

set_session(value)

Sets the value of a request session by passing a dictionary.

Source code in lilya/_internal/_connection.py

def set_session(self, value: Any) -> None:
    """
    Sets the value of a request session by passing a dictionary.
    """
    self.scope["session"] = value

clear_session

clear_session()

Clears the scope session.

Source code in lilya/_internal/_connection.py

def clear_session(self) -> None:
    """
    Clears the scope session.
    """
    self.scope["session"] = None

is_secure

is_secure()

Check if the connection is secure (HTTPS).

RETURNS	DESCRIPTION
bool	True if the connection is secure (HTTPS), False otherwise. TYPE: bool

Source code in lilya/_internal/_connection.py

def is_secure(self) -> bool:
    """
    Check if the connection is secure (HTTPS).

    Returns:
        bool: True if the connection is secure (HTTPS), False otherwise.
    """
    return self.url.is_secure

path_for

path_for(name, /, **path_params)

Source code in lilya/_internal/_connection.py

def path_for(self, name: str, /, **path_params: Any) -> URL:
    return self.url_path_for(name, **path_params)

url_path_for

url_path_for(name, /, **path_params)

Source code in lilya/_internal/_connection.py

def url_path_for(self, name: str, /, **path_params: Any) -> URL:
    router: Router | Lilya | None = self.scope.get("router") or self.scope.get("app")

    if router is None:
        raise RuntimeError(
            "The `url_for` method can only be used inside a Lilya application or with a router."
        )

    url_path = router.url_path_for(name, **path_params)
    return cast(URL, url_path.make_absolute_url(base_url=self.base_url))

raise_for_disconnect

raise_for_disconnect(message)

Source code in lilya/websockets.py

def raise_for_disconnect(self, message: Message) -> None:
    if message["type"] == Event.WEBSOCKET_DISCONNECT:
        raise WebSocketDisconnect(message["code"], message.get("reason"))

raise_for_connection_state async

raise_for_connection_state()

Source code in lilya/websockets.py

async def raise_for_connection_state(self) -> None:
    if self.application_state != WebSocketState.CONNECTED:
        raise WebSocketRuntimeError('WebSocket is not connected. Need to call "accept" first.')

receive async

receive()

Receive ASGI websocket messages, ensuring valid state transitions.

Source code in lilya/websockets.py

async def receive(self) -> Message:
    """
    Receive ASGI websocket messages, ensuring valid state transitions.
    """
    if self.client_state not in {WebSocketState.CONNECTING, WebSocketState.CONNECTED}:
        raise WebSocketRuntimeError(
            'Cannot call "receive" once a disconnect message has been received.'
        )

    message = await self._receive()
    message_type = message["type"]

    if self.client_state == WebSocketState.CONNECTING:
        if message_type != Event.WEBSOCKET_CONNECT:
            raise WebSocketRuntimeError(
                f'Expected ASGI message "websocket.connect", but got {message_type!r}'
            )
        self.client_state = WebSocketState.CONNECTED
    elif self.client_state == WebSocketState.CONNECTED:
        if message_type not in {Event.WEBSOCKET_RECEIVE, Event.WEBSOCKET_DISCONNECT}:
            raise WebSocketRuntimeError(
                "Expected ASGI message websocket.receive or "
                f'"websocket.disconnect", but got {message_type!r}'
            )

        self.client_state = (
            WebSocketState.DISCONNECTED
            if message_type == Event.WEBSOCKET_DISCONNECT
            else self.client_state
        )
    return message

add_cleanup

add_cleanup(fn)

Source code in lilya/websockets.py

def add_cleanup(self, fn: Callable[[], Any]) -> None:
    self._cleanup_callbacks.append(fn)

send async

send(message)

Send ASGI websocket messages, ensuring valid state transitions.

Source code in lilya/websockets.py

async def send(self, message: Message) -> None:
    """
    Send ASGI websocket messages, ensuring valid state transitions.
    """
    if self.application_state not in {WebSocketState.CONNECTING, WebSocketState.CONNECTED}:
        raise WebSocketRuntimeError('Cannot call "send" once a close message has been sent.')

    message_type = message["type"]

    if self.application_state == WebSocketState.CONNECTING:
        if message_type not in {Event.WEBSOCKET_ACCEPT, Event.WEBSOCKET_CLOSE}:
            raise WebSocketRuntimeError(
                'Expected ASGI message "websocket.accept" or '
                f'"websocket.close", but got {message_type!r}'
            )
        self.application_state = (
            WebSocketState.DISCONNECTED
            if message_type == Event.WEBSOCKET_CLOSE
            else WebSocketState.CONNECTED
        )
    elif self.application_state == WebSocketState.CONNECTED:
        if message_type not in {Event.WEBSOCKET_SEND, Event.WEBSOCKET_CLOSE}:
            raise WebSocketRuntimeError(
                'Expected ASGI message "websocket.send" or "websocket.close", '
                f"but got {message_type!r}"
            )

        self.application_state = (
            WebSocketState.DISCONNECTED
            if message_type == Event.WEBSOCKET_CLOSE
            else self.application_state
        )
    await self._send(message)

accept async

accept(subprotocol=None, headers=None)

Source code in lilya/websockets.py

async def accept(
    self,
    subprotocol: str | None = None,
    headers: Iterable[tuple[bytes, bytes]] | None = None,
) -> None:
    headers = headers or []

    await self._connect()
    message = {
        "type": "websocket.accept",
        "headers": headers,
        "subprotocol": subprotocol,
    }
    await self.send(message)

receive_text async

receive_text()

Source code in lilya/websockets.py

async def receive_text(self) -> str:
    await self.raise_for_connection_state()

    message = await self.receive()
    self.raise_for_disconnect(message)
    return cast(str, message["text"])

receive_bytes async

receive_bytes()

Source code in lilya/websockets.py

async def receive_bytes(self) -> bytes:
    await self.raise_for_connection_state()

    message = await self.receive()
    self.raise_for_disconnect(message)
    return cast(bytes, message["bytes"])

receive_json async

receive_json(mode='text')

Source code in lilya/websockets.py

async def receive_json(self, mode: str = "text") -> Any:
    if mode not in {MessageMode.TEXT, MessageMode.BINARY}:
        raise WebSocketRuntimeError('The "mode" argument should be "text" or "binary".')

    await self.raise_for_connection_state()

    message = await self.receive()
    self.raise_for_disconnect(message)

    if mode == "text":
        text = message["text"]
    else:
        text = message["bytes"].decode("utf-8")
    return serializer.loads(text)

iter_text async

iter_text()

Source code in lilya/websockets.py

async def iter_text(self) -> AsyncIterator[str]:
    try:
        while True:
            yield await self.receive_text()
    except WebSocketDisconnect:
        ...  # pragma: no cover

iter_bytes async

iter_bytes()

Source code in lilya/websockets.py

async def iter_bytes(self) -> AsyncIterator[bytes]:
    try:
        while True:
            yield await self.receive_bytes()
    except WebSocketDisconnect:
        ...  # pragma: no cover

iter_json async

iter_json()

Source code in lilya/websockets.py

async def iter_json(self) -> AsyncIterator[Any]:
    try:
        while True:
            yield await self.receive_json()
    except WebSocketDisconnect:
        ...  # pragma: no cover

send_text async

send_text(data)

Source code in lilya/websockets.py

async def send_text(self, data: str) -> None:
    await self.send({"type": Event.WEBSOCKET_SEND, "text": data})

send_bytes async

send_bytes(data)

Source code in lilya/websockets.py

async def send_bytes(self, data: bytes) -> None:
    await self.send({"type": Event.WEBSOCKET_SEND, "bytes": data})

send_json async

send_json(data, mode='text')

Source code in lilya/websockets.py

async def send_json(self, data: Any, mode: str = "text") -> None:
    if mode not in {MessageMode.TEXT, MessageMode.BINARY}:
        raise WebSocketRuntimeError('The "mode" argument should be "text" or "binary".')

    text = serializer.dumps(data, separators=(",", ":"), ensure_ascii=False)

    (
        await self.send({"type": Event.WEBSOCKET_SEND, "text": text})
        if mode == MessageMode.TEXT
        else await self.send({"type": Event.WEBSOCKET_SEND, "bytes": text.encode("utf-8")})
    )

close async

close(code=1000, reason=None)

Source code in lilya/websockets.py

async def close(self, code: int = 1000, reason: str | None = None) -> None:
    await self.send({"type": Event.WEBSOCKET_CLOSE, "code": code, "reason": reason or ""})

    for fn in self._cleanup_callbacks:
        maybe_await = fn()
        if inspect.isawaitable(maybe_await):
            await maybe_await

ravyn.WebSocketDisconnect

WebSocketDisconnect(code=1000, reason=None)

Bases: WebSocketDisconnect

Ravyn WebSocketDisconnect

Source code in ravyn/websockets.py

def __init__(self, code: int = 1000, reason: Optional[str] = None) -> None:
    super().__init__(code, reason)  # pragma: no cover

code instance-attribute

code = code

reason instance-attribute

reason = reason or ''

ravyn.WebSocketState

Bases: IntEnum

CONNECTING class-attribute instance-attribute

CONNECTING = 0

CONNECTED class-attribute instance-attribute

CONNECTED = 1

DISCONNECTED class-attribute instance-attribute

DISCONNECTED = 2