network_inventory/devices/views.py

from django.contrib.auth.decorators import login_required
from django.contrib.auth.mixins import LoginRequiredMixin
from django.shortcuts import get_object_or_404
from django.shortcuts import render
from django.urls import reverse
from django.views.generic import CreateView
from django.views.generic import DetailView
from django.views.generic import UpdateView
from django.views.generic import DeleteView

from django_tables2 import RequestConfig

from customers.decorators import customer_view_permission
from customers.models import Customer
from core.utils import get_objects

from .decorators import device_view_permission

from .forms import DeviceCreateForm
from .forms import DeviceInNetCreateForm
from .forms import DeviceInNetUpdateForm
from .forms import DeviceUpdateForm
from .forms import WarrantyCreateForm
from .forms import WarrantyUpdateForm

from .models import Device
from .models import DeviceInNet
from .models import DeviceManufacturer
from .models import Warranty

from .tables import DevicesTable
from .tables import WarrantiesTable


@login_required
@device_view_permission
def device_detail_view(request, pk):
    device = get_object_or_404(Device, pk=pk)
    warranty_relations = Warranty.objects.filter(device=pk)
    net_relations = DeviceInNet.objects.filter(device=pk)
    return render(request,
                  'devices/device_details.html',
                  {'device': device,
                   'warranty_relations': warranty_relations,
                   'net_relations': net_relations,
                   'pk': pk})


@login_required
@customer_view_permission
def devices_table_view(request, pk):
    table = DevicesTable(Device.objects.filter(customer=pk))
    RequestConfig(request).configure(table)
    return render(request,
                  'devices/device_list.html',
                  {'devices': table,
                   'pk': pk})


@login_required
def warranties_view(request):
    table = WarrantiesTable(get_objects("Warranty", request.user))
    RequestConfig(request).configure(table)
    return render(request,
                  'devices/warranties_list.html',
                  {'devices': table})


class DeviceCreateFromCustomerView(LoginRequiredMixin, CreateView):
    """
    A view to show a DeviceCreateForm after comming from a customers device
    table. The customer will be preselected in the form.
    """
    form_class = DeviceCreateForm
    template_name = 'devices/device_create.html'

    def get_success_url(self):
        return reverse('device_update', args=(self.object.pk,))

    def get_form_kwargs(self):
        """
        Pass the request user to the form.
        """
        kwargs = super(DeviceCreateFromCustomerView, self).get_form_kwargs()
        kwargs.update({'user': self.request.user})
        return kwargs

    def get_initial(self):
        """
        Set the customer dropdown to the customer from the previews view.
        """
        customer = get_object_or_404(Customer, id=self.kwargs.get('pk'))
        return {
            'customer': customer,
        }


class DeviceUpdateView(LoginRequiredMixin, UpdateView):
    model = Device
    form_class = DeviceUpdateForm
    template_name = 'devices/device_update.html'

    def get_success_url(self):
        return self.request.POST.get('previous_page')


class DeviceDeleteView(LoginRequiredMixin, DeleteView):
    model = Device

    def get_success_url(self):
        return reverse('devices', args=(self.object.customer.pk,))


class WarrantyCreateView(LoginRequiredMixin, CreateView):
    model = Warranty
    form_class = WarrantyCreateForm
    template_name = 'devices/warranty_create.html'

    def get_success_url(self):
        return self.request.POST.get('previous_page')

    def get_initial(self):
        """
        Set the device and customer dropdown to the device from the previous
        view and the customer related to the device.
        """
        self.device = get_object_or_404(Device, id=self.kwargs.get('pk'))
        return {
            'device': self.device,
            'customer': self.device.customer,
        }


class WarrantyUpdateView(LoginRequiredMixin, UpdateView):
    model = Warranty
    form_class = WarrantyUpdateForm
    template_name = 'devices/warranty_update.html'

    def get_success_url(self):
        return self.request.POST.get('previous_page')


class WarrantyDeleteView(LoginRequiredMixin, DeleteView):
    model = Warranty

    def get_success_url(self):
        return self.request.POST.get('previous_page')


class DeviceInNetCreateView(LoginRequiredMixin, CreateView):
    model = DeviceInNet
    form_class = DeviceInNetCreateForm
    template_name = 'devices/device_in_net_create.html'

    def get_success_url(self):
        return self.request.POST.get('previous_page')

    def get_initial(self):
        """
        Set the device and customer dropdown to the device from the previous
        view and the customer related to the device.
        """
        self.device = get_object_or_404(Device, id=self.kwargs.get('pk'))
        return {
            'device': self.device,
        }


class DeviceInNetUpdateView(LoginRequiredMixin, UpdateView):
    model = DeviceInNet
    form_class = DeviceInNetUpdateForm
    template_name = 'devices/device_in_net_update.html'

    def get_success_url(self):
        return self.request.POST.get('previous_page')


class DeviceInNetDeleteView(LoginRequiredMixin, DeleteView):
    model = DeviceInNet
    template_name = 'devices/device_in_net_confirm_delete.html'

    def get_success_url(self):
        return self.request.POST.get('previous_page')


class DeviceManufacturerDetailView(LoginRequiredMixin, DetailView):
    model = DeviceManufacturer
    template_name = 'devices/manufacturer_details.html'