mirror of
https://github.com/freedombox/FreedomBox.git
synced 2026-01-21 07:55:00 +00:00
153 lines
5.0 KiB
Python
153 lines
5.0 KiB
Python
from django import forms
|
|
from django.contrib import messages
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.contrib.auth.models import User
|
|
from django.core import validators
|
|
from django.core.urlresolvers import reverse_lazy
|
|
from django.template import RequestContext
|
|
from django.template.loader import render_to_string
|
|
from django.template.response import TemplateResponse
|
|
from gettext import gettext as _
|
|
import logging
|
|
|
|
from plinth import cfg
|
|
from plinth.modules.lib.auth import add_user
|
|
|
|
|
|
LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
def init():
|
|
"""Intialize the module"""
|
|
menu = cfg.main_menu.get('system:index')
|
|
menu.add_urlname(_('Users and Groups'), 'icon-user', 'users:index', 15)
|
|
|
|
|
|
@login_required
|
|
def index(request):
|
|
"""Return a rendered users page"""
|
|
menu = {'title': _('Users and Groups'),
|
|
'items': [{'url': reverse_lazy('users:add'),
|
|
'text': _('Add User')},
|
|
{'url': reverse_lazy('users:edit'),
|
|
'text': _('Edit Users')}]}
|
|
|
|
sidebar_right = render_to_string('menu_block.html', {'menu': menu},
|
|
RequestContext(request))
|
|
|
|
return TemplateResponse(request, 'base.html',
|
|
{'title': _('Manage Users and Groups'),
|
|
'sidebar_right': sidebar_right})
|
|
|
|
|
|
class UserAddForm(forms.Form): # pylint: disable-msg=W0232
|
|
"""Form to add a new user"""
|
|
|
|
username = forms.CharField(
|
|
label=_('Username'),
|
|
help_text=_('Must be lower case alphanumeric and start with \
|
|
and alphabet'),
|
|
validators=[
|
|
validators.RegexValidator(r'^[a-z][a-z0-9]*$',
|
|
_('Invalid username'))])
|
|
|
|
password = forms.CharField(label=_('Password'),
|
|
widget=forms.PasswordInput())
|
|
full_name = forms.CharField(label=_('Full name'), required=False)
|
|
email = forms.EmailField(label=_('Email'), required=False)
|
|
|
|
|
|
@login_required
|
|
def add(request):
|
|
"""Serve the form"""
|
|
form = None
|
|
|
|
if request.method == 'POST':
|
|
form = UserAddForm(request.POST, prefix='user')
|
|
# pylint: disable-msg=E1101
|
|
if form.is_valid():
|
|
_add_user(request, form.cleaned_data)
|
|
form = UserAddForm(prefix='user')
|
|
else:
|
|
form = UserAddForm(prefix='user')
|
|
|
|
return TemplateResponse(request, 'users_add.html',
|
|
{'title': _('Add User'),
|
|
'form': form})
|
|
|
|
|
|
def _add_user(request, data):
|
|
"""Add a user"""
|
|
if User.objects.filter(username=data['username']).exists():
|
|
messages.error(request, _('User "{username}" already exists').format(
|
|
username=data['username']))
|
|
return
|
|
|
|
add_user(data['username'], data['password'], data['full_name'],
|
|
data['email'], False)
|
|
messages.success(request, _('User "{username}" added').format(
|
|
username=data['username']))
|
|
|
|
|
|
class UserEditForm(forms.Form): # pylint: disable-msg=W0232
|
|
"""Form to edit/delete a user"""
|
|
def __init__(self, *args, **kwargs):
|
|
# pylint: disable-msg=E1002
|
|
super(forms.Form, self).__init__(*args, **kwargs)
|
|
|
|
for user in User.objects.all():
|
|
label = '%s (%s)' % (user.first_name, user.username)
|
|
field = forms.BooleanField(label=label, required=False)
|
|
# pylint: disable-msg=E1101
|
|
self.fields['delete_user_' + user.username] = field
|
|
|
|
|
|
@login_required
|
|
def edit(request):
|
|
"""Serve the edit form"""
|
|
form = None
|
|
|
|
if request.method == 'POST':
|
|
form = UserEditForm(request.POST, prefix='user')
|
|
# pylint: disable-msg=E1101
|
|
if form.is_valid():
|
|
_apply_edit_changes(request, form.cleaned_data)
|
|
form = UserEditForm(prefix='user')
|
|
else:
|
|
form = UserEditForm(prefix='user')
|
|
|
|
return TemplateResponse(request, 'users_edit.html',
|
|
{'title': _('Edit or Delete User'),
|
|
'form': form})
|
|
|
|
|
|
def _apply_edit_changes(request, data):
|
|
"""Apply form changes"""
|
|
for field, value in data.items():
|
|
if not value:
|
|
continue
|
|
|
|
if not field.startswith('delete_user_'):
|
|
continue
|
|
|
|
username = field.split('delete_user_')[1]
|
|
|
|
requesting_user = request.user.username
|
|
LOGGER.info('%s asked to delete %s', requesting_user, username)
|
|
|
|
if username == requesting_user:
|
|
messages.error(
|
|
request, _('Can not delete current account - "%s"') % username)
|
|
continue
|
|
|
|
if not User.objects.filter(username=username).exists():
|
|
messages.error(request, _('User "%s" does not exist') % username)
|
|
continue
|
|
|
|
try:
|
|
User.objects.filter(username=username).delete()
|
|
messages.success(request, _('User "%s" deleted') % username)
|
|
except IOError as exception:
|
|
messages.error(request, _('Error deleting "%s" - %s') %
|
|
(username, exception))
|