diff --git a/mongoengine/context_managers.py b/mongoengine/context_managers.py index 0c58b57c..6891e21d 100644 --- a/mongoengine/context_managers.py +++ b/mongoengine/context_managers.py @@ -14,6 +14,7 @@ __all__ = ( "no_sub_classes", "query_counter", "set_write_concern", + "set_read_write_concern", ) diff --git a/mongoengine/queryset/base.py b/mongoengine/queryset/base.py index 0788f563..d8a4b96f 100644 --- a/mongoengine/queryset/base.py +++ b/mongoengine/queryset/base.py @@ -20,8 +20,8 @@ from mongoengine.base import get_document from mongoengine.common import _import_class from mongoengine.connection import get_db from mongoengine.context_managers import ( - set_write_concern, set_read_write_concern, + set_write_concern, switch_db, ) from mongoengine.errors import (