From 0157daa0ab4a57c8563de07352568cc9b05571af Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Fri, 11 Oct 2024 15:41:55 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E8=8E=B7=E5=8F=96=E6=A0=87=E7=AD=BE?=
 =?UTF-8?q?=E6=98=8E=E7=BB=863?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/cm/views.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/apps/cm/views.py b/apps/cm/views.py
index 25f5db29..2ed04daa 100644
--- a/apps/cm/views.py
+++ b/apps/cm/views.py
@@ -5,13 +5,14 @@ from apps.cm.serializers import TidSerializer, LabelMatSerializer
 from apps.inm.models import MaterialBatch
 from rest_framework.exceptions import ParseError
 from rest_framework.response import Response
-from apps.utils.viewsets import CustomGenericViewSet, RetrieveModelMixin
+from apps.utils.viewsets import CustomGenericViewSet, RetrieveModelMixin, CustomListModelMixin
 # Create your views here.
 
-class LableMatViewSet(RetrieveModelMixin, CustomGenericViewSet):
+class LableMatViewSet(CustomListModelMixin, RetrieveModelMixin, CustomGenericViewSet):
     perms_map = {"post": "*", "get": "*"}
     serializer_class = LabelMatSerializer
     queryset = LableMat.objects.all()
+    select_related_fields = ["material", "material_origin", "supplier"]
 
     @action(methods=['post'], detail=False, serializer_class=TidSerializer)
     def get_from_mb(self, request, pk=None):