# coding:utf-8 from __future__ import absolute_import, unicode_literals from django.utils.translation import ugettext as _ from django.conf import settings from django.views.generic import TemplateView, ListView, View from django.views.generic.edit import CreateView, DeleteView, FormView, UpdateView from django.urls import reverse_lazy from django.contrib.messages.views import SuccessMessageMixin from django.views.generic.detail import DetailView, SingleObjectMixin from .. import forms from ..models import Asset, AssetGroup, AdminUser, IDC, SystemUser from ..hands import AdminUserRequiredMixin __all__ = ['AdminUserCreateView', 'AdminUserDetailView', 'AdminUserDeleteView', 'AdminUserListView', 'AdminUserUpdateView', ] class AdminUserListView(AdminUserRequiredMixin, TemplateView): model = AdminUser template_name = 'assets/admin_user_list.html' def get_context_data(self, **kwargs): context = { 'app': _('Assets'), 'action': _('Admin user list'), } kwargs.update(context) return super(AdminUserListView, self).get_context_data(**kwargs) class AdminUserCreateView(AdminUserRequiredMixin, SuccessMessageMixin, CreateView): model = AdminUser form_class = forms.AdminUserForm template_name = 'assets/admin_user_create_update.html' success_url = reverse_lazy('assets:admin-user-list') def get_context_data(self, **kwargs): context = { 'app': 'assets', 'action': 'Create admin user' } kwargs.update(context) return super(AdminUserCreateView, self).get_context_data(**kwargs) def get_success_message(self, cleaned_data): success_message = _( 'Create admin user {name} successfully.'.format( url=reverse_lazy('assets:admin-user-detail', kwargs={'pk': self.object.pk}), name=self.object.name, )) return success_message def form_invalid(self, form): return super(AdminUserCreateView, self).form_invalid(form) class AdminUserUpdateView(AdminUserRequiredMixin, UpdateView): model = AdminUser form_class = forms.AdminUserForm template_name = 'assets/admin_user_create_update.html' def get_context_data(self, **kwargs): context = { 'app': 'assets', 'action': 'Update admin user' } kwargs.update(context) return super(AdminUserUpdateView, self).get_context_data(**kwargs) def get_success_url(self): success_url = reverse_lazy('assets:admin-user-detail', kwargs={'pk': self.object.pk}) return success_url class AdminUserDetailView(AdminUserRequiredMixin, SingleObjectMixin, ListView): paginate_by = settings.CONFIG.DISPLAY_PER_PAGE template_name = 'assets/admin_user_detail.html' context_object_name = 'admin_user' def get(self, request, *args, **kwargs): self.object = self.get_object(queryset=AdminUser.objects.all()) return super(AdminUserDetailView, self).get(request, *args, **kwargs) def get_queryset(self): return self.object.assets.all() def get_context_data(self, **kwargs): asset_groups = AssetGroup.objects.all() assets = self.get_queryset() context = { 'app': 'assets', 'action': 'Admin user detail', 'assets_remain': [asset for asset in Asset.objects.all() if asset not in assets], 'asset_groups': asset_groups, } kwargs.update(context) return super(AdminUserDetailView, self).get_context_data(**kwargs) class AdminUserDeleteView(AdminUserRequiredMixin, DeleteView): model = AdminUser template_name = 'assets/delete_confirm.html' success_url = reverse_lazy('assets:admin-user-list')