From 2addfc8433c13dcb2f82c1dc46d0f927a11d0ef0 Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Wed, 6 Mar 2024 21:52:55 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20handover=5Fsubmit=E9=92=88=E5=AF=B9?=
 =?UTF-8?q?=E6=B7=B7=E6=96=99=E7=89=B9=E6=AE=8A=E5=A4=84=E7=90=86=E7=9A=84?=
 =?UTF-8?q?bug2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/wpm/services.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/apps/wpm/services.py b/apps/wpm/services.py
index 57e63bbe..d24b7112 100644
--- a/apps/wpm/services.py
+++ b/apps/wpm/services.py
@@ -282,6 +282,7 @@ def handover_submit(handover: Handover, user: User, now: Union[datetime.datetime
         return
     now = timezone.now()
     need_add = True
+    wm_from_need_delete = False
     material = handover.material
     batch = handover.batch
     try:
@@ -295,7 +296,7 @@ def handover_submit(handover: Handover, user: User, now: Union[datetime.datetime
     if count_x < 0:
         raise ParseError('车间库存不足!')
     elif count_x == 0:
-        wm_from.delete()
+        wm_from_need_delete = True
     else:
         wm_from.count = count_x
         wm_from.save()
@@ -309,3 +310,5 @@ def handover_submit(handover: Handover, user: User, now: Union[datetime.datetime
     handover.submit_user = user
     handover.submit_time = now
     handover.save()
+    if wm_from_need_delete:
+        wm_from.delete()