Python转换IP为具体地理位置小脚本

     近期工作中涉及到处理后台日志，要将大量IP数据转换成具体的地理位置，如“国家-省-市”。去网上找了一下，有很多可以拉过来用的方法，参考这篇文章，http://ju.outofmemory.cn/entry/77846，提供了比较完整的实现方式和具体代码。以此为基础，经过小小修改，任务得以顺利完成。

# !/usr/bin/python
# -*- coding: utf-8 -*-
"""
author : 蛙鳜鸡鹳狸猿
create_time : 2017年 01月 01日 星期日 00:11:00 CST
program : *_* script to transform IP to specific geographic position *_*
"""

import json
import urllib
import urllib2

class Location_Sina():
"""
Interface of sina
Under my test is of higher recognition and performance at home
"""
def __init__(self, ip):
self.ip = urllib.quote_plus(ip)
self.api_url = 'http://int.dpool.sina.com.cn/iplookup/iplookup.php?format=js&ip=%s' % self.ip

def get_geoinfo(self):
urlobj = urllib2.urlopen(self.api_url)
data = urlobj.read()[20:][:-1]
datadict = json.loads(data, encoding='utf-8')
return datadict

def get_country(self):
datadict = self.get_geoinfo()
return datadict['country']

def get_province(self):
datadict = self.get_geoinfo()
return datadict['province']

def get_city(self):
datadict = self.get_geoinfo()
return datadict['city']

class Location_Taobao():
"""
Interface of taobao
Under my test can be the 2ed choice
"""
def __init__(self, ip):
self.ip = urllib.quote_plus(ip)
self.api_url = 'http://ip.taobao.com/service/getIpInfo.php?ip=%s' % self.ip

def get_geoinfo(self):
urlobj = urllib2.urlopen(self.api_url)
data = urlobj.read()
datadict = json.loads(data, encoding='utf-8')
return datadict['data']

def get_country(self):
datadict = self.get_geoinfo()
return datadict['country']

def get_province(self):
datadict = self.get_geoinfo()
return datadict['region']

def get_city(self):
datadict = self.get_geoinfo()
return datadict['city']

def get_isp(self):
datadict = self.get_geoinfo()
return datadict['isp']

class Location_Freegeoip():
"""
Interface of an excellent foreign service freegeoip
The only shortage may be it return result in English
"""
def __init__(self, ip):
self.ip = urllib.quote_plus(ip)
self.api_url = 'http://freegeoip.net/json/%s' % (self.ip)

def get_geoinfo(self):
urlobj = urllib2.urlopen(self.api_url)
data = urlobj.read()
datadict = json.loads(data, encoding='utf-8')
return datadict

def get_country(self):
datadict = self.get_geoinfo()
return datadict['country_name']

def get_province(self):
datadict = self.get_geoinfo()
return datadict['region_name']

def get_city(self):
datadict = self.get_geoinfo()
return datadict['city']

def ip2loc(ip):
"""
An integrated user defined function via three APIs to transform IP to specific geographic position
:param ip:IP
:return:a list of ["country(or other)", "province(or other)", "city(or other)"]
"""
try:
LOC = Location_Sina(ip)
loc = [LOC.get_country(), LOC.get_province(), LOC.get_city()]
if all(loc): pass
else:
for i in range(len(loc)):
if loc[i]: pass
else:
LT_ = Location_Taobao(ip)
if i == 0: loc[i] = LT_.get_country()
elif i == 1: loc[i] = LT_.get_province()
elif i == 2: loc[i] = LT_.get_city()
if all(loc): pass
else:
for j in range(len(loc)):
if loc[j]: pass
else:
LF_ = Location_Freegeoip(ip)
if j == 0: loc[j] = LF_.get_country()
elif j == 1: loc[j] = LF_.get_province()
elif j == 2: loc[j] = LF_.get_city()
if all(loc): pass
else:
for k in range(len(loc)):
if loc[k]: pass
else: loc[k] = "暂未识别"
return loc
except:
loc = ["无法处理", "无法处理", "无法处理"]
return loc

if __name__ == '__main__':
ip = '127.0.0.1'

# Location_Sina
# iploc = Location_Sina(ip)
# print iploc.get_geoinfo()
# print iploc.get_country()
# print iploc.get_province()
# print iploc.get_city()

# Location_Taobao
# iploc = Location_Taobao(ip)
# print iploc.get_geoinfo()
# print iploc.get_country()
# print iploc.get_province()
# print iploc.get_city()
# print iploc.get_isp()

# Location_Freegeoip
# iploc = Location_Freegeoip(ip)
# print iploc.get_geoinfo()
# print iploc.get_country()
# print iploc.get_province()
# print iploc.get_city()

# ip2loc
loc = ip2loc(ip)
print(loc)
print(loc[0])
print(loc[1])
print(loc[2])


     对于IP地址的转换工作，最主要的是关注最后的识别率。个人以自己环境的测试结果为基础，最终选用了新浪、淘宝和国外的Freegeoip三个接口，然后以这三个接口写了一个“ip2loc”函数，确保只要传入一个IP，都能返回一个结果。