jumpserver/apps/ops/views.py

# ~*~ coding: utf-8 ~*~
from __future__ import unicode_literals

from django.conf import settings
from django.views.generic.list import ListView, MultipleObjectMixin
from django.views.generic.edit import CreateView, DeleteView, UpdateView
from django.views.generic.detail import DetailView, SingleObjectMixin

from users.utils import AdminUserRequiredMixin
from ops.utils.mixins import CreateSudoPrivilegesMixin, ListSudoPrivilegesMixin
from ops.models import  *


class SudoListView(AdminUserRequiredMixin, ListSudoPrivilegesMixin, ListView):
    paginate_by = settings.CONFIG.DISPLAY_PER_PAGE
    model = Sudo
    context_object_name = 'sudos'
    template_name = 'sudo/list.html'


class SudoCreateView(AdminUserRequiredMixin, CreateSudoPrivilegesMixin, CreateView):
    model = Sudo
    template_name = 'sudo/create.html'


class SudoUpdateView(AdminUserRequiredMixin, UpdateView):
    model = Sudo
    template_name = 'sudo/update.html'


class SudoDetailView(DetailView):
    model = Sudo
    context_object_name = 'sudo'
    template_name = 'sudo/detail.html'


class CronListView(AdminUserRequiredMixin, ListView):
    paginate_by = settings.CONFIG.DISPLAY_PER_PAGE
    model = CronTable
    context_object_name = 'crons'
    template_name = 'cron/list.html'


class CronCreateView(AdminUserRequiredMixin, CreateView):
    model = CronTable
    template_name = 'cron/create.html'


class CronUpdateView(AdminUserRequiredMixin, UpdateView):
    model = CronTable
    template_name = 'cron/update.html'


class CronDetailView(DetailView):
    model = CronTable
    context_object_name = 'cron'
    template_name = 'cron/detail.html'

class TaskListView(AdminUserRequiredMixin, ListView):
    paginate_by = settings.CONFIG.DISPLAY_PER_PAGE
    model = Task
    context_object_name = 'tasks'
    template_name = 'task/list.html'

    def get_context_data(self, **kwargs):
        context = {
            'task': 'Assets',
            'action': 'Create asset',
        }
        kwargs.update(context)
        return super(TaskListView, self).get_context_data(**kwargs)


class TaskCreateView(AdminUserRequiredMixin, CreateView):
    model = Task
    template_name = 'task/create.html'


class TaskUpdateView(AdminUserRequiredMixin, UpdateView):
    model = Task
    template_name = 'task/update.html'


class TaskDetailView(DetailView):
    model = Task
    context_object_name = 'task'
    template_name = 'task/detail.html'