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

Refactor set_read_write_concern so read_conern is consistent with the write_concerns argument.

Browse Source
This commit is contained in:
Agustin Barto 2020-03-17 19:08:43 -03:00
parent a3f9016ae9
commit 3d80637fa4
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mongoengine/context_managers.py
View File

@ -261,10 +261,14 @@ def set_write_concern(collection, write_concerns):
@contextmanager @contextmanager
def set_read_write_concern(collection, write_concerns, read_concern): def set_read_write_concern(collection, write_concerns, read_concerns):
combined_write_concerns = dict(collection.write_concern.document.items()) combined_write_concerns = dict(collection.write_concern.document.items())
combined_write_concerns.update(write_concerns) combined_write_concerns.update(write_concerns)
combined_read_concerns = dict(collection.read_concern.document.items())
combined_read_concerns.update(read_concerns)
yield collection.with_options( yield collection.with_options(
write_concern=WriteConcern(**combined_write_concerns), read_concern=read_concern write_concern=WriteConcern(**combined_write_concerns),
read_concern=ReadConcern(**combined_read_concerns)
) )