From 0141b240bc6206abbfc15e4c381e2d42d4caa1e2 Mon Sep 17 00:00:00 2001 From: Matt Wright Date: Tue, 12 Jun 2012 12:32:59 -0400 Subject: [PATCH] Better... --- flask_security/core.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/flask_security/core.py b/flask_security/core.py index 9be9103..feddbe1 100644 --- a/flask_security/core.py +++ b/flask_security/core.py @@ -136,10 +136,10 @@ def roles_accepted(*roles): class RoleMixin(object): """Mixin for `Role` model definitions""" def __eq__(self, other): - return self.name == getattr(other, 'name', None) + return self.name == other or self.name == getattr(other, 'name', None) def __ne__(self, other): - return self.name != getattr(other, 'name', None) + return self.name != other and self.name != getattr(other, 'name', None) def __str__(self): return '' % self.name