class GroupsController(BaseController):
def __init__(self):
super(GroupsController, self).__init__()
self.lmf = LdapModelFactory()
def __before__(self, action, **param):
super(GroupsController, self).__before__()
def _sidebar(self):
super(GroupsController, self)._sidebar()
c.actions.append({'name' : _('Show all groups'), 'args' : {'controller' : 'groups', 'action' : 'listGroups'}})
c.actions.append({'name' : _('Add Group'), 'args' : {'controller' : 'groups', 'action' : 'editGroup'}})
c.actions.append({'name' : _('Members'), 'args' : {'controller' : 'members', 'action' : 'index'}})
def index(self):
if self.lmf.isUserInGroup(self.identity, 'office') or self.lmf.isUserInGroup(self.identity, 'sysops'):
return self.listGroups()
return redirect(url(controller='profile', action='index'))
@BaseController.needGroup('superadmins')
def listGroups(self):
c.heading = _('Managed groups')
c.groups = self.lmf.getManagedGroupList()
return render('/groups/listGroups.mako')
@BaseController.needGroup('superadmins')
def editGroup(self):
# vary form depending on mode (do that over ajax)
if request.params.get('gid', '') == '':
c.group = Group()
action = 'Adding'
c.gid = ''
else:
try:
ParamChecker.checkUsername('gid', param=True)
except:
redirect(url(controller='groups', action='index'))
action = 'Editing'
c.gid = request.params['gid']
try:
c.group = self.lmf.getGroup(request.params['gid'])
users = ''
for u in c.group.users:
if not users == '':
users += '\n'
users += u
c.group.users = users
except LookupError:
# @TODO implement better handler
print 'No such group!'
redirect(url(controller='groups', action='index'))
c.heading = '{0} group'.format(action)
return render('/groups/editGroup.mako')
def checkEdit(f):
def new_f(self):
if not 'gid' in request.params:
redirect(url(controller='groups', action='index'))
else:
formok = True
errors = []
items = {}
try:
ParamChecker.checkUsername('gid', param=True)
except:
formok = False
errors.append(_('Invalid group ID'))
items['users'] = []
if 'users' in request.params:
try:
#ParamChecker.checkString('users', param=True, min_len=-1, max_len=9999, regex=r'([\w]{1,20}\n?)*')
for k in request.params['users'].split('\n'):
m = k.replace('\r', '').replace(' ', '')
if m == '':
continue
else:
ParamChecker.checkUsername(m, param=False)
items['users'].append(m)
except InvalidParameterFormat as ipf:
formok = False
errors.append(_('Invalid user name(s)'))
if not formok:
session['errors'] = errors
session['reqparams'] = {}
# @TODO request.params may contain multiple values per key... test & fix
for k in request.params.iterkeys():
session['reqparams'][k] = request.params[k]
session.save()
redirect(url(controller='groups', action='editGroup', gid=request.params['gid']))
else:
items['gid'] = request.params['gid']
return f(self, items)
return new_f
@BaseController.needGroup('superadmins')
@checkEdit
@restrict('POST')
def doEditGroup(self, items):
if not self.lmf.addGroup(items['gid']):
session['flash'] = _('Failed to add group!')
session['flash_class'] = 'error'
session.save()
else:
try:
lgrp_members = self.lmf.getGroupMembers(items['gid'])
except LookupError:
lgrp_members = []
# Adding new members
for m in items['users']:
if not m in lgrp_members:
#print 'adding -> ' + str(m)
self.lmf.changeUserGroup(m, items['gid'], True)
# Removing members
for m in lgrp_members:
if not m in items['users']:
#print 'removing -> ' + str(m)
self.lmf.changeUserGroup(m, items['gid'], False)
# @TODO add group if not exist
session['flash'] = _('Group saved successfully')
session['flash_class'] = 'success'
session.save()
redirect(url(controller='groups', action='index'))
@BaseController.needGroup('superadmins')
def unmanageGroup(self):
try:
ParamChecker.checkUsername('gid', param=True)
except:
redirect(url(controller='groups', action='index'))
result = self.lmf.unmanageGroup(request.params['gid'])
if result:
session['flash'] = _('Group no longer managed')
session['flash_class'] = 'success'
else:
session['flash'] = _('Failed to remove group from management!')
session['flash_class'] = 'error'
session.save()
redirect(url(controller='groups', action='index'))
@BaseController.needGroup('superadmins')
def deleteGroup(self):
try:
ParamChecker.checkUsername('gid', param=True)
except:
redirect(url(controller='groups', action='index'))
result = self.lmf.deleteGroup(request.params['gid'])
if result:
session['flash'] = _('Group successfully deleted')
session['flash_class'] = 'success'
else:
session['flash'] = _('Failed to delete group!')
session['flash_class'] = 'error'
session.save()
redirect(url(controller='groups', action='index'))
