140 lines
		
	
	
		
			4.6 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			140 lines
		
	
	
		
			4.6 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
"""passbook PropertyMapping 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.http import Http404
 | 
						|
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.core.models import PropertyMapping
 | 
						|
from passbook.lib.utils.reflection import path_to_class
 | 
						|
from passbook.lib.views import CreateAssignPermView
 | 
						|
 | 
						|
 | 
						|
def all_subclasses(cls):
 | 
						|
    """Recursively return all subclassess of cls"""
 | 
						|
    return set(cls.__subclasses__()).union(
 | 
						|
        [s for c in cls.__subclasses__() for s in all_subclasses(c)]
 | 
						|
    )
 | 
						|
 | 
						|
 | 
						|
class PropertyMappingListView(LoginRequiredMixin, PermissionListMixin, ListView):
 | 
						|
    """Show list of all property_mappings"""
 | 
						|
 | 
						|
    model = PropertyMapping
 | 
						|
    permission_required = "passbook_core.view_propertymapping"
 | 
						|
    template_name = "administration/property_mapping/list.html"
 | 
						|
    ordering = "name"
 | 
						|
    paginate_by = 40
 | 
						|
 | 
						|
    def get_context_data(self, **kwargs):
 | 
						|
        kwargs["types"] = {
 | 
						|
            x.__name__: x._meta.verbose_name for x in all_subclasses(PropertyMapping)
 | 
						|
        }
 | 
						|
        return super().get_context_data(**kwargs)
 | 
						|
 | 
						|
    def get_queryset(self):
 | 
						|
        return super().get_queryset().select_subclasses()
 | 
						|
 | 
						|
 | 
						|
class PropertyMappingCreateView(
 | 
						|
    SuccessMessageMixin,
 | 
						|
    LoginRequiredMixin,
 | 
						|
    DjangoPermissionRequiredMixin,
 | 
						|
    CreateAssignPermView,
 | 
						|
):
 | 
						|
    """Create new PropertyMapping"""
 | 
						|
 | 
						|
    model = PropertyMapping
 | 
						|
    permission_required = "passbook_core.add_propertymapping"
 | 
						|
 | 
						|
    template_name = "generic/create.html"
 | 
						|
    success_url = reverse_lazy("passbook_admin:property-mappings")
 | 
						|
    success_message = _("Successfully created Property Mapping")
 | 
						|
 | 
						|
    def get_context_data(self, **kwargs):
 | 
						|
        kwargs = super().get_context_data(**kwargs)
 | 
						|
        property_mapping_type = self.request.GET.get("type")
 | 
						|
        model = next(
 | 
						|
            x
 | 
						|
            for x in all_subclasses(PropertyMapping)
 | 
						|
            if x.__name__ == property_mapping_type
 | 
						|
        )
 | 
						|
        kwargs["type"] = model._meta.verbose_name
 | 
						|
        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):
 | 
						|
        property_mapping_type = self.request.GET.get("type")
 | 
						|
        model = next(
 | 
						|
            x
 | 
						|
            for x in all_subclasses(PropertyMapping)
 | 
						|
            if x.__name__ == property_mapping_type
 | 
						|
        )
 | 
						|
        if not model:
 | 
						|
            raise Http404
 | 
						|
        return path_to_class(model.form)
 | 
						|
 | 
						|
 | 
						|
class PropertyMappingUpdateView(
 | 
						|
    SuccessMessageMixin, LoginRequiredMixin, PermissionRequiredMixin, UpdateView
 | 
						|
):
 | 
						|
    """Update property_mapping"""
 | 
						|
 | 
						|
    model = PropertyMapping
 | 
						|
    permission_required = "passbook_core.change_propertymapping"
 | 
						|
 | 
						|
    template_name = "generic/update.html"
 | 
						|
    success_url = reverse_lazy("passbook_admin:property-mappings")
 | 
						|
    success_message = _("Successfully updated Property Mapping")
 | 
						|
 | 
						|
    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 (
 | 
						|
            PropertyMapping.objects.filter(pk=self.kwargs.get("pk"))
 | 
						|
            .select_subclasses()
 | 
						|
            .first()
 | 
						|
        )
 | 
						|
 | 
						|
 | 
						|
class PropertyMappingDeleteView(
 | 
						|
    SuccessMessageMixin, LoginRequiredMixin, PermissionRequiredMixin, DeleteView
 | 
						|
):
 | 
						|
    """Delete property_mapping"""
 | 
						|
 | 
						|
    model = PropertyMapping
 | 
						|
    permission_required = "passbook_core.delete_propertymapping"
 | 
						|
 | 
						|
    template_name = "generic/delete.html"
 | 
						|
    success_url = reverse_lazy("passbook_admin:property-mappings")
 | 
						|
    success_message = _("Successfully deleted Property Mapping")
 | 
						|
 | 
						|
    def get_object(self, queryset=None):
 | 
						|
        return (
 | 
						|
            PropertyMapping.objects.filter(pk=self.kwargs.get("pk"))
 | 
						|
            .select_subclasses()
 | 
						|
            .first()
 | 
						|
        )
 | 
						|
 | 
						|
    def delete(self, request, *args, **kwargs):
 | 
						|
        messages.success(self.request, self.success_message)
 | 
						|
        return super().delete(request, *args, **kwargs)
 |