From 7f2804dff319cf4348d9951d16d3bc64bc020fb4 Mon Sep 17 00:00:00 2001
From: Oliver Walters <oliver.henry.walters@gmail.com>
Date: Thu, 5 Dec 2019 10:29:16 +1100
Subject: [PATCH] Add button to mark a purchase order as complete, even if not
 all line items are received

---
 InvenTree/order/forms.py                      | 11 ++++++
 InvenTree/order/models.py                     |  6 ++++
 .../order/templates/order/order_complete.html | 13 +++++++
 .../order/purchase_order_detail.html          | 11 ++++++
 InvenTree/order/urls.py                       |  1 +
 InvenTree/order/views.py                      | 36 +++++++++++++++++++
 6 files changed, 78 insertions(+)
 create mode 100644 InvenTree/order/templates/order/order_complete.html

diff --git a/InvenTree/order/forms.py b/InvenTree/order/forms.py
index 8432d9a6ef..97e0f0bdd9 100644
--- a/InvenTree/order/forms.py
+++ b/InvenTree/order/forms.py
@@ -27,6 +27,17 @@ class IssuePurchaseOrderForm(HelperForm):
         ]
 
 
+class CompletePurchaseOrderForm(HelperForm):
+
+    confirm = forms.BooleanField(required=False, help_text=_("Mark order as complete"))
+
+    class Meta:
+        model = PurchaseOrder
+        fields = [
+            'confirm',
+        ]
+
+
 class CancelPurchaseOrderForm(HelperForm):
 
     confirm = forms.BooleanField(required=False, help_text=_('Cancel order'))
diff --git a/InvenTree/order/models.py b/InvenTree/order/models.py
index b1b93b990a..c0be887fe1 100644
--- a/InvenTree/order/models.py
+++ b/InvenTree/order/models.py
@@ -188,6 +188,12 @@ class PurchaseOrder(Order):
 
         return self.lines.filter(quantity__gt=F('received'))
 
+    @property
+    def is_complete(self):
+        """ Return True if all line items have been received """
+
+        return self.pending_line_items().count() == 0
+
     @transaction.atomic
     def receive_line_item(self, line, location, quantity, user):
         """ Receive a line item (or partial line item) against this PO
diff --git a/InvenTree/order/templates/order/order_complete.html b/InvenTree/order/templates/order/order_complete.html
new file mode 100644
index 0000000000..f6b9f14338
--- /dev/null
+++ b/InvenTree/order/templates/order/order_complete.html
@@ -0,0 +1,13 @@
+{% extends "modal_form.html" %}
+
+{% block pre_form_content %}
+
+Mark this order as complete?
+{% if not order.is_complete %}
+<div class='alert alert-warning alert-block'>
+    This order has line items which have not been marked as received.
+    Marking this order as complete will remove these line items.
+</div>
+{% endif %}
+
+{% endblock %}
\ No newline at end of file
diff --git a/InvenTree/order/templates/order/purchase_order_detail.html b/InvenTree/order/templates/order/purchase_order_detail.html
index 0ee447ef0e..d7d34103b1 100644
--- a/InvenTree/order/templates/order/purchase_order_detail.html
+++ b/InvenTree/order/templates/order/purchase_order_detail.html
@@ -44,6 +44,9 @@ InvenTree | {{ order }}
                             <button type='button' class='btn btn-default btn-glyph' id='receive-order' title='Receive items'>
                                 <span class='glyphicon glyphicon-check'></span>
                             </button>
+                            <button type='button' class='btn btn-default btn-glyph' id='complete-order' title='Mark order as complete'>
+                                <span class='glyphicon glyphicon-ok'></span>
+                            </button>
                             {% endif %}
                             {% if order.status == OrderStatus.PENDING or order.status == OrderStatus.PLACED %}
                             <button type='button' class='btn btn-default btn-glyph' id='cancel-order' title='Cancel order'>
@@ -171,12 +174,14 @@ InvenTree | {{ order }}
 
 {% block js_ready %}
 
+{% if order.status == OrderStatus.PENDING and order.lines.count > 0 %}
 $("#place-order").click(function() {
     launchModalForm("{% url 'purchase-order-issue' order.id %}",
     {
         reload: true,
     });
 });
+{% endif %}
 
 $("#edit-order").click(function() {
     launchModalForm("{% url 'purchase-order-edit' order.id %}",
@@ -228,6 +233,12 @@ $("#receive-order").click(function() {
     });
 });
 
+$("#complete-order").click(function() {
+    launchModalForm("{% url 'purchase-order-complete' order.id %}", {
+        reload: true,
+    });
+});
+
 $("#export-order").click(function() {
     location.href = "{% url 'purchase-order-export' order.id %}";
 });
diff --git a/InvenTree/order/urls.py b/InvenTree/order/urls.py
index 994e6b054d..d30879cf8e 100644
--- a/InvenTree/order/urls.py
+++ b/InvenTree/order/urls.py
@@ -15,6 +15,7 @@ purchase_order_detail_urls = [
     url(r'^edit/?', views.PurchaseOrderEdit.as_view(), name='purchase-order-edit'),
     url(r'^issue/?', views.PurchaseOrderIssue.as_view(), name='purchase-order-issue'),
     url(r'^receive/?', views.PurchaseOrderReceive.as_view(), name='purchase-order-receive'),
+    url(r'^complete/?', views.PurchaseOrderComplete.as_view(), name='purchase-order-complete'),
 
     url(r'^export/?', views.PurchaseOrderExport.as_view(), name='purchase-order-export'),
 
diff --git a/InvenTree/order/views.py b/InvenTree/order/views.py
index d78c9c059e..8e3f32f907 100644
--- a/InvenTree/order/views.py
+++ b/InvenTree/order/views.py
@@ -184,6 +184,42 @@ class PurchaseOrderIssue(AjaxUpdateView):
         return self.renderJsonResponse(request, form, data)
 
 
+class PurchaseOrderComplete(AjaxUpdateView):
+    """ View for marking a PurchaseOrder as complete.
+    """
+
+    form_class = order_forms.CompletePurchaseOrderForm
+    model = PurchaseOrder
+    ajax_template_name = "order/order_complete.html"
+    ajax_form_title = "Complete Order"
+    context_object_name = 'order'
+
+    def get_context_data(self):
+
+        ctx = {
+            'order': self.get_object(),
+        }
+
+        return ctx
+
+    def post(self, request, *args, **kwargs):
+
+        confirm = str2bool(request.POST.get('confirm', False))
+
+        if confirm:
+            po = self.get_object()
+            po.status = OrderStatus.COMPLETE
+            po.save()
+
+        data = {
+            'form_valid': confirm
+        }
+
+        form = self.get_form()
+
+        return self.renderJsonResponse(request, form, data)
+
+
 class PurchaseOrderExport(AjaxView):
     """ File download for a purchase order