Mirrored_Repos/InvenTree
1
0
Fork 0
mirror of https://github.com/inventree/InvenTree synced 2024-08-30 18:33:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Reduced SQL load for 'Require Stock to Complete Build' widget on homepage

Browse Source
This commit is contained in:
eeintech 2020-09-22 11:24:09 -05:00
parent 12e43dcc46
commit 120a17241e
1 changed files with 19 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
InvenTree/part/api.py
View File

@ -7,7 +7,7 @@ from __future__ import unicode_literals
from django_filters.rest_framework import DjangoFilterBackend from django_filters.rest_framework import DjangoFilterBackend
from django.http import JsonResponse from django.http import JsonResponse
from django.db.models import Q, F, Count from django.db.models import Q, F, Count, Prefetch, Sum
from rest_framework import status from rest_framework import status
from rest_framework.response import Response from rest_framework.response import Response
@ -22,11 +22,14 @@ from .models import PartParameter, PartParameterTemplate
from .models import PartAttachment, PartTestTemplate from .models import PartAttachment, PartTestTemplate
from .models import PartSellPriceBreak from .models import PartSellPriceBreak
from build.models import Build
from . import serializers as part_serializers from . import serializers as part_serializers
from InvenTree.views import TreeSerializer from InvenTree.views import TreeSerializer
from InvenTree.helpers import str2bool, isNull from InvenTree.helpers import str2bool, isNull
from InvenTree.api import AttachmentMixin from InvenTree.api import AttachmentMixin
from InvenTree.status_codes import BuildStatus
class PartCategoryTree(TreeSerializer): class PartCategoryTree(TreeSerializer):
@ -531,12 +534,24 @@ class PartList(generics.ListCreateAPIView):
if stock_to_build is not None: if stock_to_build is not None:
# Filter only active parts # Filter only active parts
queryset = queryset.filter(active=True) queryset = queryset.filter(active=True)
# Prefetch current active builds
build_active_queryset = Build.objects.filter(status__in=BuildStatus.ACTIVE_CODES)
build_active_prefetch = Prefetch('builds',
queryset=build_active_queryset,
to_attr='current_builds')
parts = queryset.prefetch_related(build_active_prefetch)
# Store parts with builds needing stock
parts_need_stock = [] parts_need_stock = []
# Find parts with active builds # Find parts with active builds
# where any subpart's stock is lower than quantity being built # where any subpart's stock is lower than quantity being built
for part in queryset: for part in parts:
if part.active_builds and part.can_build < part.quantity_being_built: if part.current_builds:
builds_ids = [build.id for build in part.current_builds]
total_build_quantity = build_active_queryset.filter(pk__in=builds_ids).aggregate(quantity=Sum('quantity'))['quantity']
if part.can_build < total_build_quantity:
parts_need_stock.append(part.pk) parts_need_stock.append(part.pk)
queryset = queryset.filter(pk__in=parts_need_stock) queryset = queryset.filter(pk__in=parts_need_stock)