fixed export

Author: eadwinCode

ellar/socket_io/__init__.py

@@ -1,4 +1,24 @@
+try:
+    import socketio  # noqa
+except Exception:  # pragma: no cover
+    raise Exception("socketio package is required. Use `pip install socketio`.")
+
+from .decorators import (
+    WebSocketGateway,
+    on_connected,
+    on_disconnected,
+    subscribe_message,
+)
 from .factory import GatewayRouterFactory
 from .responses import WsResponse
+from .testing import TestGateway
 
-__all__ = ["WsResponse", "GatewayRouterFactory"]
+__all__ = [
+    "WsResponse",
+    "GatewayRouterFactory",
+    "on_disconnected",
+    "on_connected",
+    "subscribe_message",
+    "WebSocketGateway",
+    "TestGateway",
+]

ellar/socket_io/decorators/gateway.py

@@ -51,15 +51,11 @@ def WebSocketGateway(
     kwargs.setdefault("async_mode", "asgi")
     kwargs.setdefault("cors_allowed_origins", "*")
 
-    _path = path
-
-    if path and isinstance(path, type):
-        _path = "/socket.io"
-
     def _decorator(cls: t.Type) -> t.Type:
+        assert path == "" or path.startswith("/"), "Routed paths must start with '/'"
         _kwargs = AttributeDict(
             socket_init_kwargs=dict(kwargs),
-            path=_path,
+            path=path,
             name=name,
             include_in_schema=False,
         )
@@ -105,5 +101,7 @@ def _decorator(cls: t.Type) -> t.Type:
         return _gateway_type
 
     if callable(path):
-        return _decorator(path)  # type:ignore[arg-type]
+        func = path
+        path = "/socket.io"
+        return _decorator(func)  # type:ignore[arg-type]
     return _decorator