app.py


1
2
3
4
5
6
7
8
9
10
11
12
13
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 get_ip(request)