From 0cd2802b04dc40c1a8b1477f37138dba3049f465 Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Thu, 21 Sep 2023 16:27:22 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=9B=B4=E6=96=B0material=E7=9A=84coun?=
 =?UTF-8?q?t?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/inm/services.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/apps/inm/services.py b/apps/inm/services.py
index f42ca335..6e32a836 100644
--- a/apps/inm/services.py
+++ b/apps/inm/services.py
@@ -1,5 +1,6 @@
 from apps.inm.models import MIO, MIOItem, MaterialBatch
 from rest_framework.exceptions import ValidationError
+from django.db.models.aggregates import Sum
 
 
 class InmService:
@@ -39,4 +40,8 @@ class InmService:
                 mb.count = mb.count - i.count
                 if mb.count < 0:
                     raise ValidationError('批次库存不足,操作失败')
-                mb.save()
\ No newline at end of file
+                mb.save()
+            else:
+                raise ValidationError('不支持的操作')
+            material.count = MaterialBatch.objects.filter(material=material).aggregate(total=Sum('count')).get('total', 0)
+            material.save()
\ No newline at end of file