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