From e0189be5a6ec0e92ba8ee67fa63d2a5f0cd1fbea Mon Sep 17 00:00:00 2001 From: Oliver Walters Date: Fri, 29 Apr 2022 01:19:36 +1000 Subject: [PATCH] Adds ability to filter StockItemTestresult API list by Build ID - Allows us to retrieve stock item test results in a single API query --- InvenTree/stock/api.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/InvenTree/stock/api.py b/InvenTree/stock/api.py index d4fc5c93d1..f6b21ca5af 100644 --- a/InvenTree/stock/api.py +++ b/InvenTree/stock/api.py @@ -23,6 +23,8 @@ from rest_framework.serializers import ValidationError from rest_framework.response import Response from rest_framework import generics, filters +from build.models import Build + import common.settings import common.models @@ -1159,6 +1161,20 @@ class StockItemTestResultList(generics.ListCreateAPIView): queryset = super().filter_queryset(queryset) + # Filter by 'build' + build = params.get('build', None) + + if build is not None: + + try: + build = Build.objects.get(pk=build) + + queryset = queryset.filter(stock_item__build=build) + + except (ValueError, Build.DoesNotExist): + pass + + # Filter by stock item item = params.get('stock_item', None)