From 30bb96cf6cbb9b79584dce167209fd7b27fb2a2a Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Mon, 7 Jul 2025 10:34:40 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=8C=89=E4=B8=AA=E9=80=89=E5=85=A5?=
 =?UTF-8?q?=E6=B6=88=E8=80=97=E6=97=B6=E7=BB=9F=E8=AE=A1mlog=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/wpm/models.py | 1 +
 apps/wpm/views.py  | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/apps/wpm/models.py b/apps/wpm/models.py
index edda8964..56cfe1f2 100644
--- a/apps/wpm/models.py
+++ b/apps/wpm/models.py
@@ -506,6 +506,7 @@ class Mlogbw(BaseModel):
         if mlogb.material_in:
             mlogb.count_use = count
             mlogb.save(update_fields=["count_use"])
+            mlogb.cal_count_notok(cal_mlog=True)
         elif mlogb.material_out:
             mlogb.count_real = count
             count_notok = 0
diff --git a/apps/wpm/views.py b/apps/wpm/views.py
index 47ea6a60..67c4e71d 100644
--- a/apps/wpm/views.py
+++ b/apps/wpm/views.py
@@ -579,7 +579,7 @@ class MlogbInViewSet(CreateModelMixin, UpdateModelMixin, DestroyModelMixin, Cust
 
     @transaction.atomic
     def perform_update(self, serializer):
-        ins = serializer.save()
+        ins:Mlogb = serializer.save()
         ins.mlog.cal_mlog_count_from_mlogb()
     
     @transaction.atomic