From 4ea85bcdc857caed20705c7a0a3b51ee3f0bc522 Mon Sep 17 00:00:00 2001
From: viveksantayana <me@viveksantayana.co.uk>
Date: Mon, 6 Dec 2021 20:10:27 +0000
Subject: [PATCH] Proxy Fix

---
 ref-test/main.py | 3 +++
 1 file changed, 3 insertions(+)

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()