diff --git a/InvenTree/order/forms.py b/InvenTree/order/forms.py index 4844ef4344..8432d9a6ef 100644 --- a/InvenTree/order/forms.py +++ b/InvenTree/order/forms.py @@ -8,8 +8,11 @@ from __future__ import unicode_literals from django import forms from django.utils.translation import ugettext as _ +from mptt.fields import TreeNodeChoiceField + from InvenTree.forms import HelperForm +from stock.models import StockLocation from .models import PurchaseOrder, PurchaseOrderLineItem @@ -35,6 +38,17 @@ class CancelPurchaseOrderForm(HelperForm): ] +class ReceivePurchaseOrderForm(HelperForm): + + location = TreeNodeChoiceField(queryset=StockLocation.objects.all(), required=True, help_text=_('Receive parts to this location')) + + class Meta: + model = PurchaseOrder + fields = [ + 'location', + ] + + class EditPurchaseOrderForm(HelperForm): """ Form for editing a PurchaseOrder object """ diff --git a/InvenTree/order/templates/order/purchase_order_detail.html b/InvenTree/order/templates/order/purchase_order_detail.html index 83bd526a96..d988f20739 100644 --- a/InvenTree/order/templates/order/purchase_order_detail.html +++ b/InvenTree/order/templates/order/purchase_order_detail.html @@ -108,14 +108,12 @@ InvenTree | {{ order }}
Location of received parts
- {% endif %} -Select parts to receive against this order.
@@ -64,6 +47,8 @@ Receive outstanding parts for {{ order }} - {{ order.description }} {% endfor %} + + {% crispy form %} {% endblock %} \ No newline at end of file diff --git a/InvenTree/order/views.py b/InvenTree/order/views.py index 49d2ed01ed..01ad682e6a 100644 --- a/InvenTree/order/views.py +++ b/InvenTree/order/views.py @@ -206,7 +206,7 @@ class PurchaseOrderExport(AjaxView): return DownloadFile(filedata, filename) -class PurchaseOrderReceive(AjaxView): +class PurchaseOrderReceive(AjaxUpdateView): """ View for receiving parts which are outstanding against a PurchaseOrder. Any parts which are outstanding are listed. @@ -214,6 +214,7 @@ class PurchaseOrderReceive(AjaxView): """ + form_class = order_forms.ReceivePurchaseOrderForm ajax_form_title = "Receive Parts" ajax_template_name = "order/receive_parts.html" @@ -225,12 +226,34 @@ class PurchaseOrderReceive(AjaxView): ctx = { 'order': self.order, 'lines': self.lines, - 'locations': StockLocation.objects.all(), - 'destination': self.destination, } return ctx + def get_lines(self): + """ + Extract particular line items from the request, + or default to *all* pending line items if none are provided + """ + + lines = None + + if 'line' in self.request.GET: + line_id = self.request.GET.get('line') + + try: + lines = PurchaseOrderLineItem.objects.filter(pk=line_id) + except (PurchaseOrderLineItem.DoesNotExist, ValueError): + pass + + # TODO - Option to pass multiple lines? + + # No lines specified - default selection + if lines is None: + lines = self.order.pending_line_items() + + return lines + def get(self, request, *args, **kwargs): """ Respond to a GET request. Determines which parts are outstanding, and presents a list of these parts to the user. @@ -239,13 +262,13 @@ class PurchaseOrderReceive(AjaxView): self.request = request self.order = get_object_or_404(PurchaseOrder, pk=self.kwargs['pk']) - self.lines = self.order.pending_line_items() + self.lines = self.get_lines() for line in self.lines: # Pre-fill the remaining quantity line.receive_quantity = line.remaining() - return self.renderJsonResponse(request) + return self.renderJsonResponse(request, form=self.get_form()) def post(self, request, *args, **kwargs): """ Respond to a POST request. Data checking and error handling. @@ -260,8 +283,8 @@ class PurchaseOrderReceive(AjaxView): self.destination = None # Extract the destination for received parts - if 'receive_location' in request.POST: - pk = request.POST['receive_location'] + if 'location' in request.POST: + pk = request.POST['location'] try: self.destination = StockLocation.objects.get(id=pk) except (StockLocation.DoesNotExist, ValueError): @@ -316,7 +339,7 @@ class PurchaseOrderReceive(AjaxView): 'success': 'Items marked as received', } - return self.renderJsonResponse(request, data=data) + return self.renderJsonResponse(request, data=data, form=self.get_form()) @transaction.atomic def receive_parts(self):