fail2map.py

#! /usr/bin/python -B
# -*- coding: utf-8 -*-

import json
import os
import sys
import urllib2
import random

JSON_FILE = "places.geojson"
GEOIP_API = "http://www.telize.com/geoip/%s"

# free IP geolocation usually gives very approximate results
# there is a big chance that we get the same place multiple times, so
# we might want to add a small offset to GPS coordinates in order to
# view the markers better
ADD_RANDOM_OFFSET = True


def find_lat_lng(ipaddr):
    point = {
        "type": "Feature",
        "geometry": {"type": "Point", "coordinates": [None, None]},
        "properties": {"name": ipaddr, "place": "", "show_on_map": False}
    }
    req = urllib2.urlopen(GEOIP_API % ipaddr)
    geo_value = json.loads(req.read())

    if len(geo_value) > 0:
        point["geometry"]["coordinates"] = [float(geo_value['longitude']), float(geo_value['latitude'])]
        point["properties"]["place"] = "{0}, {1}".format(geo_value.get('city', ''), geo_value.get('country', ''))
        point["properties"]["show_on_map"] = True

    return point


def add(ipaddr):
    newPoint = find_lat_lng(ipaddr)

    # open GPS JSON file
    data = {
            "type": "FeatureCollection",
            "features": [],
    }

    try:
        with open(JSON_FILE, "r") as fh:
            data = json.loads(fh.read())
    except:
        pass

    for place in data["features"]:
        if place["properties"]["name"] == newPoint["properties"]["name"]:
            print("We already have this IP in the file")
            return 0

        if ADD_RANDOM_OFFSET:
            if place["geometry"]["coordinates"] == newPoint["geometry"]["coordinates"]:
                newPoint["geometry"]["coordinates"][0] += random.uniform(-0.1, 0.1)
                newPoint["geometry"]["coordinates"][1] += random.uniform(-0.1, 0.1)

    data["features"].append(newPoint)

    with open(JSON_FILE, 'w') as fh:
        fh.write(json.dumps(data, sort_keys=True, indent=4))

    return 0


if __name__ == '__main__':
    if len(sys.argv) == 3:
        if sys.argv[1] == "add":
            sys.exit(add(sys.argv[2]))
    else:
        print("%s must be called with 'add' and a target IP as first and second arguments." % sys.argv[0])
        print("    ex: python fail2map.py add 0.0.0.0")
        sys.exit(1)