refector: 更新物料总量用update

This commit is contained in:
caoqianming 2023-10-18 16:26:45 +08:00
parent 5e34558c02
commit 84db42e7c4
1 changed files with 4 additions and 2 deletions

View File

@ -2,6 +2,7 @@ from apps.inm.models import MIO, MIOItem, MaterialBatch
from rest_framework.exceptions import ValidationError from rest_framework.exceptions import ValidationError
from django.db.models.aggregates import Sum from django.db.models.aggregates import Sum
from apps.wpm.services import do_out, do_in from apps.wpm.services import do_out, do_in
from apps.mtm.models import Material
class InmService: class InmService:
@ -48,6 +49,7 @@ class InmService:
mb.save() mb.save()
else: else:
raise ValidationError('不支持的操作') raise ValidationError('不支持的操作')
material.count = MaterialBatch.objects.filter( material_count = MaterialBatch.objects.filter(
material=material).aggregate(total=Sum('count')).get('total', 0) material=material).aggregate(total=Sum('count')).get('total', 0)
material.save() Material.objects.filter(id=material.id).update(
count=material_count)