wassname/flask-security
1
0
Fork 0
mirror of https://github.com/wassname/flask-security.git synced 2026-06-27 16:10:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Make find_user method for MongoEngineUserDatastore add contraints to query

Browse Source
This commit is contained in:
Matt Wright
2013-04-04 15:50:46 -04:00
parent 99ac732d10
commit 4815b1afed
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
flask_security/datastore.py
+4 -1
View File
@@ -191,7 +191,10 @@ class MongoEngineUserDatastore(MongoEngineDatastore, UserDatastore):
UserDatastore.__init__(self, user_model, role_model)
def find_user(self, **kwargs):
return self.user_model.objects(**kwargs).first()
from mongoengine.queryset 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()
def find_role(self, role):
return self.role_model.objects(name=role).first()