From 33a4a2ab86f05f059a641110e0a06a7829a0316b Mon Sep 17 00:00:00 2001 From: yuema137 <3124558229@qq.com> Date: Wed, 19 Jun 2024 20:58:40 -0500 Subject: [PATCH] add xent_collection_admin --- utilix/rundb.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/utilix/rundb.py b/utilix/rundb.py index 1c5dc73..0587180 100644 --- a/utilix/rundb.py +++ b/utilix/rundb.py @@ -525,6 +525,7 @@ def pymongo_collection(collection='runs', **kwargs): user = kwargs.get('user') pw = kwargs.get('password') database = kwargs.get('database') + read_preference = kwargs.get('read_preference', 'secondary') if not url: url = uconfig.get('RunDB', 'pymongo_url') @@ -535,7 +536,7 @@ def pymongo_collection(collection='runs', **kwargs): if not database: database = uconfig.get('RunDB', 'pymongo_database') uri = uri.format(user=user, pw=pw, url=url) - c = pymongo.MongoClient(uri, readPreference='secondary') + c = pymongo.MongoClient(uri, readPreference=read_preference) DB = c[database] coll = DB[collection] # Checkout the collection we are returning and raise errors if you want @@ -567,7 +568,7 @@ def _collection(experiment, collection, url=None, user=None, password=None, data connect_timeout = uconfig.get('RunDB', 'connect_timeout', fallback=60000) force_single_server = uconfig.get('RunDB', 'force_single_server', fallback=True) direct_connection = uconfig.get('RunDB', 'direct_connection', fallback=True) - read_preference = uconfig.get('RunDB', 'read_preference', fallback='secondaryPreferred') + read_preference = uconfig.get('RunDB', 'read_preference', fallback='secondary') # By default, use only the last server in the url if force_single_server: @@ -595,6 +596,11 @@ def _collection(experiment, collection, url=None, user=None, password=None, data def xent_collection(collection='runs', **kwargs): return _collection('xent', collection, **kwargs) +def xent_collection_admin(collection='runs', **kwargs): + # for admin purposes, the read preference should be primary + kwargs['read_preference'] = 'primary' + return _collection('xent', collection, **kwargs) + def xe1t_collection(collection='runs_new', **kwargs): return _collection('xe1t', collection, **kwargs)