fix: client namespace

Author: phi-friday

src/socketio-stubs/async_client.pyi

@@ -16,7 +16,9 @@ _P = ParamSpec("_P")
 
 default_logger: logging.Logger
 
-class AsyncClient(BaseClient[Literal[True], engineio.AsyncClient]):
+class AsyncClient(
+    BaseClient[Literal[True], engineio.AsyncClient, AsyncClientNamespace]
+):
     connection_url: str  # pyright: ignore[reportIncompatibleVariableOverride]
     connection_headers: dict[str, str]  # pyright: ignore[reportIncompatibleVariableOverride]
     connection_auth: Any
@@ -85,4 +87,4 @@ class AsyncClient(BaseClient[Literal[True], engineio.AsyncClient]):
         self, target: Callable[_P, Awaitable[_T]], *args: _P.args, **kwargs: _P.kwargs
     ) -> asyncio.Task[_T]: ...
     async def sleep(self, seconds: int = ...) -> None: ...
-    def register_namespace(self, namespace_handler: AsyncClientNamespace) -> None: ...  # pyright: ignore[reportIncompatibleMethodOverride]
+    def register_namespace(self, namespace_handler: AsyncClientNamespace) -> None: ...

src/socketio-stubs/async_namespace.pyi

@@ -75,7 +75,7 @@ class AsyncNamespace(BaseServerNamespace[Literal[True]], Generic[_A]):
     async def disconnect(self, sid: str, namespace: str | None = ...) -> None: ...
 
 class AsyncClientNamespace(BaseClientNamespace[Literal[True]]):
-    client: AsyncClient  # pyright: ignore[reportIncompatibleVariableOverride]
+    client: AsyncClient  # type: ignore[assignment]
     async def trigger_event(self, event: str, *args: Any) -> Any: ...
     async def emit(
         self,

src/socketio-stubs/base_client.pyi

@@ -23,6 +23,9 @@ from socketio.base_namespace import BaseClientNamespace
 from socketio.packet import Packet
 
 _T_co = TypeVar("_T_co", bound=Client | AsyncClient, covariant=True, default=Any)
+_T_namespace = TypeVar(
+    "_T_namespace", bound=BaseClientNamespace[Any], default=BaseClientNamespace[Any]
+)
 _IsAsyncio = TypeVar("_IsAsyncio", bound=bool, default=Literal[False])
 _F = TypeVar("_F", bound=Callable[..., Any])
 _F_event = TypeVar("_F_event", bound=EventHandler)
@@ -40,7 +43,7 @@ def signal_handler(sig: int, frame: FrameType | None) -> Any: ...
 
 original_signal_handler: Callable[[int, FrameType | None], Any] | None
 
-class BaseClient(Generic[_IsAsyncio, _T_co]):
+class BaseClient(Generic[_IsAsyncio, _T_co, _T_namespace]):
     reserved_events: ClassVar[list[str]]
     reason: ClassVar[type[engineio.Client.reason]]
     reconnection: bool
@@ -126,7 +129,7 @@ class BaseClient(Generic[_IsAsyncio, _T_co]):
     def event(self, handler: EventHandler) -> None: ...
     @overload
     def event(self, namespace: str | None) -> Callable[[_F_event], _F_event]: ...
-    def register_namespace(self, namespace_handler: BaseClientNamespace) -> None: ...
+    def register_namespace(self, namespace_handler: _T_namespace) -> None: ...
     def get_sid(self, namespace: str | None = ...) -> str | None: ...
     def transport(self) -> TransportType: ...
     def _engineio_client_class(self) -> type[_T_co]: ...

src/socketio-stubs/client.pyi

@@ -14,7 +14,7 @@ from socketio.packet import Packet
 _T = TypeVar("_T")
 _P = ParamSpec("_P")
 
-class Client(BaseClient[Literal[False], engineio.Client]):
+class Client(BaseClient[Literal[False], engineio.Client, ClientNamespace]):
     connection_url: str  # pyright: ignore[reportIncompatibleVariableOverride]
     connection_headers: dict[str, str]  # pyright: ignore[reportIncompatibleVariableOverride]
     connection_auth: Any
@@ -83,4 +83,4 @@ class Client(BaseClient[Literal[False], engineio.Client]):
         self, target: Callable[_P, _T], *args: _P.args, **kwargs: _P.kwargs
     ) -> Thread: ...
     def sleep(self, seconds: int = ...) -> None: ...
-    def register_namespace(self, namespace_handler: ClientNamespace) -> None: ...  # pyright: ignore[reportIncompatibleMethodOverride]
+    def register_namespace(self, namespace_handler: ClientNamespace) -> None: ...

src/socketio-stubs/namespace.pyi

@@ -69,7 +69,7 @@ class Namespace(BaseServerNamespace[Literal[False]], Generic[_A]):
     def disconnect(self, sid: str, namespace: str | None = ...) -> None: ...
 
 class ClientNamespace(BaseClientNamespace[Literal[False]]):
-    client: Client  # pyright: ignore[reportIncompatibleVariableOverride]
+    client: Client  # type: ignore[assignment]
     def trigger_event(self, event: str, *args: Any) -> Any: ...
     def emit(
         self,