2
0
mirror of https://github.com/inventree/InvenTree.git synced 2025-04-30 20:46:47 +00:00

save return of part.id

This commit is contained in:
Matthias 2021-05-07 07:13:23 +02:00
parent 053793288b
commit 985967fccb

View File

@ -1582,22 +1582,26 @@ class LineItemPricing(PartPricing):
form_class = EnhancedForm form_class = EnhancedForm
def get_part(self): def get_part(self, id=False):
if 'line_item' in self.request.GET: if 'line_item' in self.request.GET:
try: try:
part_id = self.request.GET.get('line_item') 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: except Part.DoesNotExist:
return None return None
elif 'pk' in self.request.POST: elif 'pk' in self.request.POST:
try: try:
part_id = self.request.POST.get('pk') part_id = self.request.POST.get('pk')
return Part.objects.get(id=part_id) part = Part.objects.get(id=part_id)
except Part.DoesNotExist: except Part.DoesNotExist:
return None return None
else: else:
return None return None
if id:
return part.id
return part
def get_quantity(self): def get_quantity(self):
""" Return set quantity in decimal format """ """ Return set quantity in decimal format """
qty = Decimal(self.request.GET.get('quantity', 1)) qty = Decimal(self.request.GET.get('quantity', 1))
@ -1607,5 +1611,6 @@ class LineItemPricing(PartPricing):
def get_initials(self): def get_initials(self):
initials = super().get_initials() initials = super().get_initials()
initials['pk'] = self.get_part().id initials['pk'] = self.get_part(id=True)
return initials return initials