From 5c5641d884a5eeb9a66cfcd31d9dd8e3ab898e63 Mon Sep 17 00:00:00 2001
From: Oliver Walters <oliver.henry.walters@gmail.com>
Date: Fri, 30 Oct 2020 10:12:42 +1100
Subject: [PATCH] Update calls to post_save

---
 InvenTree/order/views.py | 20 ++++++++++----------
 InvenTree/stock/views.py | 12 ++++++------
 2 files changed, 16 insertions(+), 16 deletions(-)

diff --git a/InvenTree/order/views.py b/InvenTree/order/views.py
index d2d02bb2c9..3a5bec3a7f 100644
--- a/InvenTree/order/views.py
+++ b/InvenTree/order/views.py
@@ -100,9 +100,9 @@ class PurchaseOrderAttachmentCreate(AjaxCreateView):
     ajax_template_name = "modal_form.html"
     role_required = 'purchase_order.add'
 
-    def post_save(self, **kwargs):
-        self.object.user = self.request.user
-        self.object.save()
+    def post_save(self, attachment, form, **kwargs):
+        attachment.user = self.request.user
+        attachment.save()
 
     def get_data(self):
         return {
@@ -148,7 +148,7 @@ class SalesOrderAttachmentCreate(AjaxCreateView):
     ajax_form_title = _('Add Sales Order Attachment')
     role_required = 'sales_order.add'
 
-    def post_save(self, **kwargs):
+    def post_save(self, attachment, form, **kwargs):
         self.object.user = self.request.user
         self.object.save()
 
@@ -319,11 +319,11 @@ class PurchaseOrderCreate(AjaxCreateView):
 
         return initials
 
-    def post_save(self, **kwargs):
+    def post_save(self, order, form, **kwargs):
         # Record the user who created this purchase order
 
-        self.object.created_by = self.request.user
-        self.object.save()
+        order.created_by = self.request.user
+        order.save()
 
 
 class SalesOrderCreate(AjaxCreateView):
@@ -351,10 +351,10 @@ class SalesOrderCreate(AjaxCreateView):
 
         return initials
 
-    def post_save(self, **kwargs):
+    def post_save(self, order, form, **kwargs):
         # Record the user who created this sales order
-        self.object.created_by = self.request.user
-        self.object.save()
+        order.created_by = self.request.user
+        order.save()
 
 
 class PurchaseOrderEdit(AjaxUpdateView):
diff --git a/InvenTree/stock/views.py b/InvenTree/stock/views.py
index 65c8f30893..45c862da83 100644
--- a/InvenTree/stock/views.py
+++ b/InvenTree/stock/views.py
@@ -164,11 +164,11 @@ class StockItemAttachmentCreate(AjaxCreateView):
     ajax_template_name = "modal_form.html"
     role_required = 'stock.add'
 
-    def post_save(self, **kwargs):
+    def post_save(self, attachment, form, **kwargs):
         """ Record the user that uploaded the attachment """
         
-        self.object.user = self.request.user
-        self.object.save()
+        attachment.user = self.request.user
+        attachment.save()
 
     def get_data(self):
         return {
@@ -440,11 +440,11 @@ class StockItemTestResultCreate(AjaxCreateView):
     ajax_form_title = _("Add Test Result")
     role_required = 'stock.add'
 
-    def post_save(self, **kwargs):
+    def post_save(self, result, form, **kwargs):
         """ Record the user that uploaded the test result """
 
-        self.object.user = self.request.user
-        self.object.save()
+        result.user = self.request.user
+        result.save()
 
     def get_initial(self):