pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #2131 from NMFR/master

Browse Source 
Set default database when using mongomock.
This commit is contained in:
Bastien Gérard
2019-07-22 11:56:56 +02:00
committed by GitHub
parent ac416aeeb3 f4a06ad65d
commit 0d01365751
2 changed files with 33 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
mongoengine/connection.py
View File

@@ -103,7 +103,14 @@ def _get_connection_settings(
if entity.startswith("mongomock://"):
conn_settings["is_mock"] = True
# `mongomock://` is not a valid url prefix and must be replaced by `mongodb://`
resolved_hosts.append(entity.replace("mongomock://", "mongodb://", 1))
new_entity = entity.replace("mongomock://", "mongodb://", 1)
resolved_hosts.append(new_entity)
uri_dict = uri_parser.parse_uri(new_entity)
database = uri_dict.get("database")
if database:
conn_settings["name"] = database
# Handle URI style connections, only updating connection params which
# were explicitly specified in the URI.
@@ -111,8 +118,9 @@ def _get_connection_settings(
uri_dict = uri_parser.parse_uri(entity)
resolved_hosts.append(entity)
if uri_dict.get("database"):
conn_settings["name"] = uri_dict.get("database")
database = uri_dict.get("database")
if database:
conn_settings["name"] = database
for param in ("read_preference", "username", "password"):
if uri_dict.get(param):