Add ProxyFix middleware

This is needed for the proper request URLs to be detected during login.

1 files changed, 3 insertions, 2 deletions

diff --git a/fbin/__init__.py b/fbin/__init__.py
index f36d74b..7bcdf0b 100644
--- a/fbin/__init__.py
+++ b/fbin/__init__.py
@@ -1,13 +1,14 @@
 from flask import Flask, url_for, Markup, request
 from flask_login import current_user
 from werkzeug.routing import BaseConverter
+from werkzeug.middleware.proxy_fix import ProxyFix
 
 app = Flask(__name__)
 app.config.from_pyfile('fbin.cfg')
-
-# Set up some custom converters. These are needed for file URLs to be properly parsed.
+app.wsgi_app = ProxyFix(app.wsgi_app)
 
 
+# Set up some custom converters. These are needed for file URLs to be properly parsed.
 class HashConverter(BaseConverter):
     regex = r'\w+'