ip 位置信息查询，支持管道、文件、命令行参数三种输入方式，信息查询命令行参数,#!/usr/bin/e

ip 位置信息查询，支持管道、文件、命令行参数三种输入方式，信息查询命令行参数,#!/usr/bin/e

#!/usr/bin/env python#-*-coding:utf-8-*-"""    调用 淘宝 ip  restful api 进行查询，    支持直接指定 ip 地址查询 ，    也支持读取文件按行解析查询地址    同时支持从管道读取数据进行查询"""import requestsimport argparseimport sysURL = "http://ip.taobao.com/service/getIpInfo.php?ip="def query_ip_info(ip):    if not ip:        print 'argument shoud not be none'        return    url = URL + str(ip)    try:        r = requests.get(url, timeout=5)    except Exception as e:        print e    else:        data = r.json()        if data['code'] == 0:            data = data['data']            print ip, data['country'], "/", data['area'], "/", data['region'], "/", data['city'], "/", data['isp']        else:            print "查询失败: ", ip, data['data']def parse_args():    parser = argparse.ArgumentParser()    parser.add_argument("-a", "--addr", help="输入单个 ip 地址进行查询")    parser.add_argument("-f", "--file", help="输入文件路径，从文件中按行进行解析读取数据")    parser.add_argument("-p", "--pipe", default=sys.stdin, const=sys.stdin, nargs="?", help="从管道中读取数据")    args = parser.parse_args()    if not any((args.addr, args.file, args.pipe)):        parser.print_usage()        sys.exit()    elif args.addr:        return ("addr", args.addr)    elif args.file:        return ("file", args.file)    else:        return ("pipe", sys.stdin)def run():    source, ips = parse_args()    if source == "pipe":        for ip in sys.stdin:            query_ip_info(ip.strip())    elif source == "addr":        query_ip_info(ips)    elif source == "file":        try:            with open(ips, "r") as f:                while True:                    ip = f.readline().strip()                    if not ip:                        break                    query_ip_info(ip)        except Exception as e:            print e    else:        print "unknown error"def test():    run()if __name__ == "__main__":    test()