extend the computer details with permissions

network_inventory/inventory/views.py

@@ -1,41 +1,50 @@
+from django.contrib.auth.decorators import login_required
 from django.shortcuts import get_object_or_404, render
 from django.views.generic import ListView
 from guardian.shortcuts import get_objects_for_user
+from inventory.decorators import computer_view_permission
 from inventory.models import (Device, Computer, ComputerRamRelation,
-                     ComputerDiskRelation,
-                     ComputerCpuRelation, Customer)
+                              ComputerDiskRelation, ComputerCpuRelation,
+                              Customer)
 
 
 def device_details(request, device_id):
-        device = get_object_or_404(Device, pk=device_id)
-        return render(request, 'inventory/device_details.html',
-                      {'device': device})
+    device = get_object_or_404(Device, pk=device_id)
+    return render(request, 'inventory/device_details.html',
+                  {'device': device})
 
 
+@login_required
+@computer_view_permission
 def computer_details(request, computer_id):
-        computer = get_object_or_404(Computer, pk=computer_id)
-        disks_list = ComputerDiskRelation.objects.filter(computer=computer_id)
-        ram = ComputerRamRelation.objects.get(computer=computer_id)
-        cpu = ComputerCpuRelation.objects.get(computer=computer_id)
-        return render(request, 'inventory/computer_details.html',
-                      {'computer': computer,
-                       'disks_list': disks_list,
-                       'ram': ram,
-                       'cpu': cpu})
+    computer = get_object_or_404(Computer, pk=computer_id)
+    disks_list = ComputerDiskRelation.objects.filter(computer=computer_id)
+    ram = ComputerRamRelation.objects.get(computer=computer_id)
+    cpu = ComputerCpuRelation.objects.get(computer=computer_id)
+    return render(request, 'inventory/computer_details.html',
+                  {'computer': computer,
+                   'disks_list': disks_list,
+                   'ram': ram,
+                   'cpu': cpu})
 
 
 class CustomerList(ListView):
     model = Customer
     template_name = 'inventory/customer_list.html'
 
+    def get_queryset(self):
+        queryset = get_objects_for_user(self.request.user,
+                                        'inventory.view_customer',
+                                        klass=Customer)
+        return queryset
+
 
 class ComputerList(ListView):
     model = Computer
     template_name = 'inventory/computer_list.html'
 
     def get_queryset(self):
-        queryset = get_objects_for_user(self.request.user, 'inventory.view_computer', klass=Computer)
-        #return super().get_queryset()
+        queryset = Computer.objects.filter(customer=self.kwargs['customer_id'])
         return queryset