Python實(shí)現(xiàn)一個(gè)轉(zhuǎn)存純真IP數(shù)據(jù)庫的腳本分享

更新時(shí)間：2017年05月21日 09:46:50 作者：KBdancer

工作中我們常需要使用純真IP數(shù)據(jù)庫內(nèi)的數(shù)據(jù)做分析，下面這篇文章主要給大家介紹了利用Python如何實(shí)現(xiàn)一個(gè)轉(zhuǎn)存純真IP數(shù)據(jù)庫的相關(guān)資料，對(duì)大家具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面來一起看看吧。

前言

之前寫過很多關(guān)于掃描腳本的文章，一直都沒寫自己的掃描IP段是哪里搞來的，也會(huì)有朋友經(jīng)常來問一些掃描經(jīng)驗(yàn)，說實(shí)話我覺得這個(gè)工具并沒有實(shí)際的技術(shù)含量，但是能提高工作效率，就共享出來給大家耍耍～

談到掃描經(jīng)驗(yàn)，我個(gè)人通常都會(huì)針對(duì)不同的設(shè)備，不同的應(yīng)用選擇不同類型的段。

比如我現(xiàn)在掃描的目標(biāo)是一款電信光貓，那自然是選擇電信的IP段，光貓一般是家庭用戶，我們篩選下家庭用戶的活躍IP段，這樣我們就有針對(duì)性了。

再比如我現(xiàn)在想掃一款企業(yè)路由設(shè)備，那么我就可以選擇企業(yè)公司多的段。

純真IP真心是個(gè)不錯(cuò)的工具，我通常用來檢索某個(gè)地區(qū)的IP段，但是這個(gè)工具有個(gè)非常致命的缺點(diǎn)，就是不能聯(lián)合查詢，而且這玩意每次用都要切換到win下面操作，對(duì)于我這種Linux來說自然是無法忍，索性寫個(gè)把純真IP轉(zhuǎn)存到mysql數(shù)據(jù)庫的腳本，這樣不用每次查詢都去win下面，還可以直接部署到遠(yuǎn)程，查詢方便了很多，最主要的還是支持多條件查詢。

需求

Python寫這個(gè)腳本技術(shù)上難度不大，主要還是用到了MySQLdb庫，關(guān)于MySQLdb庫的安裝就不多說了，博客之前有寫相關(guān)的文章。這里咱們需要先分析下純真IP數(shù)據(jù)庫的數(shù)據(jù)文件結(jié)構(gòu)，發(fā)現(xiàn)其實(shí)每一行的結(jié)構(gòu)都是固定的，那么寫起來就簡單了。

編碼這塊我就不詳細(xì)說了，簡單幾個(gè)方法就能輕松實(shí)現(xiàn)，貼出來代碼

#!/usr/bin/env python
# coding=utf-8
# kbdancer@92ez.com

import MySQLdb
import sys

reload(sys)
sys.setdefaultencoding('utf8')


def save_data_to_mysql(mysql_object, ip_line):
 try:
  begin = ip_line[0:16].replace(' ', '')
  end = ip_line[16:32].replace(' ', '')
  try:
   location = line[32:].split(' ')[0]
  except:
   location = ''
  try:
   isp_type = line[32:].replace(' ', ' ').split(' ')[1].replace('\n', '').replace('\r', '')
  except:
   isp_type = ''

  this_line_value = [begin + "-" + end, location, isp_type]
  do_insert(mysql_object, this_line_value)
 except Exception, e:
  print e


def do_insert(mysql_object, row_data):
 try:
  insert_sql = """INSERT INTO `ipdb` (`iprange`,`location`, `type`) VALUES ( %s, %s, %s )"""
  mysql_object.insert(insert_sql, row_data)
 except Exception, e:
  print row_data
  print e


class Database:
 host = 'localhost'
 user = 'ipdb'
 password = '3u9whrpcEUBTnNNn'
 db = 'ipinfo'
 charset = 'utf8'

 def __init__(self):
  self.connection = MySQLdb.connect(self.host, self.user, self.password, self.db, charset=self.charset)
  self.cursor = self.connection.cursor()

 def insert(self, query, params):
  try:
   self.cursor.execute(query, params)
   self.connection.commit()
  except Exception, e:
   print e
   self.connection.rollback()

 def query(self, query, params):
  cursor = self.connection.cursor(MySQLdb.cursors.DictCursor)
  cursor.execute(query, params)
  return cursor.fetchall()

 def __del__(self):
  self.connection.close()


if __name__ == '__main__':
 mysql = Database()
 ip_file = open(sys.path[0] + "/ip.txt")
 print 'Start save to mysql ...'
 for line in ip_file:
  save_data_to_mysql(mysql, line)
 ip_file.close()
 print 'Save complete.'

注意

由于純真IP數(shù)據(jù)庫導(dǎo)出的txt文件并不是標(biāo)準(zhǔn)的無BOM UTF8編碼，直接解析肯定是失敗的，建議使用Notepad++先轉(zhuǎn)碼一下

效果

原始數(shù)據(jù)