jumpserver/apps/assets/views/system_user.py

# ~*~ coding: utf-8 ~*~

from django.utils.translation import ugettext as _
from django.views.generic import TemplateView
from django.views.generic.edit import CreateView, DeleteView, UpdateView
from django.urls import reverse_lazy
from django.contrib.messages.views import SuccessMessageMixin
from django.views.generic.detail import DetailView

from common.const import create_success_msg, update_success_msg
from ..forms import SystemUserForm
from ..models import SystemUser, Node, CommandFilter
from common.permissions import PermissionsMixin, IsOrgAdmin


__all__ = [
    'SystemUserCreateView', 'SystemUserUpdateView',
    'SystemUserDetailView', 'SystemUserDeleteView',
    'SystemUserAssetView', 'SystemUserListView',
]


class SystemUserListView(PermissionsMixin, TemplateView):
    template_name = 'assets/system_user_list.html'
    permission_classes = [IsOrgAdmin]

    def get_context_data(self, **kwargs):
        context = {
            'app': _('Assets'),
            'action': _('System user list'),
        }
        kwargs.update(context)
        return super().get_context_data(**kwargs)


class SystemUserCreateView(PermissionsMixin, SuccessMessageMixin, CreateView):
    model = SystemUser
    form_class = SystemUserForm
    template_name = 'assets/system_user_create.html'
    success_url = reverse_lazy('assets:system-user-list')
    success_message = create_success_msg
    permission_classes = [IsOrgAdmin]

    def get_context_data(self, **kwargs):
        context = {
            'app': _('Assets'),
            'action': _('Create system user'),
        }
        kwargs.update(context)
        return super().get_context_data(**kwargs)


class SystemUserUpdateView(PermissionsMixin, SuccessMessageMixin, UpdateView):
    model = SystemUser
    form_class = SystemUserForm
    template_name = 'assets/system_user_update.html'
    success_url = reverse_lazy('assets:system-user-list')
    success_message = update_success_msg
    permission_classes = [IsOrgAdmin]

    def get_context_data(self, **kwargs):
        context = {
            'app': _('Assets'),
            'action': _('Update system user')
        }
        kwargs.update(context)
        return super().get_context_data(**kwargs)


class SystemUserDetailView(PermissionsMixin, DetailView):
    template_name = 'assets/system_user_detail.html'
    context_object_name = 'system_user'
    model = SystemUser
    permission_classes = [IsOrgAdmin]

    def get_context_data(self, **kwargs):
        cmd_filters_remain = CommandFilter.objects.exclude(system_users=self.object)
        context = {
            'app': _('Assets'),
            'action': _('System user detail'),
            'cmd_filters_remain': cmd_filters_remain,
        }
        kwargs.update(context)
        return super().get_context_data(**kwargs)


class SystemUserDeleteView(PermissionsMixin, DeleteView):
    model = SystemUser
    template_name = 'delete_confirm.html'
    success_url = reverse_lazy('assets:system-user-list')
    permission_classes = [IsOrgAdmin]


class SystemUserAssetView(PermissionsMixin, DetailView):
    model = SystemUser
    template_name = 'assets/system_user_assets.html'
    context_object_name = 'system_user'
    permission_classes = [IsOrgAdmin]

    def get_context_data(self, **kwargs):
        from ..utils import NodeUtil
        nodes_remain = Node.objects.exclude(systemuser=self.object)
        util = NodeUtil()
        nodes_remain = util.get_nodes_by_queryset(nodes_remain)
        context = {
            'app': _('assets'),
            'action': _('System user asset'),
            'nodes_remain': nodes_remain
        }
        kwargs.update(context)
        return super().get_context_data(**kwargs)