geoip.py

from ipaddress import IPv4Address
import json
from urllib3.poolmanager import PoolManager


base_url = 'http://ip-api.com/json/{query}?fields=status,country,countryCode'
http = PoolManager(num_pools=20)


def _ip_detail(ip: IPv4Address):
    assert isinstance(ip, IPv4Address), 'wrong ip type'
    url = base_url.format(query=str(ip))
    resp = http.request('GET', url)
    if resp.status != 200:
        return
    data = json.loads(resp.data)
    if data['status'] == 'faild':
        return
    return data


def get_country(ip: IPv4Address):
    d = _ip_detail(ip) or {}
    return d.get('country', '').lower()


def _2ipv4(ip: str):
    try:
        r = IPv4Address(ip)
        return r
    except:
        raise ValueError('Invali IPv4')


if __name__ == '__main__':
    print(get_country(_2ipv4(input('enter ip: ').strip())))