jumpserver/juser/views.py

# coding: utf-8

from django.shortcuts import render_to_response

from juser.models import UserGroup, User


class AddError(Exception):
    pass


def group_add(request):
    error = ''
    msg = ''
    if request.method == 'POST':
        group_name = request.POST.get('j_group_name', None)
        comment = request.POST.get('j_comment', None)

        try:
            if not group_name:
                error = u'组名不能为空'
                raise AddError

            group = UserGroup.objects.filter(name=group_name)
            if group:
                error = u'组 %s 已存在' % group_name
                raise AddError

            group = UserGroup(name=group_name, comment=comment)
            group.save()
        except AddError:
            pass

        except TypeError:
            error = u'保存用户失败'

        else:
            msg = u'添加组 %s 成功' % group_name

    return render_to_response('juser/group_add.html',
                              {'header_title': u'添加属组 | Add Group',
                               'path1': 'juser', 'path2': 'group_add',
                               'error': error, 'msg': msg})


def group_list(request):
    groups = UserGroup.objects.all()
    return render_to_response('juser/group_list.html',
                              {'header_title': u'查看属组 | Add Group',
                               'path1': 'juser', 'path2': 'group_add',
                               'groups': groups})


def user_list(request):
    pass


def user_add(request):
    error = ''
    msg = ''

    user_role = {'SU': 'SuperUser', 'GA': 'GroupAdmin', 'CU': 'CommonUser'}
    groups = UserGroup.objects.all()
    if request.method == 'POST':
        username = request.POST.get('j_username', None)
        password = request.POST.get('j_password', None)
        name = request.POST.get('j_name', None)
        email = request.POST.get('j_email', '')
        groups = request.POST.getlist('j_group', None)
        role = request.POST.get('j_role', None)
        ssh_pwd = request.POST.get('j_ssh_pwd', None)
        is_active = request.POST.get('j_is_active', None)

    return render_to_response('juser/user_add.html',
                              {'header_title': u'添加用户 | Add User',
                               'path1': 'juser', 'path2': 'user_add',
                               'roles': user_role, 'groups': groups})