python实现获取Ip归属地等信息
如果你有一批IP地址想要获得这些IP具体的信息，比如归属国家，城市等，最好的办法当时是调用现有的api接口来获取，我在之前就写过一篇文章，是关于我的博客被莫名攻击的时，就有获取过一批IP，攻击的时候当时是恢复业务重要，IP该封的就要封，攻击过后这个攻击者的IP信息，自己就可以分析下都来自哪里，有没有什么特征，帮助提示自己网站的安全性，今天这个脚本就是根据提供的IP获得IP归属的具体信息，脚本如下：
#!/usr/bin/envpython
importrequests
importcsv
defgetIp(file):
iplist=[]
fi=open(file,'r')
foripinfi:
ip=ip.strip()
iplist.append(ip)
returniplist
defget_geolocation(ip):
r=requests.get('https://freegeoip.net/json/'+ip)
info=[str(r.json()['country_name']),str(r.json()['city'])]
return{'ip':ip,'country_name':info[0],'city_name':info[1]}
if__name__=='__main__':
iplist=getIp('/root/ipfile')
f=open('outputinfo.csv','a+')
fieldnames=['ip','country_name','city_name']
dict_writer=csv.DictWriter(f,fieldnames=fieldnames)
dict_writer.writerow(dict(zip(fieldnames,fieldnames)))
foripiniplist:
data=get_geolocation(ip)
dict_writer.writerow(data)
脚本解释，定义了2个函数，getIp()函数从文件中读取IP信息，返回一个列表，get_geolocation()函数负责实际得到信息，返回一个字典，接着在主函数打开一个可写的csv文件，通过循环将得到的信息写入一个CSV文件中，在写入csv文件时用到了csv.DictWriter字典写的功能，这个还是比较实用，可以将数据结构类型为字典的数据直接写入csv文件，换句话说就是你只要把你要写的数据转换成字典的形式，就可以很方便的写入了，比如我这句dict(zip(fieldnames,fieldnames))其实就是这个功能，大家以后用到操作csv文件时，可以自己实践实践。
最后附上一则网友的代码
#!/usr/bin/envpython
#-*-coding:utf-8-*-
#查找IP地址归属地
#writerbykeery_log
#Createtime:2013-10-30
#Lastupdate:2013-10-30
#用法:pythonchk_ip.pywww.google.com|pythonchk_ip.py8.8.8.8|pythonchk_ip.pyip.txt
importsignal
importurllib
importjson
importsys,os,re
importsocket
iflen(sys.argv)<=1:
print"Pleaseinputipaddress!"
sys.exit(0)
defhandler(signum,frame):
sys.exit(0)
signal.signal(signal.SIGINT,handler)
url="http://ip.taobao.com/service/getIpInfo.php?ip="
#查找IP地址
defip_location(ip):
data=urllib.urlopen(url+ip).read()
datadict=json.loads(data)
foroneinfoindatadict:
if"code"==oneinfo:
ifdatadict[oneinfo]==0:
returndatadict["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})+\.?')
ifos.path.isfile(sys.argv[1]):#如果参数是文件,迭代查找
file_path=sys.argv[1]
fh=open(file_path,'r')
forlineinfh.readlines():
ifre_ipaddress.match(line):
city_address=ip_location(line)
printline.strip()+":"+city_address
else:
ip_address=sys.argv[1]
ifre_ipaddress.match(ip_address):#如果参数是单个IP地址
city_address=ip_location(ip_address)
printip_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)
printip_address.strip()+":"+city_address