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

read_preference from parse_uri only PYMONGO_3 #1665

Browse Source
This commit is contained in:
Erdenezul 2017-10-03 09:23:17 +08:00
parent 5c4ce8754e
commit 6e2db1ced6
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mongoengine/connection.py
View File

@ -104,14 +104,15 @@ def register_connection(alias, name=None, host=None, port=None,
if 'authmechanism' in uri_options: if 'authmechanism' in uri_options:
conn_settings['authentication_mechanism'] = uri_options['authmechanism'] conn_settings['authentication_mechanism'] = uri_options['authmechanism']
if IS_PYMONGO_3 and 'readpreference' in uri_options: if IS_PYMONGO_3 and 'readpreference' in uri_options:
read_preferences = (ReadPreference.NEAREST, read_preferences = (
ReadPreference.NEAREST,
ReadPreference.PRIMARY, ReadPreference.PRIMARY,
ReadPreference.PRIMARY_PREFERRED, ReadPreference.PRIMARY_PREFERRED,
ReadPreference.SECONDARY, ReadPreference.SECONDARY,
ReadPreference.SECONDARY_PREFERRED) ReadPreference.SECONDARY_PREFERRED)
read_pf_mode = uri_options['readpreference'] read_pf_mode = uri_options['readpreference'].lower()
for preference in read_preferences: for preference in read_preferences:
if preference.mode == read_pf_mode: if preference.name.lower() == read_pf_mode:
conn_settings['read_preference'] = preference conn_settings['read_preference'] = preference
break break
else: else: