ipserv/ipserv.py

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

app = Flask(__name__)
logging.basicConfig(level=logging.INFO)


@app.get("/")
def ip():
    if "X-Forwarded-For" in request.headers:
        ip = request.headers["X-Forwarded-For"]
    else:
        ip = request.remote_addr
    logging.info("looking up IP info for %s", ip)
    try:
        response = requests.get(
            "https://ip.shronk.net/lookup?ip=" + ip,
            headers=request.headers
        )
    except Exception as e:
        logging.error("Failed to get data for %s: %s", ip, e, exc_info=True)
        return {"error": str(e)}, 500
    try:
        data = response.json()
    except json.JSONDecodeError as e:
        logging.error("Failed to parse data for %s: %s", ip, e, exc_info=True)
        return {"error": str(e)}
    logging.info("%s -> %r", ip, data)
    data["ip"] = ip
    data.pop("legalese", None)
    data.pop("source", None)
    data.pop("brexitRequired", None)
    return data, response.status_code