Linode IP Lookup

lip.py

#!/usr/bin/python
import sys
import os
import os.path
import argparse
import json
import linode.api

key = None

def getArgs():
    parser = argparse.ArgumentParser(prog=sys.argv[0])
    parser.add_argument('--refresh', action='store_true', help="Contact the server to refresh the list")
    parser.add_argument('-a', '--all', action='store_true', help="Print all hosts in the list")
    parser.add_argument('-l', action='store_true', help="Display the host names along with the IPs")
    parser.add_argument('names', nargs='*')

    return parser.parse_args()

def getList(listPath, forceRefresh):
    if not os.path.exists(listPath) or forceRefresh:
        sys.stderr.write("Refreshing data...\n")
        api = linode.api.Api(key=key)

        nodeList = api.linode_list()
        ipList = api.linode_ip_list()

        rawIps = { x["LINODEID"]: x["IPADDRESS"] for x in ipList if x["ISPUBLIC"] }
        ips = { x["LABEL"]: rawIps[x["LINODEID"]] for x in nodeList }

        with open(listPath, "w") as f:
            json.dump(ips, f)
    else:
        with open(listPath, "r") as f:
            ips = json.load(f)

    return ips

def printVerboseData(host, ip):
    print "{0}\t{1}".format(host, ip)

def printSimpleData(host, ip):
    print ip

def main():
    args = getArgs()

    dataPath = os.path.expanduser('~/.lip.cache')
    data = getList(dataPath, args.refresh)

    if args.l:
        printFunc = printVerboseData
    else:
        printFunc = printSimpleData

    if args.all or len(args.names) == 0:
        for host in sorted(data.iterkeys()):
            printFunc(host, data[host])
    else:
        for host in args.names:
            printFunc(host, data[host])

if __name__ == "__main__":
    main()