231 lines
7.4 KiB
Python
231 lines
7.4 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, redirect
|
|
from django.shortcuts import render
|
|
from django.template.context_processors import csrf
|
|
from django.template.response import TemplateResponse
|
|
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 crispy_forms.utils import render_crispy_form
|
|
from crispy_forms.templatetags.crispy_forms_filters import as_crispy_field
|
|
from django_tables2 import RequestConfig
|
|
|
|
from customers.decorators import customer_view_permission
|
|
from customers.models import Customer
|
|
from core import utils
|
|
|
|
from .decorators import device_view_permission
|
|
|
|
from .forms import DeviceCategoryForm
|
|
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(utils.objects_for_allowed_customers(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,
|
|
}
|
|
|
|
|
|
@login_required
|
|
def device_update_view(request, pk):
|
|
"""
|
|
A view to create a customer.
|
|
"""
|
|
template_name = "devices/device_update.html"
|
|
request.session["device_to_update"] = pk
|
|
device = utils.get_object_with_view_permission(Device, user=request.user, pk=pk)
|
|
if request.method == "POST" and "save_device" in request.POST:
|
|
form = DeviceUpdateForm(request, request.POST, instance=device)
|
|
if form.is_valid():
|
|
device = form.save()
|
|
return redirect(device)
|
|
else:
|
|
form = DeviceUpdateForm(request, instance=device)
|
|
return TemplateResponse(request, template_name, {"form": form})
|
|
|
|
|
|
class DeviceDeleteView(LoginRequiredMixin, DeleteView): # type: ignore
|
|
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): # type: ignore
|
|
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): # type: ignore
|
|
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"
|
|
|
|
|
|
@login_required
|
|
def htmx_create_device_cagetory(request):
|
|
context = {}
|
|
if request.method == "POST" and "save_category" in request.POST:
|
|
form = DeviceCategoryForm(request.POST)
|
|
if form.is_valid():
|
|
category = form.save(commit=True)
|
|
pk = request.session.get("device_to_update")
|
|
device = utils.get_object_with_view_permission(
|
|
Device, user=request.user, pk=pk
|
|
)
|
|
device.category = category
|
|
device_form = DeviceUpdateForm(request, instance=device)
|
|
form_html = as_crispy_field(device_form["category"])
|
|
else:
|
|
context.update(csrf(request))
|
|
form.helper.attrs["hx-swap-oob"] = "true"
|
|
form_html = render_crispy_form(form, context=context)
|
|
context["valid"] = form.is_valid()
|
|
context["form"] = form_html
|
|
template_path = "devices/partials/device_category_response.html"
|
|
return TemplateResponse(request, template_path, context)
|
|
form = DeviceCategoryForm()
|
|
context["form"] = form
|
|
template_path = "devices/partials/device_category_create.html"
|
|
return TemplateResponse(request, template_path, context)
|