wassname/flask-security
1
0
Fork 0
mirror of https://github.com/wassname/flask-security.git synced 2026-08-09 12:00:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #111 from joshpurvis/issue110

Browse Source 
Changed has_role to accept strings with mongoengine. Fixes #110
This commit is contained in:
Matt Wright
2013-04-14 10:05:34 -07:00
parent 38874433c7 3b81ec57ea
commit 6f3c163ee7
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
flask_security/core.py
+5 -3
View File
@@ -241,8 +241,7 @@ class RoleMixin(object):
self.name == getattr(other, 'name', None))
def __ne__(self, other):
return (self.name != other and
self.name != getattr(other, 'name', None))
return not self.__eq__(other)
class UserMixin(BaseUserMixin):
@@ -261,7 +260,10 @@ class UserMixin(BaseUserMixin):
"""Returns `True` if the user identifies with the specified role.
:param role: A role name or `Role` instance"""
return role in self.roles
if isinstance(role, basestring):
return role in (role.name for role in self.roles)
else:
return role in self.roles
class AnonymousUser(AnonymousUserBase):