From 85b21dcbac4fa915a660e6d1ee8ef134ac1efc61 Mon Sep 17 00:00:00 2001 From: caoqianming Date: Thu, 14 Oct 2021 15:34:16 +0800 Subject: [PATCH] =?UTF-8?q?=E7=94=9F=E6=88=90=E5=AD=90=E8=AE=A1=E5=88=92?= =?UTF-8?q?=E5=BA=8F=E5=88=97=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- hb_server/apps/pm/views.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/hb_server/apps/pm/views.py b/hb_server/apps/pm/views.py index 2d5b0d2..b6442ed 100644 --- a/hb_server/apps/pm/views.py +++ b/hb_server/apps/pm/views.py @@ -1,3 +1,4 @@ +from rest_framework import serializers from rest_framework.views import APIView from apps.em.models import Equipment from apps.em.serializers import EquipmentSerializer @@ -40,7 +41,9 @@ class ProductionPlanViewSet(CreateUpdateModelAMixin, ListModelMixin, CreateModel def get_serializer_class(self): if self.action in ['create']: return ProductionPlanCreateFromOrderSerializer - return ProductionPlanSerializer + elif self.action == 'list': + return ProductionPlanSerializer + return super().get_serializer_class() def create(self, request, *args, **kwargs): data = request.data @@ -56,7 +59,7 @@ class ProductionPlanViewSet(CreateUpdateModelAMixin, ListModelMixin, CreateModel updateOrderPlanedCount(instance.order) return Response() - @action(methods=['post'], detail=True, perms_map={'post':'*'}) + @action(methods=['post'], detail=True, perms_map={'post':'*'}, serializer_class=serializers.Serializer) def gen_subplan(self, request, pk=None): """ 生成子计划