ipserv/ipserv.py

import requests
import json
from flask import Flask, request, make_response

app = Flask(__name__)


@app.get("/")
def ip():
    if "X-Forwarded-For" in request.headers:
        ip = request.headers["X-Forwarded-For"]
    else:
        ip = request.remote_addr
    try:
        response = requests.get("https://ip.shronk.net/lookup?ip=" + ip)
    except Exception as e:
        return {"error": str(e)}, 500
    try:
        data = response.json()
    except json.JSONDecodeError as e:
        return {"error": str(e)}
    data.pop("legalese", None)
    data.pop("source", None)
    data.pop("brexitRequired", None)
    return data, response.status_code