Add support for DB open modes

The tests won't pass in Travis until we merge the maxminddb changes.

Author: oschwald

geoip2/database.py

@@ -11,6 +11,9 @@
 import geoip2.errors
 import maxminddb
 
+from maxminddb import (MODE_AUTO, MODE_MMAP, MODE_MMAP_EXT, MODE_FILE,
+                       MODE_MEMORY)
+
 
 class Reader(object):
 
@@ -40,10 +43,10 @@ class Reader(object):
 
 """
 
-    def __init__(self, filename, locales=None):
+    def __init__(self, filename, locales=None, mode=MODE_AUTO):
         if locales is None:
             locales = ['en']
-        self._db_reader = maxminddb.Reader(filename)
+        self._db_reader = maxminddb.open_database(filename, mode)
         self._locales = locales
 
     def country(self, ip_address):

tests/database_test.py

@@ -7,6 +7,12 @@
 sys.path.append('..')
 
 import geoip2.database
+import maxminddb
+
+try:
+    import maxminddb.extension
+except ImportError:
+    maxminddb.extension = None
 
 if sys.version_info[:2] == (2, 6):
     import unittest2 as unittest
@@ -18,7 +24,7 @@
     unittest.TestCase.assertRegex = unittest.TestCase.assertRegexpMatches
 
 
-class TestReader(unittest.TestCase):
+class BaseTestReader(object):
 
     def test_default_locale(self):
         reader = geoip2.database.Reader(
@@ -142,3 +148,24 @@ def test_isp(self):
                          "ISP repr can be eval'd")
 
         reader.close()
+
+
+@unittest.skipUnless(maxminddb.extension, 'No C extension module found. Skipping tests')
+class TestExtensionReader(BaseTestReader, unittest.TestCase):
+    mode = geoip2.database.MODE_MMAP_EXT
+
+
+class TestMMAPReader(BaseTestReader, unittest.TestCase):
+    mode = geoip2.database.MODE_MMAP
+
+
+class TestFileReader(BaseTestReader, unittest.TestCase):
+    mode = geoip2.database.MODE_FILE
+
+
+class TestMemoryReader(BaseTestReader, unittest.TestCase):
+    mode = geoip2.database.MODE_MEMORY
+
+
+class TestMemoryReader(BaseTestReader, unittest.TestCase):
+    mode = geoip2.database.MODE_AUTO