arno/etesync-server
1
0
Fork 0
mirror of https://github.com/etesync/server synced 2024-12-29 10:48:07 +00:00
Code Issues Releases Wiki Activity

Add trusted host middleware.

Browse Source
This commit is contained in:
Tom Hacohen 2020-12-28 10:57:40 +02:00
parent 38884fead8
commit ad2205e596
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
etebase_fastapi/app.py
View File

@ -1,7 +1,6 @@
import os import os
from django.core.wsgi import get_wsgi_application from django.core.wsgi import get_wsgi_application
from fastapi.middleware.cors import CORSMiddleware
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "etebase_server.settings") os.environ.setdefault("DJANGO_SETTINGS_MODULE", "etebase_server.settings")
application = get_wsgi_application() application = get_wsgi_application()
@ -10,6 +9,8 @@ from django.conf import settings
# Not at the top of the file because we first need to setup django # Not at the top of the file because we first need to setup django
from fastapi import FastAPI, Request from fastapi import FastAPI, Request
from fastapi.middleware.cors import CORSMiddleware
from fastapi.middleware.trustedhost import TrustedHostMiddleware
from .exceptions import CustomHttpException from .exceptions import CustomHttpException
from .authentication import authentication_router from .authentication import authentication_router
@ -35,6 +36,7 @@ if settings.DEBUG:
app.add_middleware( app.add_middleware(
CORSMiddleware, allow_origin_regex="https?://.*", allow_credentials=True, allow_methods=["*"], allow_headers=["*"] CORSMiddleware, allow_origin_regex="https?://.*", allow_credentials=True, allow_methods=["*"], allow_headers=["*"]
) )
app.add_middleware(TrustedHostMiddleware, allowed_hosts=settings.ALLOWED_HOSTS)
@app.exception_handler(CustomHttpException) @app.exception_handler(CustomHttpException)