From e12a1042b9249b9097c9c57a08349ee1b5df4930 Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Wed, 4 Sep 2024 16:40:37 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E4=BC=98=E5=8C=96add=5Fmtaskb?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/pm/views.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/apps/pm/views.py b/apps/pm/views.py
index dd136345..a1c8ce7c 100644
--- a/apps/pm/views.py
+++ b/apps/pm/views.py
@@ -215,12 +215,15 @@ class MtaskViewSet(CustomModelViewSet):
         sr.is_valid(raise_exception=True)
         vdata = sr.validated_data
         total = 0
+        userIds = []
         for v in vdata:
             total += v['count']
+            userIds.append(v['handle_user'].id)
         if total != mtask.count:
             raise ParseError('任务数量与分配数量不匹配')
         for v in vdata:
             Mtaskb.objects.get_or_create(mtask=mtask, handle_user=vdata['handle_user'], count=vdata['count'])
+        Mtaskb.objects.filter(mtask=mtask).exclude(handle_user__in=userIds).delete()
         return Response()
 
 class MtaskbViewSet(CustomModelViewSet):