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,))