first adaptations after comments and find-outs

This commit is contained in:
mrigal
2015-04-09 10:57:19 +02:00
committed by Matthieu Rigal
parent 46817caa68
commit ccbd128fa2
4 changed files with 13 additions and 12 deletions

View File

@@ -8,8 +8,9 @@ __all__ = ['ConnectionError', 'connect', 'register_connection',
DEFAULT_CONNECTION_NAME = 'default'
if pymongo.version_tuple[0] >= 3:
READ_PREFERENCE = ReadPreference.SECONDARY_PREFERRED
READ_PREFERENCE = ReadPreference.PRIMARY
else:
from pymongo import MongoReplicaSetClient
READ_PREFERENCE = False
@@ -126,6 +127,8 @@ def get_connection(alias=DEFAULT_CONNECTION_NAME, reconnect=False):
if conn_settings == connection_settings and _connections.get(db_alias, None):
connection = _connections[db_alias]
break
if pymongo.version_tuple[0] < 3:
connection_class = MongoReplicaSetClient
_connections[alias] = connection if connection else connection_class(**conn_settings)
except Exception, e: