diff --git a/back/inventory/views.py b/back/inventory/views.py index 3254bfb..fb96300 100644 --- a/back/inventory/views.py +++ b/back/inventory/views.py @@ -85,6 +85,13 @@ class ElementViewSet(viewsets.ModelViewSet): queryset = Element.objects.all() serializer_class = ElementSerializer + + def get_queryset(self): + queryset = Element.objects.all() + inventory = self.request.query_params.get('inventory_id') + if inventory is not None: + queryset = queryset.filter(inventory=inventory) + return queryset def create(self, request, **kwargs): data = request.data