python開發ip2region 離線IP庫地址文件

一、項目簡介

ip2region - 離線的ip地址查詢庫，ip到地區的映射庫，提供二進制，B樹，內存搜索三種查詢算法，查詢速度非常快。
支持Java，PHP，C，Python，Nodejs，Golang，C#等語言，本文以Python為例

下載地址：https://github.com/lionsoul2014/ip2region

項目測試代碼：本例中ip2region.db與testSearcher.py在同一個目錄

D:\code\ip2region\binding\python>python testSearcher.py ip2region.db
initializing b-tree...
+----------------------------------+
| ip2region test program |
| Author: [e-mail]. |
| Type 'quit' to exit program |
+----------------------------------+
ip2region>> 45.15.11.150
0|中國|0|香港|0|0 in 98.978027 millseconds

ip2region>> 206.189.50.204
14|新加坡|0|0|0|0 in 0.000000 millseconds
ip2region>> 106.13.211.178
2140|中國|0|廣東省|廣州市|電信 in 0.000000 millseconds
ip2region>>

二、詳細分析

 

 

python 示例代碼：testSearcher.py、ip2Region.py、benchmark.py

 

 

 

離線代碼庫ip2region.db在data目錄下。

執行test.py示例如下，生成結果文件result.csv。

result.csv

 

 輸入文件是ip.txt。

三、代碼test.py

import time
from ip2Region import Ip2Region
import csv

def testSearch(ip_s):
    dbFile = "ip2region.db"
    searcher = Ip2Region(dbFile)
    try:
        print("開始檢測：",(ip_s))
        sTime = time.time() * 1000
        data = searcher.binarySearch(ip_s)
        # elif algorithm == "memory":
        #     data = searcher.memorySearch(line)
        # else:
        #     data = searcher.btreeSearch(line)
        eTime = time.time() * 1000
        ip_info = ("%s|%s|%s" % (ip_s, data["city_id"], data["region"].decode('utf-8')))
        print("檢測完成：" + ip_info)
        return ip_info

    except Exception as e:
        print("[Error]: %s" % e)
    searcher.close()


def all_in():

    c = open("result.csv","w", newline="")
    writer=csv.writer(c)
    title = ['源ip', '國家', '省市','運營商']
    writer.writerow(title)
    with open('ip.txt' ,'r') as file:
        for line in file.readlines():
            ip = line.strip()
            try:
                data = testSearch(ip)
                ct = data.split('|')[2].strip()
                pv = data.split('|')[4].strip()
                city = data.split('|')[5].strip()
                yys = data.split('|')[6].strip()
                if ct == "0":
                    print("地址庫中未找到對應的IP歸屬地，請更新地址庫或者確定ip准確性！")
                    writer.writerow([ip, "/", "/","/"])
                else:
                    if pv == "0":
                        print("省市查詢為空！")
                        writer.writerow([ip, ct, "/","/"])
                    else:
                        if city == "0":
                            writer.writerow([ip, ct, pv + "-" + "///","/"])
                        else:
                            if yys == "0":
                                writer.writerow([ip, ct, pv + "-" + city,"/"])
                            else:
                                writer.writerow([ip, ct, pv + "-" + city,yys])
            except Exception as e:
                print("[Error]: %s" % e)
                writer.writerow([ip,"檢測異常，請手動檢測！"])
    c.close()

if __name__ == "__main__":
    all_in()