如果你有一批IP位址想要獲得這些IP具體的資訊,比如歸屬國家,城市等,最好的辦法當時是調用現有的api接口來擷取,具體應用場景可以根據自己需要,比如我在之前就寫過一篇文章,是關于我的部落格被莫名攻擊的時,就有擷取過一批IP,攻擊的時候當然是恢複業務重要,IP該封的就要封,攻擊過後這個攻擊者的IP資訊,自己就可以分析下都來自哪裡,有沒有什麼特征,幫助提示自己網站的安全性,今天這個腳本就是根據提供的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個函數,getIp()函數從檔案中讀取IP資訊,傳回一個清單,get_geolocation()函數負責實際得到資訊,傳回一個字典,接着在主函數打開一個可寫的csv檔案,通過循環将得到的資訊寫入一個CSV檔案中,在寫入csv 檔案時用到了csv.DictWriter字典寫的功能,這個還是比較實用,可以将資料結構類型為字典的資料直接寫入csv檔案,換句話說就是你隻要把你要寫的資料轉換成字典的形式,就可以很友善的寫入了,比如我這句dict(zip(fieldnames, fieldnames))其實就是這個功能,大家以後用到操作csv檔案時,可以自己實踐實踐。