90 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			90 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
"""passbook PolicyBinding administration"""
 | 
						|
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.db.models import QuerySet
 | 
						|
from django.urls import reverse_lazy
 | 
						|
from django.utils.translation import gettext as _
 | 
						|
from django.views.generic import ListView, UpdateView
 | 
						|
from guardian.mixins import PermissionListMixin, PermissionRequiredMixin
 | 
						|
from guardian.shortcuts import get_objects_for_user
 | 
						|
 | 
						|
from passbook.admin.views.utils import DeleteMessageView
 | 
						|
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", "target"]
 | 
						|
    template_name = "administration/policy_binding/list.html"
 | 
						|
 | 
						|
    def get_queryset(self) -> QuerySet:
 | 
						|
        # Since `select_subclasses` does not work with a foreign key, we have to do two queries here
 | 
						|
        # First, get all pbm objects that have bindings attached
 | 
						|
        objects = (
 | 
						|
            get_objects_for_user(
 | 
						|
                self.request.user, "passbook_policies.view_policybindingmodel"
 | 
						|
            )
 | 
						|
            .filter(policies__isnull=False)
 | 
						|
            .select_subclasses()
 | 
						|
            .select_related()
 | 
						|
            .order_by("pk")
 | 
						|
        )
 | 
						|
        for pbm in objects:
 | 
						|
            pbm.bindings = get_objects_for_user(
 | 
						|
                self.request.user, self.permission_required
 | 
						|
            ).filter(target__pk=pbm.pbm_uuid)
 | 
						|
        return objects
 | 
						|
 | 
						|
 | 
						|
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-bindings")
 | 
						|
    success_message = _("Successfully created PolicyBinding")
 | 
						|
 | 
						|
 | 
						|
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-bindings")
 | 
						|
    success_message = _("Successfully updated PolicyBinding")
 | 
						|
 | 
						|
 | 
						|
class PolicyBindingDeleteView(
 | 
						|
    LoginRequiredMixin, PermissionRequiredMixin, DeleteMessageView
 | 
						|
):
 | 
						|
    """Delete policybinding"""
 | 
						|
 | 
						|
    model = PolicyBinding
 | 
						|
    permission_required = "passbook_policies.delete_policybinding"
 | 
						|
 | 
						|
    template_name = "generic/delete.html"
 | 
						|
    success_url = reverse_lazy("passbook_admin:policies-bindings")
 | 
						|
    success_message = _("Successfully deleted PolicyBinding")
 |