diff options
author | Simo Sorce <simo@redhat.com> | 2014-03-02 19:03:38 -0500 |
---|---|---|
committer | Simo Sorce <simo@redhat.com> | 2014-03-02 19:11:34 -0500 |
commit | e47edacd7eb7f4c90a244aed7313d07eaac08875 (patch) | |
tree | 82c1010e7455d91103368308de719c0a91998ab3 /ipsilon/providers/saml2/provider.py | |
parent | 2959e20b5607edab7313aa5ba4500c1f37358979 (diff) | |
download | ipsilon-e47edacd7eb7f4c90a244aed7313d07eaac08875.tar.gz ipsilon-e47edacd7eb7f4c90a244aed7313d07eaac08875.tar.xz ipsilon-e47edacd7eb7f4c90a244aed7313d07eaac08875.zip |
Improve exceptions for saml2 providers
Signed-off-by: Simo Sorce <simo@redhat.com>
Diffstat (limited to 'ipsilon/providers/saml2/provider.py')
-rwxr-xr-x | ipsilon/providers/saml2/provider.py | 16 |
1 files changed, 7 insertions, 9 deletions
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 <http://www.gnu.org/licenses/>. +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): |