69 lines
2.2 KiB
Python
69 lines
2.2 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__'
|
|
|
|
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,))
|
|
|
|
|