diff --git a/ref-test/main.py b/ref-test/main.py index 9729e0a..c1f0efe 100644 --- a/ref-test/main.py +++ b/ref-test/main.py @@ -8,6 +8,7 @@ from pymongo import MongoClient from pymongo.errors import ConnectionFailure from flask_wtf.csrf import CSRFProtect, CSRFError from flask_mail import Mail +from werkzeug.middleware.proxy_fix import ProxyFix from common.security import check_keyfile_exists, generate_keyfile import config @@ -70,6 +71,8 @@ def create_app(): Bootstrap(app) csrf = CSRFProtect(app) + app.wsgi_app = ProxyFix(app.wsgi_app, x_proto=1, x_host=1) + return app app = create_app()