ipsilon/admin/login.py

   1 #!/usr/bin/python
   2 #
   3 # Copyright (C) 2014  Simo Sorce <simo@redhat.com>
   4 #
   5 # see file 'COPYING' for use and warranty information
   6 #
   7 # This program is free software; you can redistribute it and/or modify
   8 # it under the terms of the GNU General Public License as published by
   9 # the Free Software Foundation, either version 3 of the License, or
  10 # (at your option) any later version.
  11 #
  12 # This program is distributed in the hope that it will be useful,
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 # GNU General Public License for more details.
  16 #
  17 # You should have received a copy of the GNU General Public License
  18 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19
  20 import cherrypy
  21 from ipsilon.util.page import Page
  22 from ipsilon.util.page import admin_protect
  23 from ipsilon.util.plugin import PluginObject
  24 from ipsilon.admin.common import AdminPluginPage
  25 from ipsilon.login.common import FACILITY
  26
  27
  28 class LoginPluginsOrder(Page):
  29
  30     def __init__(self, site, parent):
  31         super(LoginPluginsOrder, self).__init__(site)
  32         self.url = '%s/order' % parent.url
  33         self.menu = [parent]
  34
  35     @admin_protect
  36     def GET(self, *args, **kwargs):
  37         return self._template('admin/login_order.html',
  38                               title='login plugins order',
  39                               name='admin_login_order_form',
  40                               menu=self.menu, action=self.url,
  41                               options=self._site[FACILITY]['enabled'])
  42
  43     @admin_protect
  44     def POST(self, *args, **kwargs):
  45         message = "Nothing was modified."
  46         message_type = "info"
  47         valid = self._site[FACILITY]['enabled']
  48
  49         if 'order' in kwargs:
  50             order = kwargs['order'].split(',')
  51             if len(order) != 0:
  52                 new_values = []
  53                 try:
  54                     for v in order:
  55                         val = v.strip()
  56                         if val not in valid:
  57                             error = "Invalid plugin name: %s" % val
  58                             raise ValueError(error)
  59                         new_values.append(val)
  60                     if len(new_values) < len(valid):
  61                         for val in valid:
  62                             if val not in new_values:
  63                                 new_values.append(val)
  64
  65                     po = PluginObject()
  66                     po.name = "global"
  67                     globalconf = dict()
  68                     globalconf['order'] = ','.join(new_values)
  69                     po.set_config(globalconf)
  70                     po.save_plugin_config(FACILITY)
  71
  72                     # When all is saved update also live config
  73                     self._site[FACILITY]['enabled'] = new_values
  74
  75                     message = "New configuration saved."
  76                     message_type = "success"
  77
  78                 except ValueError, e:
  79                     message = str(e)
  80                     message_type = "error"
  81
  82                 except Exception, e:  # pylint: disable=broad-except
  83                     message = "Failed to save data!"
  84                     message_type = "error"
  85
  86         return self._template('admin/login_order.html',
  87                               message=message,
  88                               message_type=message_type,
  89                               title='login plugins order',
  90                               name='admin_login_order_form',
  91                               menu=self.menu, action=self.url,
  92                               options=self._site[FACILITY]['enabled'])
  93
  94     def root(self, *args, **kwargs):
  95         cherrypy.log.error("method: %s" % cherrypy.request.method)
  96         op = getattr(self, cherrypy.request.method, self.GET)
  97         if callable(op):
  98             return op(*args, **kwargs)
  99
 100
 101 class LoginPlugins(Page):
 102     def __init__(self, site, parent):
 103         super(LoginPlugins, self).__init__(site)
 104         self._master = parent
 105         self.title = 'Login Plugins'
 106         self.url = '%s/login' % parent.url
 107         self.facility = FACILITY
 108         parent.add_subtree('login', self)
 109
 110         for plugin in self._site[FACILITY]['available']:
 111             cherrypy.log.error('Admin login plugin: %s' % plugin)
 112             obj = self._site[FACILITY]['available'][plugin]
 113             self.__dict__[plugin] = AdminPluginPage(obj, self._site, self)
 114
 115         self.order = LoginPluginsOrder(self._site, self)
 116
 117     def root(self, *args, **kwargs):
 118         login_plugins = self._site[FACILITY]
 119         return self._template('admin/login.html', title=self.title,
 120                               available=login_plugins['available'],
 121                               enabled=login_plugins['enabled'],
 122                               menu=self._master.menu)