djadhere/services/views.py

from django.shortcuts import render, get_object_or_404
from django.views.generic import ListView, CreateView, DetailView, UpdateView, RedirectView
from django.views.generic.detail import SingleObjectMixin
from django.contrib.auth.mixins import LoginRequiredMixin
from django.contrib.auth.mixins import PermissionRequiredMixin
from django.http import HttpResponseGone
from django.core.exceptions import PermissionDenied
from django.utils import timezone

from djadhere.utils import get_active_filter
from .models import Service, ServiceType, IPResource, ResourceAllocation
from .forms import ServiceForm, ResourceAllocationForm


class ServiceMixin(PermissionRequiredMixin):
    model = Service
    permission_required = 'services.change_service'


class ServiceList(ServiceMixin, ListView):
    pass


class ServiceCreate(ServiceMixin, CreateView):
    form_class = ServiceForm


class ServiceDetail(LoginRequiredMixin, DetailView):
    def get_queryset(self):
        if self.request.user.is_superuser:
            return Service.objects.all()
        else:
            return Service.objects.filter(adhesion__pk__in=self.request.user.profile.adhesions.values_list('pk')) \
                        .order_by('-created')


class ServiceUpdate(ServiceMixin, UpdateView):
    fields = ('label', 'notes', 'active',)


class ServiceAllocate(PermissionRequiredMixin, CreateView):
    model = ResourceAllocation
    permission_required = 'services.change_resourceallocation'

    def get_form(self):
        service = get_object_or_404(Service, pk=self.kwargs['pk'])
        kwargs = super().get_form_kwargs()
        kwargs['initial'].update({'service': service.pk})
        form = ResourceAllocationForm(**kwargs)
        form.fields['service'].disabled = True
        form.fields['resource'].queryset = IPResource.objects.exclude(get_active_filter('allocation'))
        return form


class ServiceTypeMixin(PermissionRequiredMixin):
    model = ServiceType
    permission_required = 'services.change_servicetype'


class ServiceTypeList(ServiceTypeMixin, ListView):
    pass


class ServiceTypeCreate(ServiceTypeMixin, CreateView):
    fields = ('name',)


class ServiceTypeDetail(ServiceTypeMixin, DetailView):
    pass


class ServiceTypeUpdate(ServiceTypeMixin, UpdateView):
    fields = ('name',)


class IPResourceMixin(PermissionRequiredMixin):
    model = IPResource
    permission_required = 'services.change_ipresource'


class IPResourceList(IPResourceMixin, ListView):
    pass


class IPResourceDetail(IPResourceMixin, DetailView):
    pass


class IPResourceAllocate(PermissionRequiredMixin, CreateView):
    model = ResourceAllocation
    permission_required = 'services.change_resourceallocation'

    def get(self, request, *args, **kwargs):
        resource = get_object_or_404(IPResource, pk=self.kwargs['pk'])
        if resource.in_use:
            return HttpResponseGone("Cette IP est déjà alloué !")
        return super().get(request, *args, **kwargs)

    def get_form(self):
        resource = get_object_or_404(IPResource, pk=self.kwargs['pk'])
        kwargs = super().get_form_kwargs()
        kwargs['initial'].update({'resource': resource.pk})
        form = ResourceAllocationForm(**kwargs)
        form.fields['resource'].disabled = True
        return form


class ResourceAllocationMixin(PermissionRequiredMixin):
    model = ResourceAllocation
    permission_required = 'services.change_resourceallocation'


class ResourceAllocationDeallocate(ResourceAllocationMixin, SingleObjectMixin, RedirectView):
    def get_object(self, queryset=None):
        obj = super().get_object(queryset)
        if not obj.active:
            raise PermissionDenied
        obj.end = timezone.now()
        obj.save()
        return obj

    def get_redirect_url(self, *args, **kwargs):
        return self.get_object().get_absolute_url()