inventree/InvenTree
2
0
Fork 0
mirror of https://github.com/inventree/InvenTree.git synced 2025-12-16 17:28:11 +00:00
Code Activity

Fix prefetching for PartSerializer

Browse Source
This commit is contained in:
Oliver Walters
2025-11-25 01:56:53 +00:00
parent 5a4e84c83a
commit cb76020bf8
1 changed files with 0 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
src/backend/InvenTree/part/api.py
View File

@@ -58,8 +58,6 @@ from .models import (
PartCategory, PartCategory,
PartCategoryParameterTemplate, PartCategoryParameterTemplate,
PartInternalPriceBreak, PartInternalPriceBreak,
PartParameter,
PartParameterTemplate,
PartRelated, PartRelated,
PartSellPriceBreak, PartSellPriceBreak,
PartStocktake, PartStocktake,
@@ -1024,10 +1022,6 @@ class PartMixin(SerializerContextMixin):
queryset = part_serializers.PartSerializer.annotate_queryset(queryset) queryset = part_serializers.PartSerializer.annotate_queryset(queryset)
# Annotate with parameter template data?
if str2bool(self.request.query_params.get('parameters', False)):
queryset = queryset.prefetch_related('parameters', 'parameters__template')
if str2bool(self.request.query_params.get('price_breaks', True)): if str2bool(self.request.query_params.get('price_breaks', True)):
queryset = queryset.prefetch_related('salepricebreaks') queryset = queryset.prefetch_related('salepricebreaks')
@@ -1266,64 +1260,6 @@ class PartRelatedDetail(PartRelatedMixin, RetrieveUpdateDestroyAPI):
"""API endpoint for accessing detail view of a PartRelated object.""" """API endpoint for accessing detail view of a PartRelated object."""
class PartParameterTemplateFilter(FilterSet):
"""FilterSet for PartParameterTemplate objects."""
class Meta:
"""Metaclass options."""
model = PartParameterTemplate
# Simple filter fields
fields = ['name', 'units', 'checkbox']
has_choices = rest_filters.BooleanFilter(
method='filter_has_choices', label='Has Choice'
)
def filter_has_choices(self, queryset, name, value):
"""Filter queryset to include only PartParameterTemplates with choices."""
if str2bool(value):
return queryset.exclude(Q(choices=None) | Q(choices=''))
return queryset.filter(Q(choices=None) | Q(choices='')).distinct()
has_units = rest_filters.BooleanFilter(method='filter_has_units', label='Has Units')
def filter_has_units(self, queryset, name, value):
"""Filter queryset to include only PartParameterTemplates with units."""
if str2bool(value):
return queryset.exclude(Q(units=None) | Q(units=''))
return queryset.filter(Q(units=None) | Q(units='')).distinct()
part = rest_filters.ModelChoiceFilter(
queryset=Part.objects.all(), method='filter_part', label=_('Part')
)
@extend_schema_field(OpenApiTypes.INT)
def filter_part(self, queryset, name, part):
"""Filter queryset to include only PartParameterTemplates which are referenced by a part."""
parameters = PartParameter.objects.filter(part=part)
template_ids = parameters.values_list('template').distinct()
return queryset.filter(pk__in=[el[0] for el in template_ids])
# Filter against a "PartCategory" - return only parameter templates which are referenced by parts in this category
category = rest_filters.ModelChoiceFilter(
queryset=PartCategory.objects.all(),
method='filter_category',
label=_('Category'),
)
@extend_schema_field(OpenApiTypes.INT)
def filter_category(self, queryset, name, category):
"""Filter queryset to include only PartParameterTemplates which are referenced by parts in this category."""
cats = category.get_descendants(include_self=True)
parameters = PartParameter.objects.filter(part__category__in=cats)
template_ids = parameters.values_list('template').distinct()
return queryset.filter(pk__in=[el[0] for el in template_ids])
class PartStocktakeFilter(FilterSet): class PartStocktakeFilter(FilterSet):
"""Custom filter for the PartStocktakeList endpoint.""" """Custom filter for the PartStocktakeList endpoint."""