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

通过IP地址判断客户端是网通的还是电信的。

使用一个纯文本的IP纯真数据库，用Python写了一个小程序。

核心代码：
  1. #!/usr/bin/env python  
  2. #site www.jbxue.com  
  3. #   
  4. # -*- coding: utf-8 -*-   
  5.   
  6. from bisect import bisect   
  7.   
  8. _LIST1, _LIST2 = [], []   
  9. _INIT = False   
  10.   
  11. ip2int = lambda ip_str: reduce(lambda a, b: (a << 8) + b, [int(i) for i in ip_str.split('.')])   
  12.   
  13. def _init():   
  14. global _LIST, _INIT   
  15. if not _INIT:   
  16. for l in open('ipdata.txt', 'rb'):   
  17. ip1, ip2 = l.split()[:2]   
  18. addr = ' '.join(l.split()[2:])   
  19. ip1, ip2 = ip2int(ip1), ip2int(ip2)   
  20. _LIST1.append(ip1)   
  21. _LIST2.append((ip1, ip2, addr))   
  22. _INIT = True   
  23.   
  24.   
  25. def ip_from(ip):   
  26. _init()   
  27. i = ip2int(ip)   
  28. idx = bisect(_LIST1, i)   
  29. assert(idx > 0)   
  30. if len(_LIST1) <= idx:   
  31. return u'unknown ip address %s' % ip   
  32. else:   
  33. frm, to ,addr = _LIST2[idx - 1]   
  34. if frm <= i <= to:   
  35. return addr   
  36. else:   
  37. return u'unknown ip address %s' % ip   
  38.   
  39.   
  40. if __name__ == '__main__':   
  41. print ip_from('115.238.54.106')   
  42. print ip_from('220.181.29.160')   
  43. print ip_from('115.238.54.107')   
  44. print ip_from('8.8.8.8')  
posted @ 2014-01-27 21:42  un123  阅读(723)  评论(0)  编辑  收藏  举报