PublicMirrors/voice-changer
1
0
Fork 0
mirror of https://github.com/w-okada/voice-changer.git synced 2025-02-02 16:23:58 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

Improve allowed origins input and use set

Browse Source
This commit is contained in:
Yury 2024-03-17 16:26:55 +02:00
parent cf2b693334
commit ce9b599501
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
server/restapi/mods/trustedorigin.py
View File

@ -20,15 +20,17 @@ class TrustedOriginMiddleware:
if port is not None: if port is not None:
local_origins = [f'{origin}:{port}' for origin in local_origins] local_origins = [f'{origin}:{port}' for origin in local_origins]
if not allowed_origins: self.allowed_origins: set[str] = set()
allowed_origins = local_origins if allowed_origins is not None:
else:
for origin in allowed_origins: for origin in allowed_origins:
assert urlparse(origin).scheme, ENFORCE_URL_ORIGIN_FORMAT url = urlparse(origin)
allowed_origins = local_origins + allowed_origins assert url.scheme, ENFORCE_URL_ORIGIN_FORMAT
valid_origin = f'{url.scheme}://{url.hostname}'
if url.port:
valid_origin += f':{url.port}'
self.allowed_origins.add(valid_origin)
self.allowed_origins.update(local_origins)
self.app = app self.app = app
self.allowed_origins = list(allowed_origins)
async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None: async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
if scope["type"] not in ( if scope["type"] not in (