filter data

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