Commit b3a80ece by Patrick Smith

Add support for non-gzipped .dat files when in MEMORY_CACHE mode.

parent 536fbc2b
......@@ -97,8 +97,12 @@ class GeoIP(GeoIPBase):
self._filehandle = mmap.mmap(f.fileno(), 0, access=mmap.ACCESS_READ)
elif self._flags & const.MEMORY_CACHE:
self._filehandle = gzip.open(filename, 'rb')
self._memoryBuffer = self._filehandle.read()
try:
self._filehandle = gzip.open(filename, 'rb')
self._memoryBuffer = self._filehandle.read()
except IOError:
self._filehandle = codecs.open(filename, 'rb', 'latin_1')
self._memoryBuffer = self._filehandle.read()
else:
self._filehandle = codecs.open(filename, 'rb','latin_1')
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment