82 lines
2.6 KiB
Python
82 lines
2.6 KiB
Python
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,))
|