Merge pull request #384 from OperationCode/fix-rate-limiting

Add ProxyFix to rate limiting

Author: projectLewis

app/__init__.py

@@ -6,6 +6,8 @@
 from flask_sqlalchemy import SQLAlchemy
 from flask_limiter import Limiter
 from flask_limiter.util import get_remote_address
+from werkzeug.middleware.proxy_fix import ProxyFix
+from os import environ
 
 from healthcheck import HealthCheck
 # from healthcheck import EnvironmentDump
@@ -23,6 +25,8 @@
 app.config.from_object(Config)
 app.url_map.strict_slashes = False
 
+if environ['FLASK_ENV'] != 'development':
+    app = ProxyFix(app, x_for=1, x_host=1)
 limiter = Limiter(
     app,
     key_func=get_remote_address,