From 0805de506b414978517d76424d44c511110e551b Mon Sep 17 00:00:00 2001 From: Krzysztof Szularz Date: Fri, 24 Jun 2016 17:27:23 +0200 Subject: [PATCH] Delete port from connection settings only if on pymongo 2 Right now the port is dropped from connection settings even if `MongoReplicaSetClient` is not being used. --- mongoengine/connection.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mongoengine/connection.py b/mongoengine/connection.py index 4055a9b6e..be2ddd65a 100644 --- a/mongoengine/connection.py +++ b/mongoengine/connection.py @@ -128,12 +128,12 @@ def get_connection(alias=DEFAULT_CONNECTION_NAME, reconnect=False): connection_class = MongoClient if 'replicaSet' in conn_settings: - # Discard port since it can't be used on MongoReplicaSetClient - conn_settings.pop('port', None) # Discard replicaSet if not base string if not isinstance(conn_settings['replicaSet'], basestring): conn_settings.pop('replicaSet', None) if not IS_PYMONGO_3: + # Discard port since it can't be used on MongoReplicaSetClient + conn_settings.pop('port', None) connection_class = MongoReplicaSetClient conn_settings['hosts_or_uri'] = conn_settings.pop('host', None)