Nebucatnetzer/network_inventory
1
0
Fork 0
mirror of https://github.com/Nebucatnetzer/network_inventory.git synced 2024-06-27 03:35:11 +02:00
Code Issues Releases Wiki Activity

add backup_view_permission decorator

Browse Source
This commit is contained in:
Andreas Zweili 2019-12-29 14:03:31 +01:00
parent 117fbaf084
commit 1b9f461895
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
network_inventory/inventory/decorators.py
View File

@ -1,6 +1,6 @@
from django.http import HttpResponseForbidden
from django.shortcuts import get_object_or_404
from .models import Computer, Customer, Device, ConnectedDevice, Net
from .models import Backup, Computer, Customer, Device, ConnectedDevice, Net
def computer_view_permission(old_fuction):
@ -66,6 +66,13 @@ def net_view_permission(old_fuction):
"You're not allowed to access this device."
)
return new_function
def backup_view_permission(old_fuction):
def new_function(request, pk, *args, **kwargs):
backup = get_object_or_404(Backup, pk=pk)
user = request.user
if user.has_perm('inventory.view_customer', backup.computer.customer):
return old_fuction(request, pk)
else:
return HttpResponseForbidden(