Function _attach_event_handler

core/src/stdlib/pyscript/websocket.py:47–67

Given a `websocket`, and `handler_name`, attach the `handler_function` to the `WebSocket` instance, handling both synchronous and asynchronous handler functions. Creates a JavaScript proxy for the handler and wraps async handlers appropriately. Handles the `WebSocketEvent` wrap

(websocket, handler_name, handler_function)

def _attach_event_handler(websocket, handler_name, handler_function):
    """
    Given a `websocket`, and `handler_name`, attach the `handler_function`
    to the `WebSocket` instance, handling both synchronous and asynchronous
    handler functions.

    Creates a JavaScript proxy for the handler and wraps async handlers
    appropriately. Handles the `WebSocketEvent` wrapping for all handlers.
    """
    if is_awaitable(handler_function):

        async def async_wrapper(event):
            await handler_function(WebSocketEvent(event))

        wrapped_handler = create_proxy(async_wrapper)
    else:
        wrapped_handler = create_proxy(
            lambda event: handler_function(WebSocketEvent(event))
        )
    # Note: Direct assignment (websocket[handler_name]) fails in Pyodide.
    setattr(websocket, handler_name, wrapped_handler)


class WebSocketEvent: