Merge pull request #1411 from p-l-/fix-mongodb-conn

DB/Mongo: fix connection parameters
ivre · Jul 29, 2022 · c93d4e6 · c93d4e6
2 parents 7ddfc32 + 2e8f66c
commit c93d4e6
Showing 1 changed file with 16 additions and 4 deletions.
diff --git a/ivre/db/mongo.py b/ivre/db/mongo.py
@@ -138,6 +138,21 @@ def __init__(self, url):
             self.maxtime = int(params.pop("maxtime", None))
         except TypeError:
             self.maxtime = None
+        if "ssl" in params:
+            if params["ssl"].lower() == "true":
+                params["ssl"] = True
+            elif params["ssl"].lower() == "false":
+                params["ssl"] = False
+        if "readPreference" in params:
+            params["read_preference"] = {
+                "nearest": pymongo.ReadPreference.NEAREST,
+                "primary": pymongo.ReadPreference.PRIMARY,
+                "primarypreferred": pymongo.ReadPreference.PRIMARY_PREFERRED,
+                "primary_preferred": pymongo.ReadPreference.PRIMARY_PREFERRED,
+                "secondary": pymongo.ReadPreference.SECONDARY,
+                "secondarypreferred": pymongo.ReadPreference.SECONDARY_PREFERRED,
+                "secondary_preferred": pymongo.ReadPreference.SECONDARY_PREFERRED,
+            }.get(params["readPreference"].lower(), params["readPreference"])
         self.params = params
         self.schema_migrations = []
 
@@ -167,10 +182,7 @@ def db_client(self):
             if self.mongodb_srv is not None:
                 self._db_client = pymongo.MongoClient(self.mongodb_srv)
             else:
-                self._db_client = pymongo.MongoClient(
-                    host=self.host,
-                    read_preference=pymongo.ReadPreference.SECONDARY_PREFERRED,
-                )
+                self._db_client = pymongo.MongoClient(host=self.host, **self.params)
             return self._db_client
 
     @property