Move some exceptions into provider.common

[cascardo/ipsilon.git] / ipsilon / providers / common.py

diff --git a/ipsilon/providers/common.py b/ipsilon/providers/common.py
index 94f747a..6d454b5 100755 (executable)
--- a/ipsilon/providers/common.py
+++ b/ipsilon/providers/common.py
@@ -34,6 +34,21 @@ class ProviderException(Exception, Log):
         return repr(self.message)
 
 
+class AuthenticationError(ProviderException):
+
+    def __init__(self, message, code):
+        super(AuthenticationError, self).__init__(message)
+        self.code = code
+        self._debug('%s [%s]' % (message, code))
+
+
+class InvalidRequest(ProviderException):
+
+    def __init__(self, message):
+        super(InvalidRequest, self).__init__(message)
+        self._debug(message)
+
+
 class ProviderBase(PluginObject):
 
     def __init__(self, name, path):
@@ -41,7 +56,11 @@ class ProviderBase(PluginObject):
         self.name = name
         self.path = path
         self.tree = None
-        self.admin = None
+
+    def on_enable(self):
+        # this one does nothing
+        # derived classes can override with custom behavior
+        return
 
     def get_tree(self, site):
         raise NotImplementedError
@@ -83,6 +102,8 @@ class ProviderBase(PluginObject):
 
         self.set_config_value('enabled', '1')
         self.save_plugin_config(FACILITY)
+
+        self.on_enable()
         self._debug('IdP Provider enabled: %s' % self.name)
 
     def disable(self, site):
@@ -112,7 +133,13 @@ class ProviderPageBase(Page):
         raise cherrypy.HTTPError(501)
 
     def root(self, *args, **kwargs):
-        op = getattr(self, cherrypy.request.method, self.GET)
+        method = cherrypy.request.method
+
+        preop = getattr(self, 'pre_%s' % method, None)
+        if preop and callable(preop):
+            preop(*args, **kwargs)
+
+        op = getattr(self, method, self.GET)
         if callable(op):
             return op(*args, **kwargs)
         else: