diff --git a/mongoengine/pymongo_support.py b/mongoengine/pymongo_support.py index 38332c13..d9c5ee27 100644 --- a/mongoengine/pymongo_support.py +++ b/mongoengine/pymongo_support.py @@ -29,7 +29,7 @@ def count_documents(collection, filter, skip=None, limit=None, hint=None, collat try: return collection.count_documents(filter=filter, **kwargs) - except (AttributeError, OperationFailure) as ex: + except (AttributeError, OperationFailure): # AttributeError - count_documents appeared in pymongo 3.7 # OperationFailure - accounts for some operators that used to work # with .count but are no longer working with count_documents (i.e $geoNear, $near, and $nearSphere) diff --git a/mongoengine/queryset/base.py b/mongoengine/queryset/base.py index 41b394b4..303271f5 100644 --- a/mongoengine/queryset/base.py +++ b/mongoengine/queryset/base.py @@ -12,7 +12,6 @@ import pymongo.errors from pymongo.collection import ReturnDocument from pymongo.common import validate_read_preference import six -from pymongo.errors import OperationFailure from six import iteritems from mongoengine import signals