From 24a9633edc33872d13c09d30a1062b1f83d15976 Mon Sep 17 00:00:00 2001 From: RussellLuo Date: Mon, 20 Apr 2015 16:03:25 +0800 Subject: [PATCH] Override `authentication_source` by "authSource" in URI --- mongoengine/connection.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/mongoengine/connection.py b/mongoengine/connection.py index dcecdd9a..5e18efb7 100644 --- a/mongoengine/connection.py +++ b/mongoengine/connection.py @@ -1,4 +1,3 @@ -import pymongo from pymongo import MongoClient, MongoReplicaSetClient, uri_parser @@ -58,8 +57,11 @@ def register_connection(alias, name=None, host=None, port=None, 'password': uri_dict.get('password'), 'read_preference': read_preference, }) - if "replicaSet" in conn_settings['host']: + uri_options = uri_dict['options'] + if 'replicaset' in uri_options: conn_settings['replicaSet'] = True + if 'authsource' in uri_options: + conn_settings['authentication_source'] = uri_options['authsource'] # Deprecated parameters that should not be passed on kwargs.pop('slaves', None)