network_inventory/src/backups/views.py

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__"

    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):
    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):
    model = Backup
    template_name = "backups/backup_confirm_delete.html"

    def get_success_url(self):
        return reverse("backups", args=(self.object.customer.pk,))