diff --git a/InvenTree/InvenTree/views.py b/InvenTree/InvenTree/views.py index ad0d70bfe8..91261827e7 100644 --- a/InvenTree/InvenTree/views.py +++ b/InvenTree/InvenTree/views.py @@ -238,6 +238,18 @@ class AjaxCreateView(AjaxMixin, CreateView): - Handles form validation via AJAX POST requests """ + def pre_save(self, **kwargs): + """ + Hook for doing something before the form is validated + """ + pass + + def post_save(self, **kwargs): + """ + Hook for doing something with the created object after it is saved + """ + pass + def get(self, request, *args, **kwargs): """ Creates form with initial data, and renders JSON response """ @@ -255,26 +267,29 @@ class AjaxCreateView(AjaxMixin, CreateView): - Return status info (success / failure) """ self.request = request - form = self.get_form() + self.form = self.get_form() # Extra JSON data sent alongside form data = { - 'form_valid': form.is_valid(), + 'form_valid': self.form.is_valid(), } - if form.is_valid(): - obj = form.save() + if self.form.is_valid(): + + self.pre_save() + self.object = self.form.save() + self.post_save() # Return the PK of the newly-created object - data['pk'] = obj.pk - data['text'] = str(obj) + data['pk'] = self.object.pk + data['text'] = str(object) try: - data['url'] = obj.get_absolute_url() + data['url'] = self.object.get_absolute_url() except AttributeError: pass - return self.renderJsonResponse(request, form, data) + return self.renderJsonResponse(request, self.form, data) class AjaxUpdateView(AjaxMixin, UpdateView): 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..d78c9c059e 100644 --- a/InvenTree/order/views.py +++ b/InvenTree/order/views.py @@ -91,6 +91,12 @@ class PurchaseOrderCreate(AjaxCreateView): return initials + def post_save(self, **kwargs): + # Record the user who created this purchase order + + self.object.created_by = self.request.user + self.object.save() + class PurchaseOrderEdit(AjaxUpdateView): """ View for editing a PurchaseOrder using a modal form """ @@ -206,7 +212,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 +220,7 @@ class PurchaseOrderReceive(AjaxView): """ + form_class = order_forms.ReceivePurchaseOrderForm ajax_form_title = "Receive Parts" ajax_template_name = "order/receive_parts.html" @@ -225,12 +232,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 +268,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 +289,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 +345,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): diff --git a/InvenTree/stock/models.py b/InvenTree/stock/models.py index 7b5d01c58b..1cf1a870b8 100644 --- a/InvenTree/stock/models.py +++ b/InvenTree/stock/models.py @@ -129,13 +129,17 @@ class StockItem(models.Model): else: add_note = False + user = kwargs.pop('user', None) + + add_note = add_note and kwargs.pop('note', True) + super(StockItem, self).save(*args, **kwargs) if add_note: # This StockItem is being saved for the first time self.addTransactionNote( 'Created stock item', - None, + user, notes="Created new stock item for part '{p}'".format(p=str(self.part)), system=True ) @@ -466,7 +470,8 @@ class StockItem(models.Model): if location: new_item.location = location - new_item.save() + # The item already has a transaction history, don't create a new note + new_item.save(user=user, note=False) # Copy entire transaction history new_item.copyHistoryFrom(self) diff --git a/InvenTree/stock/views.py b/InvenTree/stock/views.py index 1bb9a27a83..d8833d59d0 100644 --- a/InvenTree/stock/views.py +++ b/InvenTree/stock/views.py @@ -792,6 +792,8 @@ class StockItemCreate(AjaxCreateView): form = self.get_form() + data = {} + valid = form.is_valid() if valid: @@ -850,7 +852,7 @@ class StockItemCreate(AjaxCreateView): URL=data.get('URL'), ) - item.save() + item.save(user=request.user) except ValidationError as e: form.errors['serial_numbers'] = e.messages @@ -861,11 +863,15 @@ class StockItemCreate(AjaxCreateView): # We need to call _post_clean() here because it is prevented in the form implementation form.clean() form._post_clean() - form.save() + + item = form.save(commit=False) + item.save(user=request.user) - data = { - 'form_valid': valid, - } + data['pk'] = item.pk + data['url'] = item.get_absolute_url() + data['success'] = _("Created new stock item") + + data['form_valid'] = valid return self.renderJsonResponse(request, form, data=data)