64 lines
2.1 KiB
Python
64 lines
2.1 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
|
|
# Create your views here.
|
|
|
|
|
|
@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__'
|
|
|
|
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):
|
|
model = LicenseWithComputer
|
|
template_name = 'licenses/license_with_computer_confirm_delete.html'
|
|
|
|
def get_success_url(self):
|
|
return reverse('computer', args=(self.object.computer.pk,))
|
|
|
|
|