From fc42cca10e8606bd7c85ccc0687f1104af28cb31 Mon Sep 17 00:00:00 2001
From: Oliver Walters <oliver.henry.walters@gmail.com>
Date: Wed, 1 May 2019 08:08:50 +1000
Subject: [PATCH] Automatically allocate parts when a Build is created

- If there is only one StockItem to choose from, allocate parts from that StockItem
---
 InvenTree/build/models.py | 46 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)

diff --git a/InvenTree/build/models.py b/InvenTree/build/models.py
index 3891cc512a..d4e01795a4 100644
--- a/InvenTree/build/models.py
+++ b/InvenTree/build/models.py
@@ -12,6 +12,8 @@ from django.urls import reverse
 from django.db import models
 from django.core.validators import MinValueValidator
 
+from stock.models import StockItem
+
 
 class Build(models.Model):
     """ A Build object organises the creation of new parts from the component parts.
@@ -28,6 +30,50 @@ class Build(models.Model):
         notes: Text notes
     """
 
+    def save(self, *args, **kwargs):
+        """ Called when the Build model is saved to the database. 
+        
+        If this is a new Build, try to allocate StockItem objects automatically.
+        
+        - If there is only one StockItem for a Part, use that one.
+        - If there are multiple StockItem objects, leave blank and let the user decide
+        """
+
+        allocate_parts = False
+
+        # If there is no PK yet, then this is the first time the Build has been saved
+        if not self.pk:
+            allocate_parts = True
+
+        # Save this Build first
+        super(Build, self).save(*args, **kwargs)
+
+        if allocate_parts:
+            for item in self.part.bom_items.all():
+                part = item.sub_part
+                # Number of parts required for this build
+                q_req = item.quantity * self.quantity
+
+                stock = StockItem.objects.filter(part=part)
+
+                if len(stock) == 1:
+                    stock_item = stock[0]
+
+                    # Are there any parts available?
+                    if stock_item.quantity > 0:
+                        # If there are not enough parts, reduce the amount we will take
+                        if stock_item.quantity < q_req:
+                            q_req = stock_item.quantity
+
+                        # Allocate parts to this build
+                        build_item = BuildItem(
+                            build=self, 
+                            stock_item=stock_item,
+                            quantity=q_req)
+
+                        build_item.save()
+
+
     def __str__(self):
         return "Build {q} x {part}".format(q=self.quantity, part=str(self.part))