Add task to check required stock for build order

InvenTree/build/tasks.py

@@ -0,0 +1,85 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+import logging
+
+from django.utils.translation import ugettext_lazy as _
+from django.template.loader import render_to_string
+
+from allauth.account.models import EmailAddress
+
+from common.models import NotificationEntry
+
+import build.models
+import InvenTree.helpers
+import InvenTree.tasks
+
+
+logger = logging.getLogger('inventree')
+
+
+def check_build_stock(build: build.models.Build):
+    """
+    Check the required stock for a newly created build order,
+    and send an email out to any subscribed users if stock is low.
+    """
+
+    # Iterate through each of the parts required for this build
+
+    lines = []
+
+    for bom_item in build.part.get_bom_items():
+
+        sub_part = bom_item.sub_part
+
+        # The 'in stock' quantity depends on whether the bom_item allows variants
+        in_stock = sub_part.get_stock_count(include_variants=bom_item.allow_variants)
+
+        allocated = sub_part.allocation_count()
+
+        available = max(0, in_stock - allocated)
+
+        required = bom_item.quantity * build.quantity
+
+        if available < required:
+            # There is not sufficient stock for this part
+
+            lines.append({
+                'link': InvenTree.helpers.construct_absolute_url(sub_part.get_absolute_url()),
+                'part': sub_part,
+                'in_stock': in_stock,
+                'allocated': allocated,
+                'available': available,
+                'required': required,
+            })
+
+    if len(lines) == 0:
+        # Nothing to do
+        return
+
+    # Are there any users subscribed to these parts?
+    subscribers = build.part.get_subscribers()
+
+    emails = EmailAddress.objects.filter(
+        user__in=subscribers,
+    )
+
+    if len(emails) > 0:
+
+        logger.info(f"Notifying users of stock required for build {build.pk}")
+
+        context = {
+            'link': InvenTree.helpers.construct_absolute_url(build.get_absolute_url()),
+            'build': build,
+            'part': build.part,
+            'lines': lines,
+        }
+
+        # Render the HTML message
+        html_message = render_to_string('email/build_order_required_stock.html', context)
+
+        subject = "[InvenTree] " + _("Stock required for build order")
+
+        recipients = emails.values_list('email', flat=True)
+
+        InvenTree.tasks.send_email(subject, '', recipients, html_message=html_message)

InvenTree/part/models.py

@@ -1324,6 +1324,17 @@ class Part(MPTTModel):
 
         return query
 
+    def get_stock_count(self, include_variants=True):
+        """
+        Return the total "in stock" count for this part
+        """
+
+        entries = self.stock_entries(in_stock=True, include_variants=include_variants)
+
+        query = entries.aggregate(t=Coalesce(Sum('quantity'), Decimal(0)))
+
+        return query['t']
+
     @property
     def total_stock(self):
         """ Return the total stock quantity for this part.
@@ -1332,11 +1343,7 @@ class Part(MPTTModel):
         - If this part is a "template" (variants exist) then these are counted too
         """
 
-        entries = self.stock_entries(in_stock=True)
+        return self.get_stock_count()
-
-        query = entries.aggregate(t=Coalesce(Sum('quantity'), Decimal(0)))
-
-        return query['t']
 
     def get_bom_item_filter(self, include_inherited=True):
         """

InvenTree/part/tasks.py

@@ -50,7 +50,7 @@ def notify_low_stock(part: part.models.Part):
             'link': InvenTree.helpers.construct_absolute_url(part.get_absolute_url()),
         }
 
-        subject = _(f'[InvenTree] {part.name} is low on stock')
+        subject = "[InvenTree] " + _("Low stock notification")
         html_message = render_to_string('email/low_stock_notification.html', context)
         recipients = emails.values_list('email', flat=True)
 

InvenTree/templates/email/build_order_required_stock.html

@@ -0,0 +1,37 @@
+{% extends "email/email.html" %}
+
+{% load i18n %}
+{% load inventree_extras %}
+
+{% block title %}
+{% trans "Stock is required for the following build order" %}<br>
+{% blocktrans with build=build.reference part=part.full_name quantity=build.quantity %}Build order {{ build }} - building {{ quantity }} x {{ part }}{% endblocktrans %}
+<br>
+<p>{% trans "Click on the following link to view this build order" %}: <a href='{{ link }}'>{{ link }}</a></p>
+{% endblock title %}
+
+{% block body %}
+<tr colspan='100%' style='height: 2rem; text-align: center;'>{% trans "The following parts are low on required stock" %}</tr>
+
+<tr style="height: 3rem; border-bottom: 1px solid">
+    <th>{% trans "Part" %}</th>
+    <th>{% trans "Required Quantity" %}</th>
+    <th>{% trans "Available" %}</th>
+</tr>
+
+{% for line in lines %}
+<tr style="height: 2.5rem; border-bottom: 1px solid">
+    <td style='padding-left: 1em;'><a href='{{ line.link }}'>{{ line.part.full_name }}</a></td>
+    <td style="text-align: center;">
+        {% decimal line.required %} {% if line.part.units %}{{ line.part.units }}{% endif %}
+    </td>
+    <td style="text-align: center;">{% decimal line.available %}  {% if line.part.units %}{{ line.part.units }}{% endif %}</td>
+</tr>
+
+{% endfor %}
+
+{% endblock body %}
+
+{% block footer_prefix %}
+<p><em>{% blocktrans with part=part.name %}You are receiving this email because you are subscribed to notifications for this part {% endblocktrans %}.</em></p>
+{% endblock footer_prefix %}