From 985967fccb05aeab28aa3ec1e25c582aade5fd57 Mon Sep 17 00:00:00 2001
From: Matthias <matmair@live.de>
Date: Fri, 7 May 2021 07:13:23 +0200
Subject: [PATCH] save return of part.id

---
 InvenTree/order/views.py | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/InvenTree/order/views.py b/InvenTree/order/views.py
index d9c28457d2..40f57d247c 100644
--- a/InvenTree/order/views.py
+++ b/InvenTree/order/views.py
@@ -1582,22 +1582,26 @@ class LineItemPricing(PartPricing):
 
     form_class = EnhancedForm
 
-    def get_part(self):
+    def get_part(self, id=False):
         if 'line_item' in self.request.GET:
             try:
                 part_id = self.request.GET.get('line_item')
-                return SalesOrderLineItem.objects.get(id=part_id).part
+                part = SalesOrderLineItem.objects.get(id=part_id).part
             except Part.DoesNotExist:
                 return None
         elif 'pk' in self.request.POST:
             try:
                 part_id = self.request.POST.get('pk')
-                return Part.objects.get(id=part_id)
+                part = Part.objects.get(id=part_id)
             except Part.DoesNotExist:
                 return None
         else:
             return None
 
+        if id:
+            return part.id
+        return part
+
     def get_quantity(self):
         """ Return set quantity in decimal format """
         qty = Decimal(self.request.GET.get('quantity', 1))
@@ -1607,5 +1611,6 @@ class LineItemPricing(PartPricing):
 
     def get_initials(self):
         initials = super().get_initials()
-        initials['pk'] = self.get_part().id
+        initials['pk'] = self.get_part(id=True)
+
         return initials