05全球IP归属地查询工具

# 写在前面，这篇文章的原创作者是Charles我只是在他这个程序的基础上边进行加工，另外有一些自己的改造
# 并都附上了注释和我自己的理解，这也是我一个学习的过程。
# 附上大佬的GitHub地址：https://github.com/CharlesPikachu/Tools

'''
Function:
   根据IP地址查其对应的地理信息
Author:
   Charles
微信公众号:
   Charles的皮卡丘
'''
import IPy
import time
import random
import hashlib
import argparse
import requests


headers = {
   'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36'
}



def getaliIP(ip):
   # 这里使用ali的IP地址查询功能。
   # https://market.aliyun.com/products/?keywords=ip%E5%BD%92%E5%B1%9E%E5%9C%B0
   # 需要自己去这个网址注册账号，然后进行调用。
   # 这里我们先进行定义url
   host = 'https://ips.market.alicloudapi.com'
   path = '/iplocaltion'
   method = "GET"
   appcode = '填写你自己的xxx'
   url = host + path + '?ip=' + ip
   # 定义头部。
   headers = {"Authorization": 'APPCODE ' + appcode}
   try:
      # 进行获取调用结果。
      rep = requests.get(url, headers=headers)
   except:
      return 'url参数错误'
   # 判断是否调用成功。如果调用成功就接着进行下边的动作。
   httpStatusCode = rep.status_code
   if httpStatusCode == 200:
      # 转换成json格式
      data = rep.json()
      # 然后获取其中的参数。
      ''''
      # 是以下边这种格式进行返回的。
      {
         "code": 100,
         "message": "success",
         "ip": "110.188.234.66",
         "result": {
                    "en_short": "CN", // 英文简称
         "en_name": "China", // 归属国家英文名称
      "nation": "中国", // 归属国家
      "province": "四川省", // 归属省份
      "city": "绵阳市", // 归属城市
      "district": "涪城区", // 归属县区
      "adcode": 510703, // 归属地编码
      "lat": 31.45498, // 经度
      "lng": 104.75708 // 维度
      }
      }'''
      result1 = data.get('result')
      city = result1['city']
      province = result1['province']
      nation = result1['nation']
      district = result1['district']
      latitude = result1['lat']
      longitude = result1['lng']
      # 返回我们需要的结果。
      result = '-' * 50 + '
' + 
             '''[ali.com查询结果-IP]: %s
经纬度: (%s, %s)
国家: %s
地区: %s
城市: %s
''' % (
             ip, longitude, latitude, nation, province, city) 
             + '-' * 50
   else:
      httpReason = rep.headers['X-Ca-Error-Message']
      if (httpStatusCode == 400 and httpReason == 'Invalid Param Location'):
         return "参数错误"
      elif (httpStatusCode == 400 and httpReason == 'Invalid AppCode'):
         return "AppCode错误"
      elif (httpStatusCode == 400 and httpReason == 'Invalid Url'):
         return "请求的 Method、Path 或者环境错误"
      elif (httpStatusCode == 403 and httpReason == 'Unauthorized'):
         return "服务未被授权（或URL和Path不正确）"
      elif (httpStatusCode == 403 and httpReason == 'Quota Exhausted'):
         return "套餐包次数用完"
      elif (httpStatusCode == 500):
         return "API网关错误"
      else:
         return "参数名错误 或 其他错误" + httpStatusCode + httpReason

   return result

'''淘宝API'''
def getTaobaoIP(ip):
   # 请求淘宝获取IP位置的API接口，但是现在有些不是很好用了。查不出来了。
   # 看了看接口需要进行传入秘钥
   url = 'http(s)://ips.market.alicloudapi.com/iplocaltion'
   # 使用get方法进行请求。
   res = requests.get(url+ip, headers=headers)
   # 然后进行解析参数。
   data = res.json().get('data')
   print(res.json)
   if data is None:
      return '[淘宝API查询结果-IP]: %s
无效IP' % ip
   result = '-'*50 + '
' + 
   '''[淘宝API查询结果-IP]: %s
国家: %s
地区: %s
城市: %s
''' % (ip, data.get('country'), data.get('region'), data.get('city')) 
   + '-'*50
   return result


'''ip-api.com(很不准)'''
def getIpapiIP(ip):
   url = 'http://ip-api.com/json/'
   res = requests.get(url+ip, headers=headers)
   data = res.json()
   yd = youdao()
   city = yd.translate(data.get('city'))[0][0]['tgt']
   country = yd.translate(data.get('country'))[0][0]['tgt']
   region_name = yd.translate(data.get('regionName'))[0][0]['tgt']
   latitude = data.get('lat')
   longitude = data.get('lon')
   result = '-'*50 + '
' + 
   '''[ip-api.com查询结果-IP]: %s
经纬度: (%s, %s)
国家: %s
地区: %s
城市: %s
''' % (ip, longitude, latitude, country, region_name, city) 
   + '-'*50
   return result


'''ipstack.com'''
def getIpstackIP(ip):
   # 定义url
   url = 'http://api.ipstack.com/{}?access_key=1bdea4d0bf1c3bf35c4ba9456a357ce3'
   res = requests.get(url.format(ip), headers=headers)
   data = res.json()
   # 实例化一个有道翻译的类。
   yd = youdao()
   # 调用翻译函数。获取翻译的值。
   continent_name = yd.translate(data.get('continent_name'))[0][0]['tgt']
   country_name = yd.translate(data.get('country_name'))[0][0]['tgt']
   region_name = yd.translate(data.get('region_name'))[0][0]['tgt']
   city = yd.translate(data.get('city'))[0][0]['tgt']
   # 获取经纬度。
   latitude = data.get('latitude')
   longitude = data.get('longitude')
   result = '-'*50 + '
' + 
   '''[ipstack.com查询结果-IP]: %s
经纬度: (%s, %s)
板块: %s
国家: %s
地区: %s
城市: %s
''' % (ip, longitude, latitude, continent_name, country_name, region_name, city) 
   + '-'*50
   return result


'''IP地址有效性验证'''
def isIP(ip):
   try:
      IPy.IP(ip)
      return True
   except:
      return False


'''
Function:
   有道翻译类，进行翻译上边我们查询结果的返回值。
'''
class youdao():
   def __init__(self):
      # 这里我们需要使用post方法进行调用接口。
      self.headers = {
         'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36'
      }

      self.data = {
               'i': 'hello',
               'action': 'FY_BY_CLICKBUTTION',
               'bv': 'e2a78ed30c66e16a857c5b6486a1d326',
               'client': 'fanyideskweb',
               'doctype': 'json',
               'from': 'AUTO',
               'keyfrom': 'fanyi.web',
               'salt': '15532627491296',
               'sign': 'ee5b85b35c221d9be7437297600c66df',
               'smartresult': 'dict',
               'to': 'AUTO',
               'ts': '1553262749129',
               'typoResult': 'false',
               'version': '2.1'
               }
      # 有道翻译调用接口的url
      self.url = 'http://fanyi.youdao.com/translate?smartresult=dict&smartresult=rule'
      # http: // fanyi.youdao.com / translate?smartresult = dict & smartresult = rule & sessionFrom =
   # 进行翻译。
   def translate(self, word):
      # 先判断单词是否为空。
      if word is None:
         return [word]
      # 随机生成一个时间。
      t = str(time.time()*1000 + random.randint(1,10))
      t = str(time.time()*1000 + random.randint(1, 10))
      # 传入我们需要翻译的单词和其他参数。
      self.data['i'] = word
      self.data['salt'] = t
      sign = 'fanyideskweb' + word + t + '6x(ZHw]mwzX#u0V7@yfwK'
      # 这里需要哈希一下。
      self.data['sign'] = hashlib.md5(sign.encode('utf-8')).hexdigest()
      # 进行post方法调用接口，并获取我们需要的参数。
      res = requests.post(self.url, headers=self.headers, data=self.data)
      # 返回翻译的结果。
      return res.json()['translateResult']


'''主函数'''
def main(ip):
   separator = '*' * 30 + 'IPLocQuery' + '*' * 30
   # 首先判断IP地址是否合法。
   if isIP(ip):
      # 然后分别调用几个接口进行查询。
      print(separator)
      print(getaliIP(ip))
      print(getIpstackIP(ip))
      print(getIpapiIP(ip))
      print('*' * len(separator))
   else:
      print(separator + '
[Error]: %s --> 无效IP地址...
' % ip + '*' * len(separator))
   

if __name__ == '__main__':
   # 获取终端输入的入参。
   parser = argparse.ArgumentParser(description="Query geographic information based on IP address.")
   # 可选参数，代表着文件的名字，里边存放着IP之地。
   parser.add_argument('-f', dest='filename', help='File to be queried with one ip address per line')
   # 可选参数，代表着我们需要查询的IP地址。
   parser.add_argument('-ip', dest='ipaddress', help='Single ip address to be queried')
   args = parser.parse_args()
   # 获取终端输入的参数。
   ip = args.ipaddress
   filename = args.filename
   # 判断终端是否有进行输入参数。
   if ip:
      main(ip)
   if filename:
      with open(filename) as f:
         # 获取文件中的所有IP地址，存放成一个列表的形式。
         ips = [ip.strip('
') for ip in f.readlines()]
      for ip in ips:
         main(ip)