From 727c19b9c06a7f7d44dcca36e7d429b1ae8d4df5 Mon Sep 17 00:00:00 2001 From: Matt Wright Date: Tue, 12 Jun 2012 12:11:19 -0400 Subject: [PATCH] Fix #13 --- flask_security/core.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/flask_security/core.py b/flask_security/core.py index f741ac1..9be9103 100644 --- a/flask_security/core.py +++ b/flask_security/core.py @@ -136,14 +136,10 @@ def roles_accepted(*roles): class RoleMixin(object): """Mixin for `Role` model definitions""" def __eq__(self, other): - if isinstance(other, basestring): - return self.name == other - return self.name == other.name + return self.name == getattr(other, 'name', None) def __ne__(self, other): - if isinstance(other, basestring): - return self.name != other - return self.name != other.name + return self.name != getattr(other, 'name', None) def __str__(self): return '' % self.name