network_inventory/src/backups/views.py

80 lines
2.5 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 computers.models import Computer
from customers.decorators import customer_view_permission
from .decorators import backup_view_permission
from .models import Backup
from .models import NotificationFromBackup
from .models import TargetDevice
from .tables import BackupsTable
@login_required
@customer_view_permission
def backups_table_view(request, pk):
computers = Computer.objects.filter(customer=pk)
table = BackupsTable(Backup.objects.filter(computer__in=computers))
RequestConfig(request).configure(table)
return render(request, "backups/backup_list.html", {"backups": table})
@login_required
@backup_view_permission
def backup_detail_view(request, pk):
backup = get_object_or_404(Backup, pk=pk)
target_device_list = TargetDevice.objects.filter(backup=backup)
notifications = NotificationFromBackup.objects.filter(backup=backup)
return render(
request,
"backups/backup_details.html",
{
"backup": backup,
"target_device_list": target_device_list,
"notifications": notifications,
},
)
class BackupCreateView(LoginRequiredMixin, CreateView):
model = Backup
template_name = "backups/backup_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 BackupDeleteView(LoginRequiredMixin, DeleteView): # type: ignore
model = Backup
template_name = "backups/backup_confirm_delete.html"
def get_success_url(self):
return reverse("computer", args=(self.object.computer.pk,))
class BackupDeleteFromTableView(LoginRequiredMixin, DeleteView): # type: ignore
model = Backup
template_name = "backups/backup_confirm_delete.html"
def get_success_url(self):
return reverse("backups", args=(self.object.customer.pk,))