app: Trust X-Forwarded-Proto

Bug: T361471 Change-Id: I600fe27fe9a4e5199d39583a3a11174aea77611f
toolforge · Sep 24, 2024 · 9a50f2f · 9a50f2f
1 parent 5561941
commit 9a50f2f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/quarry/web/app.py b/quarry/web/app.py
@@ -1,5 +1,6 @@
 from flask import current_app, Flask, render_template, g
 from flask_caching import Cache
+from werkzeug.middleware.proxy_fix import ProxyFix
 
 from .config import get_config
 from .connections import Connections
@@ -31,6 +32,7 @@ def kill_context(exception=None):
 
 def create_app(test_config=None):
     app = Flask(__name__)
+    app.wsgi_app = ProxyFix(app.wsgi_app, x_proto=1)
 
     if test_config is None:
         app.config.update(get_config())