implement the utils.get_objects function

This commit is contained in:
Andreas Zweili 2020-02-29 15:49:00 +01:00
parent 09d3e522ac
commit 7293d2b8b2
1 changed files with 2 additions and 6 deletions

View File

@ -9,8 +9,8 @@ from django.urls import reverse
from django_filters.views import FilterView
from django_tables2 import RequestConfig
from django_tables2.views import SingleTableMixin
from guardian.shortcuts import get_objects_for_user
from core import utils
from backups.models import Backup
from customers.models import Customer
from customers.decorators import customer_view_permission
@ -79,11 +79,7 @@ class ComputersFilterView(LoginRequiredMixin, SingleTableMixin, FilterView):
filterset_class = ComputerFilter
def get_queryset(self):
customers = get_objects_for_user(self.request.user,
'customers.view_customer',
klass=Customer)
results = Computer.objects.filter(customer__in=customers)
return results
return utils.get_objects("Computer", self.request.user)
class ComputerCreateFromCustomerView(LoginRequiredMixin, CreateView):