support gzip

Author: dylan4224

ipdb/database.py

@@ -8,6 +8,7 @@
 import sys
 
 from .util import bytes2long
+from .util import read_file
 from .exceptions import NoSupportIPv4Error, NoSupportIPv6Error, NoSupportLanguageError, DatabaseError, IPNotFound
 
 
@@ -31,9 +32,8 @@ class Reader:
     _v4offset = 0
     _v6offsetCache = {}
 
-    def __init__(self, name):
-        file = open(name, "rb")
-        self.data = file.read()
+    def __init__(self, name, compression=None):
+        self.data = read_file(name, compression=compression)
         self._file_size = len(self.data)
 
         meta_length = bytes2long(self.data[0], self.data[1], self.data[2], self.data[3])

ipdb/util.py

@@ -3,7 +3,7 @@
     :copyright: ©2018 by IPIP.net
 """
 
-
+import gzip
 import sys
 
 
@@ -18,3 +18,23 @@ def convert(v):
         return v
     else:
         return ord(v)
+
+
+def read_file(path, compression=None):
+    if compression is None:
+        return _read_file_default(path)
+    elif compression == 'gzip':
+        return _read_file_gzip(path)
+    raise Exception('unsupported compression type: {}'.format(compression))
+
+
+def _read_file_default(path):
+    with open(path, "rb") as f:
+        content = f.read()
+    return content
+
+
+def _read_file_gzip(path):
+    with gzip.open(path, "rb") as f:
+        content = f.read()
+    return content