python實現(xiàn)獲取Ip歸屬地等信息

更新時間：2016年08月27日 10:54:01 作者：mindg

本文給大家簡單介紹了下如何使用Python實現(xiàn)獲取IP歸屬地信息的方法和代碼，非常的實用，有需要的小伙伴可以參考下

如果你有一批IP地址想要獲得這些IP具體的信息，比如歸屬國家，城市等，最好的辦法當時是調(diào)用現(xiàn)有的api接口來獲取，我在之前就寫過一篇文章，是關于我的博客被莫名攻擊的時，就有獲取過一批IP，攻擊的時候當時是恢復業(yè)務重要，IP該封的就要封，攻擊過后這個攻擊者的IP信息，自己就可以分析下都來自哪里，有沒有什么特征，幫助提示自己網(wǎng)站的安全性，今天這個腳本就是根據(jù)提供的IP獲得IP歸屬的具體信息，腳本如下：

#!/usr/bin/env python
 
import requests
import csv
 
def getIp(file):
  iplist = []
  fi = open(file, 'r')
  for ip in fi:
    ip = ip.strip()
    iplist.append(ip)
  return iplist
 
def get_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)))
  for ip in iplist:
    data = get_geolocation(ip)
    dict_writer.writerow(data)

腳本解釋，定義了2個函數(shù)，getIp()函數(shù)從文件中讀取IP信息，返回一個列表，get_geolocation()函數(shù)負責實際得到信息，返回一個字典，接著在主函數(shù)打開一個可寫的csv文件，通過循環(huán)將得到的信息寫入一個CSV文件中，在寫入csv 文件時用到了csv.DictWriter字典寫的功能，這個還是比較實用，可以將數(shù)據(jù)結構類型為字典的數(shù)據(jù)直接寫入csv文件，換句話說就是你只要把你要寫的數(shù)據(jù)轉(zhuǎn)換成字典的形式，就可以很方便的寫入了，比如我這句dict(zip(fieldnames, fieldnames))其實就是這個功能，大家以后用到操作csv文件時，可以自己實踐實踐。

最后附上一則網(wǎng)友的代碼

#!/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 signal
import urllib
import json
import sys,os,re
import 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]): #如果參數(shù)是文件,迭代查找
  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_address
else:
  ip_address = sys.argv[1]
  if re_ipaddress.match(ip_address): #如果參數(shù)是單個IP地址
    city_address = ip_location(ip_address)
    print ip_address + ":" + city_address
  elif(re_domain.match(ip_address)): #如果參數(shù)是域名
    result = socket.getaddrinfo(ip_address, None)
    ip_address = result[0][4][0]
    city_address = ip_location(ip_address)
    print ip_address.strip() + ":" + city_address

您可能感興趣的文章: