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

Allow setting read_concern

Browse Source
This commit is contained in:
Agustin
2020-01-24 13:11:07 -03:00
parent b2c3acd025
commit 8e17e42e26
3 changed files with 82 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
mongoengine/context_managers.py
View File

@@ -257,3 +257,11 @@ def set_write_concern(collection, write_concerns):
combined_concerns = dict(collection.write_concern.document.items())
combined_concerns.update(write_concerns)
yield collection.with_options(write_concern=WriteConcern(**combined_concerns))
@contextmanager
def set_read_write_concern(collection, write_concerns, read_concern):
combined_write_concerns = dict(collection.write_concern.document.items())
combined_write_concerns.update(write_concerns)
yield collection.with_options(write_concern=WriteConcern(**combined_write_concerns), read_concern=read_concern)