make readpreference work again

KarrLab · Sep 4, 2019 · b0b5ac3 · b0b5ac3
1 parent 4974329
commit b0b5ac3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/datanator_query_python/util/mongo_util.py b/datanator_query_python/util/mongo_util.py
@@ -8,12 +8,12 @@ class MongoUtil:
 
     def __init__(self, cache_dirname=None, MongoDB=None, replicaSet=None, db='test',
                  verbose=False, max_entries=float('inf'), username = None, 
-                 password = None, authSource = 'admin'):
+                 password = None, authSource = 'admin', readPreference='nearest'):
         self.client = pymongo.MongoClient(
             host=MongoDB, replicaSet=replicaSet, 
             username = username, password = password,
-            authSource = authSource)  
-        self.db_obj = self.client[db]
+            authSource = authSource, readPreference=readPreference)  
+        self.db_obj = self.client.get_database(db)
 
     def list_all_collections(self):
         '''List all non-system collections within database

diff --git a/tests/util/test_mongo_util.py b/tests/util/test_mongo_util.py
@@ -30,6 +30,7 @@ def tearDownClass(cls):
     # @unittest.skip('passed')
     def test_list_all_collections(self):
         self.assertTrue('ecmdb' in self.src.list_all_collections())
+        print(self.src.client.read_preference)
 
 
     # @unittest.skip('passed')