corrected bad import preventing to run on PyMongo 2.X versions

This commit is contained in:
mrigal 2015-04-11 11:08:26 +02:00 committed by Matthieu Rigal
parent a5c2fc4f9d
commit 6ad9a56bd9

View File

@ -13,7 +13,6 @@ from bson import json_util
import pymongo import pymongo
import pymongo.errors import pymongo.errors
from pymongo.common import validate_read_preference from pymongo.common import validate_read_preference
from pymongo.collection import ReturnDocument
from mongoengine import signals from mongoengine import signals
from mongoengine.connection import get_db from mongoengine.connection import get_db
@ -27,6 +26,9 @@ from mongoengine.queryset import transform
from mongoengine.queryset.field_list import QueryFieldList from mongoengine.queryset.field_list import QueryFieldList
from mongoengine.queryset.visitor import Q, QNode from mongoengine.queryset.visitor import Q, QNode
if IS_PYMONGO_3:
from pymongo.collection import ReturnDocument
__all__ = ('BaseQuerySet', 'DO_NOTHING', 'NULLIFY', 'CASCADE', 'DENY', 'PULL') __all__ = ('BaseQuerySet', 'DO_NOTHING', 'NULLIFY', 'CASCADE', 'DENY', 'PULL')