Import check to account for new version of MongoEngine

This commit is contained in:
Matt Wright
2013-05-03 12:13:58 -04:00
parent 38874433c7
commit f2d5245bd8
+5 -1
View File
@@ -190,7 +190,11 @@ class MongoEngineUserDatastore(MongoEngineDatastore, UserDatastore):
UserDatastore.__init__(self, user_model, role_model)
def find_user(self, **kwargs):
from mongoengine.queryset import Q, QCombination
try:
from mongoengine.queryset import Q, QCombination
except ImportError:
from mongoengine.queryset.visitor import Q, QCombination
queries = map(lambda i: Q(**{i[0]: i[1]}), kwargs.items())
query = QCombination(QCombination.AND, queries)
return self.user_model.objects(query).first()