From a531f49b30bad6c72e8ee352b8ffd689a90d8fb2 Mon Sep 17 00:00:00 2001 From: Adam Griffiths Date: Thu, 16 Jun 2022 15:30:20 +1000 Subject: [PATCH] Add **kwargs to SocketManager Pass these to socketio.AsyncServer. --- fastapi_socketio/socket_manager.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/fastapi_socketio/socket_manager.py b/fastapi_socketio/socket_manager.py index 083c7f6..f10b10c 100644 --- a/fastapi_socketio/socket_manager.py +++ b/fastapi_socketio/socket_manager.py @@ -5,7 +5,7 @@ class SocketManager: """ - Integrates SocketIO with FastAPI app. + Integrates SocketIO with FastAPI app. Adds `sio` property to FastAPI object (app). Default mount location for SocketIO app is at `/ws` @@ -23,10 +23,11 @@ def __init__( mount_location: str = "/ws", socketio_path: str = "socket.io", cors_allowed_origins: Union[str, list] = '*', - async_mode: str = "asgi" + async_mode: str = "asgi", + **kwargs ) -> None: # TODO: Change Cors policy based on fastapi cors Middleware - self._sio = socketio.AsyncServer(async_mode=async_mode, cors_allowed_origins=cors_allowed_origins) + self._sio = socketio.AsyncServer(async_mode=async_mode, cors_allowed_origins=cors_allowed_origins, **kwargs) self._app = socketio.ASGIApp( socketio_server=self._sio, socketio_path=socketio_path )