From ab9b4257f21ed957d1f7a0a8dc6fb885dc6a9e13 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mose=20M=C3=BCller?= Date: Mon, 19 Aug 2024 07:24:31 +0200 Subject: [PATCH] adds overwritable sio client_manager --- src/pydase/server/web_server/sio_setup.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/src/pydase/server/web_server/sio_setup.py b/src/pydase/server/web_server/sio_setup.py index 20b1f07..bac9f3b 100644 --- a/src/pydase/server/web_server/sio_setup.py +++ b/src/pydase/server/web_server/sio_setup.py @@ -23,6 +23,7 @@ logger = logging.getLogger(__name__) # These functions can be monkey-patched by other libraries at runtime dump = pydase.utils.serialization.serializer.dump +sio_client_manager = None class UpdateDict(TypedDict): @@ -93,9 +94,16 @@ def setup_sio_server( state_manager = observer.state_manager if enable_cors: - sio = socketio.AsyncServer(async_mode="aiohttp", cors_allowed_origins="*") + sio = socketio.AsyncServer( + async_mode="aiohttp", + cors_allowed_origins="*", + client_manager=sio_client_manager, + ) else: - sio = socketio.AsyncServer(async_mode="aiohttp") + sio = socketio.AsyncServer( + async_mode="aiohttp", + client_manager=sio_client_manager, + ) setup_sio_events(sio, state_manager) setup_logging_handler(sio)