geolocate w/ ip address #python

geolocate_ip.py

import requests
import json

def geolocate_ip(ip):
    '''Get location data based on ip.

    Note: Limited (Free) to a 1000 requests per day.
    TODO(Leonard): Handler for being behind a vpn

    Inputs:
        ip: str
    Returns:
        loc: dict with keys
    '''
    try:
        geo_keys = ('continent_code',
                    'country',
                    'region',
                    'latitude',
                    'longitude',
                    'timezone',
                    'postal')
        r = requests.get(f'https://ipapi.co/{ip}/json/')
        loc = {k: r.json()[k] for k in geo_keys}

        return loc
    except requests.exceptions.HTTPError as e:
        return logger.error(e)