From 890d8aeec8464881115d5acc01c77fb05c7069dd Mon Sep 17 00:00:00 2001 From: "Matthew D. Scholefield" Date: Wed, 25 Nov 2020 14:33:04 -0600 Subject: [PATCH] Use cors_allowed_origins in AsyncServer This allows controlling cors in the socket server --- fastapi_socketio/socket_manager.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/fastapi_socketio/socket_manager.py b/fastapi_socketio/socket_manager.py index 653d6b8..c5a57ab 100644 --- a/fastapi_socketio/socket_manager.py +++ b/fastapi_socketio/socket_manager.py @@ -1,4 +1,5 @@ import socketio +from typing import Union from fastapi import FastAPI @@ -21,10 +22,10 @@ def __init__( app: FastAPI, mount_location: str = "/ws", socketio_path: str = "socket.io", - cors_allowed_origins: list = [], + cors_allowed_origins: Union[str, list] = '*', ) -> None: # TODO: Change Cors policy based on fastapi cors Middleware - self._sio = socketio.AsyncServer(async_mode="asgi", cors_allowed_origins="*") + self._sio = socketio.AsyncServer(async_mode="asgi", cors_allowed_origins=cors_allowed_origins) self._app = socketio.ASGIApp( socketio_server=self._sio, socketio_path=socketio_path )