fix: 创建pt bug

This commit is contained in:
caoqianming 2023-05-29 15:46:40 +08:00
parent 1c9e7a8769
commit 10feb5493a
2 changed files with 5 additions and 1 deletions

View File

@ -131,6 +131,8 @@ class Task2DoSerializer(serializers.ModelSerializer):
class ValidateMixin:
def validate(self, attrs):
task2do = attrs.get('task2do', None)
attrs['create_by'] = self.context['request'].user
attrs['belong_dept'] = self.context['request'].user.belong_dept
if task2do:
if task2do.task2.state == Task2.TASK2_CLOSE:
raise ParseError('任务已关闭不可处理')
@ -150,6 +152,8 @@ class PgoalDeptSerializer(serializers.ModelSerializer):
def validate(self, attrs):
task2do = attrs.get('task2do', None)
attrs['create_by'] = self.context['request'].user
attrs['belong_dept'] = self.context['request'].user.belong_dept
if task2do:
if task2do.task2.state == Task2.TASK2_CLOSE:
raise ParseError('任务已关闭不可处理')

View File

@ -565,7 +565,7 @@ class Domixin:
@transaction.atomic
def perform_create(self, serializer):
instance = super().perform_create(serializer)
instance = serializer.save()
self.countup_task2do(instance.task2do)
@transaction.atomic