network_inventory/src/licenses/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 DeleteView

from django_tables2 import RequestConfig

from customers.decorators import customer_view_permission

from computers.models import Computer

from .models import ComputerLicense
from .models import UserLicense
from .models import LicenseWithComputer
from .tables import ComputerLicensesTable
from .tables import UserLicensesTable


@login_required
@customer_view_permission
def licenses_table_view(request, pk):
    user_licenses = UserLicensesTable(UserLicense.objects.filter(customer=pk))
    computer_licenses = ComputerLicensesTable(
        ComputerLicense.objects.filter(customer=pk)
    )
    RequestConfig(request).configure(user_licenses)
    RequestConfig(request).configure(computer_licenses)
    return render(
        request,
        "licenses/license_list.html",
        {
            "user_licenses": user_licenses,
            "computer_licenses": computer_licenses,
        },
    )


class LicenseWithComputerCreateView(LoginRequiredMixin, CreateView):
    model = LicenseWithComputer
    template_name = "licenses/license_with_computer_create.html"
    fields = "__all__"  # type: ignore

    def get_success_url(self):
        return reverse("computer", args=(self.computer.pk,))

    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.computer = get_object_or_404(Computer, id=self.kwargs.get("pk"))
        return {
            "computer": self.computer,
        }


class LicenseWithComputerDeleteView(LoginRequiredMixin, DeleteView):  # type: ignore
    model = LicenseWithComputer
    template_name = "licenses/license_with_computer_confirm_delete.html"

    def get_success_url(self):
        return reverse("computer", args=(self.object.computer.pk,))


class UserLicenseDeleteView(LoginRequiredMixin, DeleteView):  # type: ignore
    model = UserLicense
    template_name = "licenses/license_confirm_delete.html"

    def get_success_url(self):
        return reverse("licenses", args=(self.object.customer.pk,))


class ComputerLicenseDeleteView(LoginRequiredMixin, DeleteView):  # type: ignore
    model = ComputerLicense
    template_name = "licenses/license_confirm_delete.html"

    def get_success_url(self):
        return reverse("licenses", args=(self.object.customer.pk,))