feat: add json

Author: phi-friday

src/socketio-stubs/async_aiopika_manager.pyi

@@ -6,6 +6,7 @@ from aio_pika.abc import (
     AbstractRobustExchange,
 )
 
+from socketio._types import JsonModule
 from socketio.async_pubsub_manager import AsyncPubSubManager
 
 class AsyncAioPikaManager(AsyncPubSubManager):
@@ -20,4 +21,5 @@ class AsyncAioPikaManager(AsyncPubSubManager):
         channel: str = ...,
         write_only: bool = ...,
         logger: logging.Logger | None = ...,
+        json: JsonModule | None = ...,
     ) -> None: ...

src/socketio-stubs/async_pubsub_manager.pyi

@@ -3,7 +3,7 @@ import logging
 from collections.abc import Callable
 from typing import Any
 
-from socketio._types import DataType
+from socketio._types import DataType, JsonModule
 from socketio.async_manager import AsyncManager
 
 class AsyncPubSubManager(AsyncManager):
@@ -17,6 +17,7 @@ class AsyncPubSubManager(AsyncManager):
         channel: str = ...,
         write_only: bool = ...,
         logger: logging.Logger | None = ...,
+        json: JsonModule | None = ...,
     ) -> None: ...
     thread: asyncio.Task[Any]
     def initialize(self) -> None: ...

src/socketio-stubs/async_redis_manager.pyi

@@ -6,7 +6,7 @@ from redis.asyncio.client import PubSub as RedisPubSub
 from valkey.asyncio import Valkey
 from valkey.asyncio.client import PubSub as ValkeyPubSub
 
-from socketio._types import RedisArgs
+from socketio._types import JsonModule, RedisArgs
 from socketio.async_pubsub_manager import AsyncPubSubManager
 
 def parse_redis_sentinel_url(
@@ -26,5 +26,6 @@ class AsyncRedisManager(AsyncPubSubManager):
         channel: str = ...,
         write_only: bool = ...,
         logger: logging.Logger | None = ...,
+        json: JsonModule | None = ...,
         redis_options: dict[str, Any] | None = ...,
     ) -> None: ...

src/socketio-stubs/base_manager.pyi

@@ -4,6 +4,7 @@ from typing import Any
 
 from bidict import bidict
 
+from socketio._types import JsonModule
 from socketio.base_server import BaseServer
 
 default_logger: logging.Logger
@@ -17,6 +18,7 @@ class BaseManager:
     eio_to_sid: dict[str, str]
     callbacks: dict[str, dict[int, Callable[..., Any]]]
     pending_disconnect: dict[str, list[str]]
+    json: JsonModule
     def __init__(self) -> None: ...
     def set_server(self, server: BaseServer[Any, Any]) -> None: ...
     def initialize(self) -> None: ...

src/socketio-stubs/kafka_manager.pyi

@@ -2,6 +2,7 @@ import logging
 
 import kafka
 
+from socketio._types import JsonModule
 from socketio.pubsub_manager import PubSubManager
 
 logger: logging.Logger
@@ -12,5 +13,10 @@ class KafkaManager(PubSubManager):
     producer: kafka.KafkaProducer
     consumer: kafka.KafkaConsumer
     def __init__(
-        self, url: str = ..., channel: str = ..., write_only: bool = ...
+        self,
+        url: str = ...,
+        channel: str = ...,
+        write_only: bool = ...,
+        logger: logging.Logger | None = ...,
+        json: JsonModule | None = ...,
     ) -> None: ...

src/socketio-stubs/kombu_manager.pyi

@@ -3,6 +3,7 @@ from typing import Any
 
 import kombu
 
+from socketio._types import JsonModule
 from socketio.pubsub_manager import PubSubManager
 
 class KombuManager(PubSubManager):
@@ -19,6 +20,7 @@ class KombuManager(PubSubManager):
         channel: str = ...,
         write_only: bool = ...,
         logger: logging.Logger | None = ...,
+        json: JsonModule | None = ...,
         connection_options: dict[str, Any] | None = ...,
         exchange_options: dict[str, Any] | None = ...,
         queue_options: dict[str, Any] | None = ...,

src/socketio-stubs/pubsub_manager.pyi

@@ -3,7 +3,7 @@ from collections.abc import Callable
 from threading import Thread
 from typing import Any
 
-from socketio._types import DataType
+from socketio._types import DataType, JsonModule
 from socketio.manager import Manager
 
 class PubSubManager(Manager):
@@ -17,6 +17,7 @@ class PubSubManager(Manager):
         channel: str = ...,
         write_only: bool = ...,
         logger: logging.Logger | None = ...,
+        json: JsonModule | None = ...,
     ) -> None: ...
     thread: Thread
     def initialize(self) -> None: ...

src/socketio-stubs/redis_manager.pyi

@@ -6,7 +6,7 @@ from redis.client import PubSub as RedisPubSub
 from valkey import Valkey
 from valkey.client import PubSub as ValkeyPubSub
 
-from socketio._types import RedisArgs
+from socketio._types import JsonModule, RedisArgs
 from socketio.pubsub_manager import PubSubManager
 
 logger: logging.Logger
@@ -28,6 +28,7 @@ class RedisManager(PubSubManager):
         channel: str = ...,
         write_only: bool = ...,
         logger: logging.Logger | None = ...,
+        json: JsonModule | None = ...,
         redis_options: dict[str, Any] | None = ...,
     ) -> None: ...
     def initialize(self) -> None: ...

src/socketio-stubs/zmq_manager.pyi

@@ -3,6 +3,7 @@ from collections.abc import Generator
 
 from eventlet.green.zmq import Socket, _Socket_recv
 
+from socketio._types import JsonModule
 from socketio.pubsub_manager import PubSubManager
 
 class ZmqManager(PubSubManager):
@@ -15,6 +16,7 @@ class ZmqManager(PubSubManager):
         url: str = ...,
         channel: str = ...,
         write_only: bool = ...,
-        logger: logging.Logger = ...,
+        logger: logging.Logger | None = ...,
+        json: JsonModule | None = ...,
     ) -> None: ...
     def zmq_listen(self) -> Generator[_Socket_recv]: ...