当前位置：Gxlcms > Python > python访问纯真IP数据库的代码

python访问纯真IP数据库的代码

时间：2021-07-01 10:21:17 帮助过：22人阅读

核心代码：

  1. #!/usr/bin/env python 
  2. # -*- coding: utf-8 -*- 
  3. from bisect import bisect 
  4. _LIST1, _LIST2 = [], [] 
  5. _INIT = False 
  6. ip2int = lambda ip_str: reduce(lambda a, b: (a << 8) + b, [int(i) for i in ip_str.split('.')]) 
  7. def _init(): 
  8. global _LIST, _INIT 
  9. if not _INIT: 
  10. for l in open('ipdata.txt', 'rb'): 
  11. ip1, ip2 = l.split()[:2] 
  12. addr = ' '.join(l.split()[2:]) 
  13. ip1, ip2 = ip2int(ip1), ip2int(ip2) 
  14. _LIST1.append(ip1) 
  15. _LIST2.append((ip1, ip2, addr)) 
  16. _INIT = True 
  17. def ip_from(ip): 
  18. _init() 
  19. i = ip2int(ip) 
  20. idx = bisect(_LIST1, i) 
  21. assert(idx > 0) 
  22. if len(_LIST1) <= idx: 
  23. return u'unknown ip address %s' % ip 
  24. else: 
  25. frm, to ,addr = _LIST2[idx - 1] 
  26. if frm <= i <= to: 
  27. return addr 
  28. else: 
  29. return u'unknown ip address %s' % ip 
  30. if __name__ == '__main__': 
  31. print ip_from('115.238.54.106') 
  32. print ip_from('220.181.29.160') 
  33. print ip_from('115.238.54.107') 
  34. print ip_from('8.8.8.8')

代码打包下载 http://xiazai.bitsCN.com/201105/yuanma/ipaddress.7z

接下来为大家分享更完美的代码：

  1. #!/usr/bin/env python
  2. # coding: utf-8
  3.  
  4. '''用Python脚本查询纯真IP库
  5.  
  6. QQWry.Dat的格式如下:
  7.  
  8. +----------+
  9. | 文件头 | (8字节)
  10. +----------+
  11. | 记录区 | （不定长）
  12. +----------+
  13. | 索引区 | （大小由文件头决定）
  14. +----------+
  15.  
  16. 文件头：4字节开始索引偏移值+4字节结尾索引偏移值
  17.  
  18. 记录区： 每条IP记录格式 ==> IP地址[国家信息][地区信息]
  19.  
  20.   对于国家记录，可以有三种表示方式：
  21.  
  22.     字符串形式(IP记录第5字节不等于0x01和0x02的情况)，
  23.     重定向模式1(第5字节为0x01),则接下来3字节为国家信息存储地的偏移值
  24.     重定向模式(第5字节为0x02),
  25.  
  26.   对于地区记录，可以有两种表示方式： 字符串形式和重定向
  27.  
  28.   最后一条规则：重定向模式1的国家记录后不能跟地区记录
  29.  
  30. 索引区： 每条索引记录格式 ==> 4字节起始IP地址 + 3字节指向IP记录的偏移值
  31.  
  32.   索引区的IP和它指向的记录区一条记录中的IP构成一个IP范围。查询信息是这个
  33.   范围内IP的信息
  34.  
  35. '''
  36.  
  37. import sys
  38. import socket
  39. from struct import pack, unpack
  40.  
  41. class IPInfo(object):
  42.   '''QQWry.Dat数据库查询功能集合
  43.   '''
  44.   def __init__(self, dbname):
  45.     ''' 初始化类，读取数据库内容为一个字符串，
  46.     通过开始8字节确定数据库的索引信息'''
  47.  
  48.     self.dbname = dbname
  49.     # f = file(dbname, 'r')
  50.  
  51.     # Demon注：在Windows下用'r'会有问题，会把\r\n转换成\n
  52.     # 详见http://demon.tw/programming/python-open-mode.html
  53.     # 还有Python文档中不提倡用file函数来打开文件，推荐用open
  54.     f = open(dbname, 'rb')
  55.  
  56.     self.img = f.read()
  57.     f.close()
  58.  
  59.     # QQWry.Dat文件的开始8字节是索引信息,前4字节是开始索引的偏移值，
  60.     # 后4字节是结束索引的偏移值。
  61.     # (self.firstIndex, self.lastIndex) = unpack('II', self.img[:8])
  62.  
  63.     # Demon注：unpack默认使用的endian是和机器有关的
  64.     # Intel x86和AMD64(x86-64)是little-endian
  65.     # Motorola 68000和PowerPC G5是big-endian
  66.     # 而纯真数据库全部采用了little-endian字节序
  67.     # 所以在某些big-endian的机器上原代码会出错
  68.     (self.firstIndex, self.lastIndex) = unpack('<ii', self.img[:8])="" #="" 每条索引长7字节，这里得到索引总个数="" self.indexcount="(self.lastIndex" -="" self.firstindex)="" 7="" +="" 1="" def="" getstring(self,="" offset="0):" '''="" 读取字符串信息，包括"国家"信息和"地区"信息="" qqwry.dat的记录区每条信息都是一个以'\0'结尾的字符串'''="" o2="self.img.find('\0'," offset)="" #return="" self.img[offset:o2]="" 有可能只有国家信息没有地区信息，="" gb2312_str="self.img[offset:o2]" try:="" utf8_str="unicode(gb2312_str,'gb2312').encode('utf-8')" except:="" return="" '未知'="" getlong3(self,="" '''qqwry.dat中的偏移记录都是3字节，本函数取得3字节的偏移量的常规表示="" qqwry.dat使用“字符串“存储这些值'''="" s="self.img[offset:" 3]="" unpack用一个'i'作为format，后面的字符串必须是4字节="" unpack('i',="" s)[0]="" demon注：和上面一样，强制使用little-endian="" unpack('<i',="" getareaaddr(self,="" 通过给出偏移值，取得区域信息字符串，'''="" byte="ord(self.img[offset])" if="" or="" 2:="" 第一个字节为1或者2时，取得2-4字节作为一个偏移量调用自己="" p="self.getLong3(offset" 1)="" self.getareaaddr(p)="" else:="" self.getstring(offset)="" getaddr(self,="" offset,="" ip="0):" img="self.img" o="offset" 1:="" 重定向模式1="" [ip][0x01][国家和地区信息的绝对偏移地址]="" 使用接下来的3字节作为偏移量调用字节取得信息="" self.getaddr(self.getlong3(o="" 1))="" 重定向模式2="" [ip][0x02][国家信息的绝对偏移][地区信息字符串]="" 使用国家信息偏移量调用自己取得字符串信息="" carea="self.getAreaAddr(self.getLong3(o" 跳过前4字节取字符串作为地区信息="" aarea="self.getAreaAddr(o)" (carea,="" aarea)="" !="1" and="" 最简单的ip记录形式，[ip][国家信息][地区信息]="" 重定向模式1有种情况就是偏移量指向包含国家和地区信息两个字符串="" 即偏移量指向的第一个字节不是1或2,就使用这里的分支="" 简单地说：取连续取两个字符串！="" #o="" 我们已经修改carea为utf-8字符编码了，len取得的长度会有变，="" 用下面方法得到offset="" "?":="" "信":="" "[":="" find(self,="" ip,="" l,="" r):="" 使用二分法查找网络字节编码的ip地址的索引记录'''="" r="" l="" <="1:" m="(l" r)="" 2="" *="" #new_ip="unpack('I'," self.img[o:="" o+4])[0]="" new_ip="unpack('<I'," self.find(ip,="" m)="" m,="" getipaddr(self,="" ip):="" 调用其他函数，取得信息！'''="" 使用网络字节编码ip地址="" socket.inet_aton(ip))[0]="" 使用="" self.find="" 函数查找ip的索引偏移="" i="self.find(ip," 0,="" 得到索引记录="" 索引记录格式是：="" 前4字节ip信息+3字节指向ip记录信息的偏移量="" 这里就是使用后3字节作为偏移量得到其常规表示（qqwry.dat用字符串表示值）="" 4)="" ip记录偏移值+4可以丢弃前4字节的ip地址信息。="" (c,="" a)="self.getAddr(o2" output(self,="" first,="" last):="" for="" in="" range(first,="" self.img[o:o+4])[0]))="" print="" "%s="" %d="" %s="" %s"="" %="" (ip,="" c,="" getip(ip):="" import="" os="" _localdir="os.path.dirname(__file__)" _curpath="os.path.normpath(os.path.join(os.getcwd(),_localDir))" curpath="_curpath" c+a="" main():="" os.path.exists(sys.argv[1]):="" line="" open(sys.argv[1],"r").readlines():="" demon注：如果是在windows命令行中运行把编码转回gb2312以避免乱码="" sys.platform="=" 'win32':="" c="unicode(c," 'utf-8').encode('gb2312')="" a="unicode(a," '%s="" %s'="" (line,="" (sys.argv[1],="" __name__="=" '__main__':="" main()<="" script=""></ii',>

用Python脚本查询纯真IP库QQWry.dat（Demon修改版）

由于要用 Python 读取一个和纯真IP数据库 QQWry.dat 格式差不多的 IPv6 数据库，所以在网上搜索了一下，在 LinuxTOY 看到了一个 Python 脚本，发现有一些小小的问题，于是修改了一下。

  1. #!/usr/bin/env python
  2. # coding: utf-8
  3. # from: http://linuxtoy.org/files/pyip.py
  4. # Blog: http://linuxtoy.org/archives/python-ip.html
  5. # Modified by Demon
  6. # Blog: http://demon.tw/programming/python-qqwry-dat.html
  7. '''用Python脚本查询纯真IP库
  8. QQWry.Dat的格式如下:
  9. +----------+
  10. | 文件头 | (8字节)
  11. +----------+
  12. | 记录区 | （不定长）
  13. +----------+
  14. | 索引区 | （大小由文件头决定）
  15. +----------+
  16. 文件头：4字节开始索引偏移值+4字节结尾索引偏移值
  17. 记录区： 每条IP记录格式 ==> IP地址[国家信息][地区信息]
  18.   对于国家记录，可以有三种表示方式：
  19.     字符串形式(IP记录第5字节不等于0x01和0x02的情况)，
  20.     重定向模式1(第5字节为0x01),则接下来3字节为国家信息存储地的偏移值
  21.     重定向模式(第5字节为0x02),
  22.   
  23.   对于地区记录，可以有两种表示方式： 字符串形式和重定向
  24.   最后一条规则：重定向模式1的国家记录后不能跟地区记录
  25. 索引区： 每条索引记录格式 ==> 4字节起始IP地址 + 3字节指向IP记录的偏移值
  26.   索引区的IP和它指向的记录区一条记录中的IP构成一个IP范围。查询信息是这个
  27.   范围内IP的信息
  28. '''
  29. import sys
  30. import socket
  31. from struct import pack, unpack
  32. class IPInfo(object):
  33.   '''QQWry.Dat数据库查询功能集合
  34.   '''
  35.   def __init__(self, dbname):
  36.     ''' 初始化类，读取数据库内容为一个字符串，
  37.     通过开始8字节确定数据库的索引信息'''
  38.     
  39.     self.dbname = dbname
  40.     # f = file(dbname, 'r')
  41.     # Demon注：在Windows下用'r'会有问题，会把\r\n转换成\n
  42.     # 详见http://demon.tw/programming/python-open-mode.html
  43.     # 还有Python文档中不提倡用file函数来打开文件，推荐用open
  44.     f = open(dbname, 'rb')
  45.     self.img = f.read()
  46.     f.close()
  47.     # QQWry.Dat文件的开始8字节是索引信息,前4字节是开始索引的偏移值，
  48.     # 后4字节是结束索引的偏移值。
  49.     # (self.firstIndex, self.lastIndex) = unpack('II', self.img[:8])
  50.     # Demon注：unpack默认使用的endian是和机器有关的
  51.     # Intel x86和AMD64(x86-64)是little-endian
  52.     # Motorola 68000和PowerPC G5是big-endian
  53.     # 而纯真数据库全部采用了little-endian字节序
  54.     # 所以在某些big-endian的机器上原代码会出错
  55.     (self.firstIndex, self.lastIndex) = unpack('<ii', self.img[:8])="" #="" 每条索引长7字节，这里得到索引总个数="" self.indexcount="(self.lastIndex" -="" self.firstindex)="" 7="" +="" 1="" def="" getstring(self,="" offset="0):" '''="" 读取字符串信息，包括"国家"信息和"地区"信息="" qqwry.dat的记录区每条信息都是一个以'\0'结尾的字符串'''="" o2="self.img.find('\0'," offset)="" #return="" self.img[offset:o2]="" 有可能只有国家信息没有地区信息，="" gb2312_str="self.img[offset:o2]" try:="" utf8_str="unicode(gb2312_str,'gb2312').encode('utf-8')" except:="" return="" '未知'="" getlong3(self,="" '''qqwry.dat中的偏移记录都是3字节，本函数取得3字节的偏移量的常规表示="" qqwry.dat使用“字符串“存储这些值'''="" s="self.img[offset:" 3]="" unpack用一个'i'作为format，后面的字符串必须是4字节="" unpack('i',="" s)[0]="" demon注：和上面一样，强制使用little-endian="" unpack('<i',="" getareaaddr(self,="" 通过给出偏移值，取得区域信息字符串，'''="" byte="ord(self.img[offset])" if="" or="" 2:="" 第一个字节为1或者2时，取得2-4字节作为一个偏移量调用自己="" p="self.getLong3(offset" 1)="" self.getareaaddr(p)="" else:="" self.getstring(offset)="" getaddr(self,="" offset,="" ip="0):" img="self.img" o="offset" 1:="" 重定向模式1="" [ip][0x01][国家和地区信息的绝对偏移地址]="" 使用接下来的3字节作为偏移量调用字节取得信息="" self.getaddr(self.getlong3(o="" 1))="" 重定向模式2="" [ip][0x02][国家信息的绝对偏移][地区信息字符串]="" 使用国家信息偏移量调用自己取得字符串信息="" carea="self.getAreaAddr(self.getLong3(o" 跳过前4字节取字符串作为地区信息="" aarea="self.getAreaAddr(o)" (carea,="" aarea)="" !="1" and="" 最简单的ip记录形式，[ip][国家信息][地区信息]="" 重定向模式1有种情况就是偏移量指向包含国家和地区信息两个字符串="" 即偏移量指向的第一个字节不是1或2,就使用这里的分支="" 简单地说：取连续取两个字符串！="" #o="" 我们已经修改carea为utf-8字符编码了，len取得的长度会有变，="" 用下面方法得到offset="" find(self,="" ip,="" l,="" r):="" 使用二分法查找网络字节编码的ip地址的索引记录'''="" r="" l="" <="1:" m="(l" r)="" 2="" *="" #new_ip="unpack('I'," self.img[o:="" o+4])[0]="" new_ip="unpack('<I'," self.find(ip,="" m)="" m,="" getipaddr(self,="" ip):="" 调用其他函数，取得信息！'''="" 使用网络字节编码ip地址="" socket.inet_aton(ip))[0]="" 使用="" self.find="" 函数查找ip的索引偏移="" i="self.find(ip," 0,="" 得到索引记录="" 索引记录格式是：="" 前4字节ip信息+3字节指向ip记录信息的偏移量="" 这里就是使用后3字节作为偏移量得到其常规表示（qqwry.dat用字符串表示值）="" 4)="" ip记录偏移值+4可以丢弃前4字节的ip地址信息。="" (c,="" a)="self.getAddr(o2" output(self,="" first,="" last):="" for="" in="" range(first,="" self.img[o:o+4])[0]))="" print="" "%s="" %d="" %s="" %s"="" %="" (ip,="" c,="" main():="" demon注：如果是在windows命令行中运行把编码转回gb2312以避免乱码="" sys.platform="=" 'win32':="" c="unicode(c," 'utf-8').encode('gb2312')="" a="unicode(a," '%s="" %s'="" (sys.argv[1],="" __name__="=" '__main__':="" main()="" changelog="" 时间：2009年5月29日="" 1.="" 工具下面网友的建议，修改"o="" 1"="" http:="" linuxtoy.org="" archives="" python-ip.html#comment-113960="" 因为这个时候我已经把得到的字符串变成utf-8编码了，长度会有变化！<="" script=""></ii',>

人气教程排行

本站所有资源全部来源于网络，若本站发布的内容侵害到您的隐私或者利益，请联系我们删除！