From dd32cf643270a0d46cf4c699100e1559abf3c1fa Mon Sep 17 00:00:00 2001 From: aarizona Date: Wed, 29 May 2024 18:22:30 +0300 Subject: [PATCH] filter data --- back/inventory/views.py | 7 +++++++ 1 file changed, 7 insertions(+) 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