From 3d80637fa414354b2d05b0ca6ac3afb71222d8f1 Mon Sep 17 00:00:00 2001 From: Agustin Barto Date: Tue, 17 Mar 2020 19:08:43 -0300 Subject: [PATCH] Refactor set_read_write_concern so read_conern is consistent with the write_concerns argument. --- mongoengine/context_managers.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/mongoengine/context_managers.py b/mongoengine/context_managers.py index 6891e21d..a87c9bb3 100644 --- a/mongoengine/context_managers.py +++ b/mongoengine/context_managers.py @@ -261,10 +261,14 @@ def set_write_concern(collection, write_concerns): @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.update(write_concerns) + combined_read_concerns = dict(collection.read_concern.document.items()) + combined_read_concerns.update(read_concerns) + 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) )