feat: 优化add_mtaskb

This commit is contained in:
caoqianming 2024-09-04 16:40:37 +08:00
parent 09b3f17025
commit e12a1042b9
1 changed files with 3 additions and 0 deletions

View File

@ -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):