Python下纯真IP库的乱码问题

网上有某位高人写的, 用python去读纯真IP库. 

从库里读出的时候, 有写乱码的情况. 读出的编码是Gb2312的. 

在上面作了一些修改, 主要是把gb2312转为utf-8:

1. #! /usr/bin/env python
2. # -*- coding: utf-8 -*-
4. """ IPLocator: locate IP in the QQWry.dat.
5.     Usage:
6.         python IPLocator.py <ip>
7.     Create and test with Python 2.2.3.
8.     spadger@bmy <echo.xjtu@gmail.com> 2008-2-19
9. """
11. import socket,string,struct,sys
13. class IPLocator :
14.     def __init__( self, ipdbFile ):
15.         self.ipdb = open( ipdbFile, "rb" )
16.         str = self.ipdb.read( 8 )
17.         (self.firstIndex,self.lastIndex) = struct.unpack('II',str)
18.         self.indexCount = (self.lastIndex - self.firstIndex)/7+1
19.         print self.getVersion()," 纪录总数: %d 条 "%(self.indexCount)
21.     def getVersion(self):
22.         s = self.getIpAddr(0xffffff00L)
23.         return s
25.     def getAreaAddr(self,offset=0):
26.         if offset :
27.             self.ipdb.seek( offset )
28.         str = self.ipdb.read( 1 )
29.         (byte,) = struct.unpack('B',str)
30.         if byte == 0x01 or byte == 0x02:
31.             p = self.getLong3()
32.             if p:
33.                 return self.getString( p )
34.             else:
35.                 return ""
36.         else:
37.             self.ipdb.seek(-1,1)
38.             return self.getString( offset )
40.     def getAddr(self,offset,ip=0):
41.         self.ipdb.seek( offset + 4)
42.         countryAddr = ""
43.         areaAddr = ""
44.         str = self.ipdb.read( 1 )
45.         (byte,) = struct.unpack('B',str)
46.         if byte == 0x01:
47.             countryOffset = self.getLong3()
48.             self.ipdb.seek( countryOffset )
49.             str = self.ipdb.read( 1 )
50.             (b,) = struct.unpack('B',str)
51.             if b == 0x02:
52.                 countryAddr = self.getString( self.getLong3() )
53.                 self.ipdb.seek( countryOffset + 4 )
54.             else:
55.                 countryAddr = self.getString( countryOffset )
56.             areaAddr = self.getAreaAddr()
57.         elif byte == 0x02:
58.             countryAddr = self.getString( self.getLong3() )
59.             areaAddr = self.getAreaAddr( offset + 8 )
60.         else:
61.             countryAddr = self.getString( offset + 4 )
62.             areaAddr = self.getAreaAddr()
63.         return countryAddr + " " + areaAddr
65.     def dump(self, first ,last ):
66.         if last > self.indexCount :
67.             last = self.indexCount
68.         for index in range(first,last):
69.             offset = self.firstIndex + index * 7
70.             self.ipdb.seek( offset )
71.             buf = self.ipdb.read( 7 )
72.             (ip,of1,of2) = struct.unpack("IHB",buf)
73.             print "%d/t%s/t%s" %(index, self.ip2str(ip), /
74.                 self.getAddr( of1 + (of2 << 16) ) )
76.     def setIpRange(self,index):
77.         offset = self.firstIndex + index * 7
78.         self.ipdb.seek( offset )
79.         buf = self.ipdb.read( 7 )
80.         (self.curStartIp,of1,of2) = struct.unpack("IHB",buf)
81.         self.curEndIpOffset = of1 + (of2 << 16)
82.         self.ipdb.seek( self.curEndIpOffset )
83.         buf = self.ipdb.read( 4 )
84.         (self.curEndIp,) = struct.unpack("I",buf)
86.     def getIpAddr(self,ip):
87.         L = 0
88.         R = self.indexCount - 1
89.         while L < R-1:
90.             M = (L + R) / 2
91.             self.setIpRange(M)
92.             if ip == self.curStartIp:
93.                 L = M
94.                 break
95.             if ip > self.curStartIp:
96.                 L = M
97.             else:
98.                 R = M
99.         self.setIpRange( L )
100.         #version information,255.255.255.X,urgy but useful
101.         if ip&0xffffff00L == 0xffffff00L:
102.             self.setIpRange( R )
103.         if self.curStartIp <= ip <= self.curEndIp:
104.             address = self.getAddr( self.curEndIpOffset )
105.             #把GB2312转为utf-8
106.             address = unicode(address,'gb2312').encode("utf-8")
107.         else:
108.             address = "未找到该IP的地址"
109.         return address
111.     def getIpRange(self,ip):
112.         self.getIpAddr(ip)
113.         range = self.ip2str(self.curStartIp) + ' - ' /
114.             + self.ip2str(self.curEndIp)
115.         return range
117.     def getString(self,offset = 0):
118.         if offset :
119.             self.ipdb.seek( offset )
120.         str = ""
121.         ch = self.ipdb.read( 1 )
122.         (byte,) = struct.unpack('B',ch)
123.         while byte != 0:
124.             str = str + ch
125.             ch = self.ipdb.read( 1 )
126.             (byte,) = struct.unpack('B',ch)
127.         return str
129.     def ip2str(self,ip):
130.         return str(ip>>24)+'.'+str((ip>>16)&0xffL)+'.' /
131.             +str((ip>>8)&0xffL)+'.'+str(ip&0xffL)
133.     def str2ip(self,s):
134.         (ip,) = struct.unpack('L',socket.inet_aton(s))
135.         return ((ip>>24)&0xffL)|((ip&0xffL)<<24) /
136.             |((ip>>8)&0xff00L)|((ip&0xff00L)<<8)
138.     def getLong3(self,offset = 0):
139.         if offset :
140.             self.ipdb.seek( offset )
141.         str = self.ipdb.read(3)
142.         (a,b) = struct.unpack('HB',str)
143.         return (b << 16) + a
145. #Demo
146. def main():
147.     IPL = IPLocator( "QQWry.dat" )
148.     ip = ""
149.     if len(sys.argv) != 2:
150.         print 'Usage: python IPLocator.py <IP>'
151.         return
152.     else:
153.         ip = sys.argv[1]
154.     address = IPL.getIpAddr( IPL.str2ip(ip) )
155.     range = IPL.getIpRange( IPL.str2ip(ip) )
156.     print "此IP %s 属于 %s/n所在网段: %s" % (ip,address, range)
159. if __name__ == "__main__" :
160.     main()