From b4600e01f4bd73981b16da5ce9788f8f32b3c5fa Mon Sep 17 00:00:00 2001
From: Yuval Adam <_@yuv.al>
Date: Mon, 8 Oct 2018 13:33:22 +0300
Subject: Return forwarded IP on prod

---
 app.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/app.py b/app.py
index 3f67127..805ca8a 100644
--- a/app.py
+++ b/app.py
@@ -1,7 +1,14 @@
 from flask import Flask, request
+from os import environ
 
 app = Flask(__name__)
 
+def get_ip(request):
+    if 'PROD' in environ:
+        return request.headers.get('X-Forwarded-For')
+    else:
+        return request.remote_addr
+
 @app.route('/')
 def index():
-    return request.remote_addr
+    return get_ip(request)
-- 
cgit v1.3.1