Function validate_registry

zappa/websocket.py:172–193

Validate that all required routes are registered. Raises WebSocketConfigurationError if handlers are registered but required routes ($connect, $default) are missing.

()

def validate_registry():
    """Validate that all required routes are registered.

    Raises WebSocketConfigurationError if handlers are registered
    but required routes ($connect, $default) are missing.
    """
    global _validated
    if _validated:
        return
    _validated = True

    if not _registry:
        return

    missing = REQUIRED_ROUTES - set(_registry.keys())
    if missing:
        route_to_decorator = {"$connect": "@on_connect", "$default": "@on_message"}
        missing_names = ", ".join(route_to_decorator.get(r, r) for r in sorted(missing))
        raise WebSocketConfigurationError(
            f"WebSocket handlers registered but missing required routes: {missing_names}. "
            f"All WebSocket apps must define handlers for $connect and $default ($disconnect is optional)."
        )


class WebSocketConfigurationError(Exception):