fix: imports

Author: phi-friday

src/socketio-stubs/async_simple_client.pyi

@@ -3,9 +3,9 @@ import types
 from typing import Any, ClassVar, Literal
 
 from _typeshed import Incomplete
-from socketio import Client
 from socketio._types import DataType, TransportType
 from socketio.async_client import AsyncClient
+from socketio.client import Client
 from typing_extensions import Self
 
 class AsyncSimpleClient:

src/socketio-stubs/base_client.pyi

@@ -5,9 +5,10 @@ from typing import Any, ClassVar, Generic, Literal, overload
 
 import engineio
 from _typeshed import Incomplete
-from engineio import AsyncClient, Client
-from socketio import base_namespace
+from engineio.async_client import AsyncClient
+from engineio.client import Client
 from socketio._types import TransportType
+from socketio.base_namespace import BaseClientNamespace
 from typing_extensions import TypeVar
 
 _T_co = TypeVar("_T_co", bound=Client | AsyncClient, covariant=True, default=Any)
@@ -90,9 +91,7 @@ class BaseClient(Generic[_IsAsyncio, _T_co]):
     def event(
         self, handler: Callable[..., Incomplete], namespace: str | None = ...
     ) -> Callable[[_F], _F] | None: ...
-    def register_namespace(
-        self, namespace_handler: base_namespace.BaseClientNamespace
-    ) -> None: ...
+    def register_namespace(self, namespace_handler: BaseClientNamespace) -> 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/base_server.pyi

@@ -4,9 +4,10 @@ from typing import Any, ClassVar, Generic, Literal, overload
 
 import engineio
 from _typeshed import Incomplete
-from engineio import AsyncServer, Server
-from socketio import base_namespace
+from engineio.async_server import AsyncServer
+from engineio.server import Server
 from socketio._types import SyncAsyncModeType, TransportType
+from socketio.base_namespace import BaseClientNamespace
 from socketio.manager import Manager
 from typing_extensions import TypeVar
 
@@ -69,7 +70,7 @@ class BaseServer(Generic[_IsAsyncio, _T_co]):
     @overload
     def event(self, namespace: str | None) -> Callable[[_F], _F]: ...
     def register_namespace(
-        self, namespace_handler: base_namespace.BaseClientNamespace[_IsAsyncio]
+        self, namespace_handler: BaseClientNamespace[_IsAsyncio]
     ) -> None: ...
     def rooms(self, sid: str, namespace: str | None = ...) -> str | list[str]: ...
     def transport(self, sid: str, namespace: str | None = ...) -> TransportType: ...

src/socketio-stubs/client.pyi

@@ -6,13 +6,13 @@ from typing import Any, Literal, ParamSpec, TypeVar
 import engineio
 import requests
 from _typeshed import Incomplete
-from socketio import base_client
 from socketio._types import DataType, TransportType
+from socketio.base_client import BaseClient
 
 _T = TypeVar("_T")
 _P = ParamSpec("_P")
 
-class Client(base_client.BaseClient[Literal[False], engineio.Client]):
+class Client(BaseClient[Literal[False], engineio.Client]):
     connection_url: str  # pyright: ignore[reportIncompatibleVariableOverride]
     connection_headers: dict[Incomplete, Incomplete]  # pyright: ignore[reportIncompatibleVariableOverride]
     connection_auth: Incomplete | None

src/socketio-stubs/msgpack_packet.pyi

@@ -1,8 +1,8 @@
 from _typeshed import Incomplete
-from socketio import packet
+from socketio.packet import Packet
 from typing_extensions import Buffer
 
-class MsgPackPacket(packet.Packet):
+class MsgPackPacket(Packet):
     uses_binary_events: bool
     def encode(self) -> bytes: ...
     packet_type: Incomplete

src/socketio-stubs/server.pyi

@@ -5,14 +5,14 @@ from typing import Any, Generic, Literal, NoReturn, ParamSpec, TypeAlias, overlo
 
 import engineio
 from _typeshed import Incomplete
-from socketio import base_server
 from socketio._types import (
     SessionContextManager,
     SocketIOModeType,
     SyncAsyncModeType,
     TransportType,
 )
 from socketio.admin import InstrumentedServer
+from socketio.base_server import BaseServer
 from socketio.manager import Manager
 from typing_extensions import TypeVar
 
@@ -23,7 +23,7 @@ _T = TypeVar("_T")
 
 default_logger: logging.Logger
 
-class Server(base_server.BaseServer[Literal[False], engineio.Server], Generic[_A]):
+class Server(BaseServer[Literal[False], engineio.Server], Generic[_A]):
     def __init__(
         self,
         client_manager: Manager | None = ...,

src/socketio-stubs/simple_client.pyi

@@ -3,8 +3,8 @@ from threading import Event
 from typing import Any, ClassVar, Literal
 
 from _typeshed import Incomplete
-from socketio import Client
 from socketio._types import DataType, TransportType
+from socketio.client import Client
 from typing_extensions import Self
 
 class SimpleClient: