Add PROXY_FIX config option to enable werkzeug ProxyFix

2 files changed, 6 insertions, 0 deletions

diff --git a/at.cfg.dist b/at.cfg.dist
index c37efdf..b2e0720 100644
--- a/at.cfg.dist
+++ b/at.cfg.dist
@@ -17,3 +17,5 @@ SPECIAL_DEVICES = {
 }
 
 SECRET_KEY = 'adaba'
+
+PROXY_FIX = False
diff --git a/at.py b/at.py
index 01347bb..8248bd7 100644
--- a/at.py
+++ b/at.py
@@ -23,6 +23,10 @@ app.jinja_env.add_extension('jinja2.ext.i18n')
 app.jinja_env.install_null_translations()
 app.updater = None
 
+if app.config.get('PROXY_FIX'):
+    from werkzeug.contrib.fixers import ProxyFix
+    app.wsgi_app = ProxyFix(app.wsgi_app)
+
 auth = SpaceAuth(app)