"""passbook PolicyBinding administration""" from django.contrib import messages from django.contrib.auth.mixins import LoginRequiredMixin from django.contrib.auth.mixins import ( PermissionRequiredMixin as DjangoPermissionRequiredMixin, ) from django.contrib.messages.views import SuccessMessageMixin from django.urls import reverse_lazy from django.utils.translation import ugettext as _ from django.views.generic import DeleteView, ListView, UpdateView from guardian.mixins import PermissionListMixin, PermissionRequiredMixin from passbook.lib.utils.reflection import path_to_class from passbook.lib.views import CreateAssignPermView from passbook.policies.forms import PolicyBindingForm from passbook.policies.models import PolicyBinding class PolicyBindingListView(LoginRequiredMixin, PermissionListMixin, ListView): """Show list of all policies""" model = PolicyBinding permission_required = "passbook_policies.view_policybinding" paginate_by = 10 ordering = "order" template_name = "administration/policybinding/list.html" class PolicyBindingCreateView( SuccessMessageMixin, LoginRequiredMixin, DjangoPermissionRequiredMixin, CreateAssignPermView, ): """Create new PolicyBinding""" model = PolicyBinding permission_required = "passbook_policies.add_policybinding" form_class = PolicyBindingForm template_name = "generic/create.html" success_url = reverse_lazy("passbook_admin:policies") success_message = _("Successfully created PolicyBinding") def get_context_data(self, **kwargs): kwargs = super().get_context_data(**kwargs) form_cls = self.get_form_class() if hasattr(form_cls, "template_name"): kwargs["base_template"] = form_cls.template_name return kwargs class PolicyBindingUpdateView( SuccessMessageMixin, LoginRequiredMixin, PermissionRequiredMixin, UpdateView ): """Update policybinding""" model = PolicyBinding permission_required = "passbook_policies.change_policybinding" form_class = PolicyBindingForm template_name = "generic/update.html" success_url = reverse_lazy("passbook_admin:policies") success_message = _("Successfully updated PolicyBinding") def get_context_data(self, **kwargs): kwargs = super().get_context_data(**kwargs) form_cls = self.get_form_class() if hasattr(form_cls, "template_name"): kwargs["base_template"] = form_cls.template_name return kwargs def get_form_class(self): form_class_path = self.get_object().form form_class = path_to_class(form_class_path) return form_class def get_object(self, queryset=None): return ( PolicyBinding.objects.filter(pk=self.kwargs.get("pk")) .select_subclasses() .first() ) class PolicyBindingDeleteView( SuccessMessageMixin, LoginRequiredMixin, PermissionRequiredMixin, DeleteView ): """Delete policybinding""" model = PolicyBinding permission_required = "passbook_policies.delete_policybinding" template_name = "generic/delete.html" success_url = reverse_lazy("passbook_admin:policies") success_message = _("Successfully deleted PolicyBinding") def delete(self, request, *args, **kwargs): messages.success(self.request, self.success_message) return super().delete(request, *args, **kwargs)