feat: mtask update在已提交状态下只更改peifen_kg

This commit is contained in:
caoqianming 2023-12-16 09:56:46 +08:00
parent e24a16b692
commit b8b106dc2e
1 changed files with 8 additions and 4 deletions

View File

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