From b63352ce200ce60f15a773bbd73644f296c7c16c Mon Sep 17 00:00:00 2001
From: Oliver Walters <oliver.henry.walters@gmail.com>
Date: Thu, 28 Apr 2022 22:58:58 +1000
Subject: [PATCH] Add a stock item transaction when installing items into a
 build output

---
 InvenTree/build/models.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/InvenTree/build/models.py b/InvenTree/build/models.py
index e5189e6073..61783fc8fb 100644
--- a/InvenTree/build/models.py
+++ b/InvenTree/build/models.py
@@ -1260,7 +1260,7 @@ class BuildItem(models.Model):
             })
 
     @transaction.atomic
-    def complete_allocation(self, user):
+    def complete_allocation(self, user, notes=''):
         """
         Complete the allocation of this BuildItem into the output stock item.
 
@@ -1286,8 +1286,13 @@ class BuildItem(models.Model):
                 self.save()
 
             # Install the stock item into the output
-            item.belongs_to = self.install_into
-            item.save()
+            self.install_into.installStockItem(
+                item,
+                self.quantity,
+                user,
+                notes
+            )
+            
         else:
             # Simply remove the items from stock
             item.take_stock(