API refactoring

- Specify prefetch_fields for optional child serializers
- Ref: inventree#11012

Author: SchrodingersGat

src/backend/InvenTree/build/serializers.py

@@ -1435,6 +1435,7 @@ class Meta:
             can_build=False,
         ),
         False,
+        prefetch_fields=['bom_item'],
     )
 
     assembly_detail = enable_filter(

src/backend/InvenTree/common/api.py

@@ -893,9 +893,7 @@ def filter_model_type(self, queryset, name, value):
 class ParameterMixin:
     """Mixin class for Parameter views."""
 
-    queryset = common.models.Parameter.objects.all().prefetch_related(
-        'model_type', 'updated_by', 'template', 'template__model_type'
-    )
+    queryset = common.models.Parameter.objects.all().prefetch_related('model_type')
     serializer_class = common.serializers.ParameterSerializer
     permission_classes = [IsAuthenticatedOrReadScope]
 

src/backend/InvenTree/common/serializers.py

@@ -808,11 +808,15 @@ def save(self, **kwargs):
     )
 
     updated_by_detail = enable_filter(
-        UserSerializer(source='updated_by', read_only=True, many=False), True
+        UserSerializer(source='updated_by', read_only=True, many=False),
+        True,
+        prefetch_fields=['updated_by'],
     )
 
     template_detail = enable_filter(
-        ParameterTemplateSerializer(source='template', read_only=True, many=False), True
+        ParameterTemplateSerializer(source='template', read_only=True, many=False),
+        True,
+        prefetch_fields=['template', 'template_model_type'],
     )
 
 

src/backend/InvenTree/company/api.py

@@ -430,13 +430,6 @@ class SupplierPriceBreakMixin:
     queryset = SupplierPriceBreak.objects.all()
     serializer_class = SupplierPriceBreakSerializer
 
-    def get_queryset(self):
-        """Return annotated queryset for the SupplierPriceBreak list endpoint."""
-        queryset = super().get_queryset()
-        queryset = SupplierPriceBreakSerializer.annotate_queryset(queryset)
-
-        return queryset
-
 
 class SupplierPriceBreakOutputOptions(OutputConfiguration):
     """Available output options for the SupplierPriceBreak endpoints."""

src/backend/InvenTree/company/serializers.py

@@ -572,12 +572,15 @@ def annotate_queryset(queryset):
     supplier_detail = enable_filter(
         CompanyBriefSerializer(
             source='part.supplier', many=False, read_only=True, allow_null=True
-        )
+        ),
+        False,
+        prefetch_fields=['part__supplier'],
     )
 
     part_detail = enable_filter(
         SupplierPartSerializer(
             source='part', brief=True, many=False, read_only=True, allow_null=True
         ),
         False,
+        prefetch_fields=['part', 'part__part', 'part__part__pricing_data'],
     )

src/backend/InvenTree/part/api.py

@@ -570,7 +570,7 @@ class PartPricingDetail(RetrieveUpdateAPI):
     """API endpoint for viewing part pricing data."""
 
     serializer_class = part_serializers.PartPricingSerializer
-    queryset = Part.objects.all()
+    queryset = Part.objects.all().select_related('sub_part', 'sub_part__pricing_data')
 
     def get_object(self):
         """Return the PartPricing object associated with the linked Part."""
@@ -1361,6 +1361,8 @@ class BomOutputOptions(OutputConfiguration):
         InvenTreeOutputOption('can_build', default=True),
         InvenTreeOutputOption('part_detail'),
         InvenTreeOutputOption('sub_part_detail'),
+        InvenTreeOutputOption('substitutes'),
+        InvenTreeOutputOption('pricing'),
     ]
 
 

src/backend/InvenTree/part/serializers.py

@@ -1563,7 +1563,15 @@ def validate_quantity(self, quantity):
     )
 
     substitutes = enable_filter(
-        BomItemSubstituteSerializer(many=True, read_only=True, allow_null=True), True
+        BomItemSubstituteSerializer(many=True, read_only=True, allow_null=True),
+        False,
+        filter_name='substitutes',
+        prefetch_fields=[
+            'substitutes',
+            'substitutes__part',
+            'substitutes__part__stock_items',
+            'substitutes__part__pricing_data',
+        ],
     )
 
     part_detail = enable_filter(
@@ -1684,9 +1692,7 @@ def annotate_queryset(queryset):
             'sub_part__stock_items',
             'sub_part__stock_items__allocations',
             'sub_part__stock_items__sales_order_allocations',
-            'substitutes',
-            'substitutes__part__stock_items',
-        ).select_related('part__pricing_data', 'sub_part__pricing_data')
+        )
 
         # Annotate with the 'total pricing' information based on unit pricing and quantity
         queryset = queryset.annotate(
@@ -1751,13 +1757,15 @@ class Meta:
             source='template', many=False, read_only=True
         ),
         True,
+        prefetch_fields=['template'],
     )
 
     category_detail = enable_filter(
         CategorySerializer(
             source='category', many=False, read_only=True, allow_null=True
         ),
         True,
+        prefetch_fields=['category'],
     )
 
 

src/backend/InvenTree/stock/serializers.py

@@ -223,7 +223,8 @@ class Meta:
         read_only_fields = ['pk', 'user', 'date']
 
     user_detail = enable_filter(
-        UserSerializer(source='user', read_only=True, allow_null=True)
+        UserSerializer(source='user', read_only=True, allow_null=True),
+        prefetch_fields=['user'],
     )
 
     template = serializers.PrimaryKeyRelatedField(
@@ -238,7 +239,8 @@ class Meta:
     template_detail = enable_filter(
         part_serializers.PartTestTemplateSerializer(
             source='template', read_only=True, allow_null=True
-        )
+        ),
+        prefetch_fields=['template'],
     )
 
     attachment = InvenTree.serializers.InvenTreeAttachmentSerializerField(
@@ -1244,11 +1246,13 @@ class Meta:
     label = serializers.CharField(read_only=True)
 
     item_detail = enable_filter(
-        StockItemSerializer(source='item', many=False, read_only=True, allow_null=True)
+        StockItemSerializer(source='item', many=False, read_only=True, allow_null=True),
+        prefetch_fields=['item'],
     )
 
     user_detail = enable_filter(
-        UserSerializer(source='user', many=False, read_only=True, allow_null=True)
+        UserSerializer(source='user', many=False, read_only=True, allow_null=True),
+        prefetch_fields=['user'],
     )
 
     deltas = serializers.JSONField(read_only=True)

src/backend/InvenTree/users/api.py

@@ -291,13 +291,6 @@ class GroupMixin(SerializerContextMixin):
     serializer_class = GroupSerializer
     permission_classes = [InvenTree.permissions.IsStaffOrReadOnlyScope]
 
-    def get_queryset(self):
-        """Return queryset for this endpoint.
-
-        Note that the queryset is filtered by the permissions of the current user.
-        """
-        return super().get_queryset().prefetch_related('rule_sets', 'user_set')
-
 
 class GroupOutputOptions(OutputConfiguration):
     """Holds all available output options for Group views."""

src/backend/InvenTree/users/serializers.py

@@ -268,11 +268,13 @@ def get_permissions(self, group: Group) -> dict:
             source='rule_sets', many=True, read_only=True, allow_null=True
         ),
         filter_name='role_detail',
+        prefetch_fields=['rule_sets'],
     )
 
     users = enable_filter(
         UserSerializer(source='user_set', many=True, read_only=True, allow_null=True),
         filter_name='user_detail',
+        prefetch_fields=['user_set'],
     )