Nebucatnetzer/network_inventory
1
0
Fork 0
mirror of https://github.com/Nebucatnetzer/network_inventory.git synced 2024-06-21 06:56:54 +02:00
Code Issues Releases Wiki Activity

add user_view_permission decorator

Browse Source
This commit is contained in:
Andreas Zweili 2020-01-01 19:57:19 +01:00
parent b3af7881b2
commit 8cb41d30e9
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
network_inventory/inventory/decorators.py
View File

@ -85,3 +85,16 @@ def backup_view_permission(old_fuction):
"You're not allowed to access this device."
)
return new_function
def user_view_permission(old_fuction):
def new_function(request, pk, *args, **kwargs):
inventory_user = get_object_or_404(User, pk=pk)
user = request.user
if user.has_perm('inventory.view_customer', inventory_user.customer):
return old_fuction(request, pk)
else:
return HttpResponseForbidden(
"You're not allowed to access this device."
)
return new_function