Python查询IP地址归属完整代码

#!/usr/bin/env python# -*- coding: utf-8 -*-#查找IP地址归属地#writer by keery_log#Create time:2013-10-30#Last update:2013-10-30#用法: python chk_ip.py www.google.com |python chk_ip.py 8.8.8.8 |python chk_ip.py ip.txt import signalimport urllibimport jsonimport sys,os,reimport socket if len(sys.argv) <= 1 :  print "Please input ip address !"  sys.exit(0) def handler(signum, frame):  sys.exit(0)signal.signal(signal.SIGINT, handler) url = "http://ip.taobao.com/service/getIpInfo.php?ip=" #查找IP地址def ip_location(ip):  data = urllib.urlopen(url + ip).read()  datadict=json.loads(data)   for oneinfo in datadict:    if "code" == oneinfo:      if datadict[oneinfo] == 0:        return datadict["data"]["country"] + datadict["data"]["region"] + datadict["data"]["city"] + datadict["data"]["isp"] #定义IP与域名正则re_ipaddress = re.compile(r'^((25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)/.){3}(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)$')re_domain = re.compile(r'[a-zA-Z0-9][-a-zA-Z0-9]{0,62}(/.[a-zA-Z0-9][-a-zA-Z0-9]{0,62})+/.?') if os.path.isfile(sys.argv[1]): #如果参数是文件,迭代查找  file_path = sys.argv[1]  fh = open(file_path,'r')  for line in fh.readlines():    if re_ipaddress.match(line):      city_address = ip_location(line)      print line.strip() + ":" + city_addresselse:  ip_address = sys.argv[1]  if re_ipaddress.match(ip_address): #如果参数是单个IP地址    city_address = ip_location(ip_address)    print ip_address + ":" + city_address  elif(re_domain.match(ip_address)): #如果参数是域名    result = socket.getaddrinfo(ip_address, None)    ip_address = result[0][4][0]    city_address = ip_location(ip_address)    print ip_address.strip() + ":" + city_address