cymruwhois_ips.py

#!/usr/bin/env python3

import sys
import json
from cymruwhois import Client

if len(sys.argv) != 3:
    sys.exit("Usage: python3 {} <input_file> <output_file>\n\
              input_file should be a text file containing a unique list of IPs\n\
              output_file will be a json formatted file of results".format(sys.argv[0]))

cy = Client()
ips = []
results = {}

with open(sys.argv[1], "r") as fh:
    for line in fh.readlines():
        ips.append(line.strip())

for ip in ips:
    result = cy.lookup(ip)
    if result.prefix in results:
        results[result.prefix]["ips"].append(result.ip)
        results[result.prefix]["count"] += 1
    else:
        results[result.prefix] = {
            "ips": [result.ip,],
            "count": 1,
            "asn": result.asn,
            "country": result.cc,
            "owner": result.owner
        }

with open(sys.argv[2], "w+") as fh:
    fh.write(json.dumps(results, indent=4))