fix function recalculate from apo

1 files changed, 11 insertions, 0 deletions

diff --git a/indoteknik_custom/models/purchase_order.py b/indoteknik_custom/models/purchase_order.py
index 9e6d8556..89c0d8f1 100755
--- a/indoteknik_custom/models/purchase_order.py
+++ b/indoteknik_custom/models/purchase_order.py
@@ -557,6 +557,9 @@ class PurchaseOrder(models.Model):
             self.approval_status = 'pengajuan1'
 
     def re_calculate(self):
+        if self.from_apo:
+            self.re_calculate_from_apo()
+            return
         for line in self.order_line:
             sale_order_line = self.env['sale.order.line'].search([
                 ('product_id', 'in', [line.product_id.id]),
@@ -565,6 +568,14 @@ class PurchaseOrder(models.Model):
             for so_line in sale_order_line:
                 so_line.purchase_price = line.price_unit
 
+    def re_calculate_from_apo(self):
+        for line in self.order_sales_match_line:
+            order_line = self.env['purchase.order.line'].search([
+                ('product_id', '=', line.product_id.id),
+                ('order_id', '=', line.purchase_order_id.id)
+            ], limit=1)
+            line.sale_line_id.purchase_price = order_line.price_unit
+    
     def button_cancel(self):
         res = super(PurchaseOrder, self).button_cancel()
         self.approval_status = False