From b8b106dc2e24715e37fbebb815c97469c1a7ec4c Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Sat, 16 Dec 2023 09:56:46 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20mtask=20update=E5=9C=A8=E5=B7=B2?=
 =?UTF-8?q?=E6=8F=90=E4=BA=A4=E7=8A=B6=E6=80=81=E4=B8=8B=E5=8F=AA=E6=9B=B4?=
 =?UTF-8?q?=E6=94=B9peifen=5Fkg?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/pm/serializers.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/apps/pm/serializers.py b/apps/pm/serializers.py
index 2fa97378..3c09d1de 100644
--- a/apps/pm/serializers.py
+++ b/apps/pm/serializers.py
@@ -32,10 +32,14 @@ class UtaskSerializer(CustomModelSerializer):
         return attrs
 
     def update(self, instance, validated_data):
-        if instance.state != Mtask.MTASK_CREATED:
-            raise ValidationError('任务非创建中不可编辑')
-        new_data = {key: validated_data[key] for key in [
-            'number', 'count', 'start_date', 'end_date']}
+        if instance.state in [Mtask.MTASK_STOP, Mtask.MTASK_DONE]:
+            raise ValidationError('任务该状态下不可编辑')
+        if instance.state == Mtask.MTASK_CREATED:
+            new_data = {key: validated_data[key] for key in [
+                'number', 'count', 'start_date', 'end_date']}
+        else:
+            new_data = {key: validated_data[key] for key in [
+                'peifen_kg']}
         return super().update(instance, new_data)