network_inventory/src/nets/decorators.py

17 lines
518 B
Python

from django.http import HttpResponseForbidden
from django.shortcuts import get_object_or_404
from .models import Net
def net_view_permission(old_fuction):
def new_function(request, pk, *args, **kwargs):
net = get_object_or_404(Net, pk=pk)
user = request.user
if user.has_perm("customers.view_customer", net.customer):
return old_fuction(request, pk)
else:
return HttpResponseForbidden("You're not allowed to access this device.")
return new_function