From e47edacd7eb7f4c90a244aed7313d07eaac08875 Mon Sep 17 00:00:00 2001 From: Simo Sorce Date: Sun, 2 Mar 2014 19:03:38 -0500 Subject: Improve exceptions for saml2 providers Signed-off-by: Simo Sorce --- ipsilon/providers/saml2/provider.py | 16 +++++++--------- 1 file changed, 7 insertions(+), 9 deletions(-) (limited to 'ipsilon/providers/saml2/provider.py') diff --git a/ipsilon/providers/saml2/provider.py b/ipsilon/providers/saml2/provider.py index acf2ee7..03efeb3 100755 --- a/ipsilon/providers/saml2/provider.py +++ b/ipsilon/providers/saml2/provider.py @@ -17,6 +17,7 @@ # You should have received a copy of the GNU General Public License # along with this program. If not, see . +from ipsilon.providers.common import ProviderException import cherrypy import lasso @@ -34,15 +35,12 @@ NAMEID_MAP = { } -class InvalidProviderId(Exception): +class InvalidProviderId(ProviderException): - def __init__(self, message): - msg = 'Invalid Provider ID: %s' % message - super(InvalidProviderId, self).__init__(msg) - self.message = msg - - def __str__(self): - return repr(self.message) + def __init__(self, code): + message = 'Invalid Provider ID: %s' % code + super(InvalidProviderId, self).__init__(message) + self._debug(message) class NameIdNotAllowed(Exception): @@ -101,7 +99,7 @@ class ServiceProvider(object): for nameid in allowed: if nip.format == NAMEID_MAP[nameid]: return nip.format - raise NameIdNotAllowed() + raise NameIdNotAllowed(nip.format) def _debug(self, fact): if cherrypy.config.get('debug', False): -- cgit