feat: quastat 加入批量接口

apps/qm/serializers.py

@@ -30,4 +30,15 @@ class QuaStatSerializer(CustomModelSerializer):
         end_time_local = localtime(end_time)
         attrs['year_s'], attrs['month_s'], attrs['day_s'] = end_time_local.year, end_time_local.month, end_time_local.day
         attrs['rate_pass'] = attrs['num_ok']/attrs['num_test']
-        return attrs
+        return attrs
+    
+class QuaStatUpdateSerializer(CustomModelSerializer):
+    belong_dept_name = serializers.CharField(source='belong_dept.name', read_only=True)
+    material_name = serializers.CharField(source='material.name', read_only=True)
+    testitem_name = serializers.CharField(source='testitem.name', read_only=True)
+
+    class Meta:
+        model = QuaStat
+        fields = '__all__'
+        read_only_fields = EXCLUDE_FIELDS + ['type', 'rate_pass', 'year_s', 'month_s', 'day_s', 'belong_dept', 'sflog', 'material', 'testitem']
+        extra_kwargs = {'val_avg': {'required': True}, 'num_test':{'required': True}, 'num_ok': {'required': True}}

apps/qm/views.py

@@ -1,10 +1,11 @@
 from django.shortcuts import render
-from rest_framework.mixins import ListModelMixin, CreateModelMixin
+from rest_framework.mixins import ListModelMixin, CreateModelMixin, UpdateModelMixin
 from rest_framework.decorators import action
 from apps.qm.models import QuaStat, TestItem
-from apps.qm.serializers import QuaStatSerializer, TestItemSerializer
+from apps.qm.serializers import QuaStatSerializer, TestItemSerializer, QuaStatUpdateSerializer
 from apps.qm.tasks import cal_quastat
 from rest_framework.response import Response
+from apps.utils.mixins import BulkCreateModelMixin, BulkUpdateModelMixin
 
 from apps.utils.viewsets import CustomGenericViewSet
 
@@ -21,19 +22,21 @@ class TestItemViewSet(ListModelMixin, CustomGenericViewSet):
     filterset_fields = []
     ordering = ['id']
 
-class QuaStatViewSet(ListModelMixin, CreateModelMixin, CustomGenericViewSet):
+class QuaStatViewSet(ListModelMixin, BulkUpdateModelMixin, CustomGenericViewSet):
     """
     list:质量分析报告
 
     质量分析报告
     """
-    perms_map = {'get': '*', 'post': 'quastat.create'}
+    perms_map = {'get': '*', 'put': 'quastat.update'}
     queryset = QuaStat.objects.all()
     serializer_class = QuaStatSerializer
+    update_serializer_class = QuaStatUpdateSerializer
     filterset_fields = ['type', 'year_s', 'month_s', 'day_s', 'material', 'testitem', 'belong_dept', 'sflog', 'sflog__mgroup']
     select_related_fields = ['belong_dept', 'material', 'testitem']
     
-    def perform_create(self, serializer):
+
+    def perform_update(self, serializer):
         ins = serializer.save()
         # 计算其他方面的统计
         cal_quastat.delay(ins.id)