fix compute margin total

1 files changed, 7 insertions, 3 deletions

diff --git a/indoteknik_custom/models/purchase_order.py b/indoteknik_custom/models/purchase_order.py
index 6654bfd2..8ad25228 100755
--- a/indoteknik_custom/models/purchase_order.py
+++ b/indoteknik_custom/models/purchase_order.py
@@ -344,9 +344,13 @@ class PurchaseOrder(models.Model):
     def compute_total_margin(self):
         sum_so_margin = sum_sales_price = sum_margin = 0
         for line in self.order_line:
-            sale_order_line = self.env['sale.order.line'].search(
-                [('id', '=', line.so_line_id.id),
-                 ('order_id', '=', line.order_id.sale_order_id.id)], limit=1, order='price_reduce_taxexcl')
+            sale_order_line = line.so_line_id
+            if not sale_order_line:
+                sale_order_line = self.env['sale.order.line'].search([
+                    ('product_id', '=', line.product_id.id),
+                    ('order_id', '=', line.order_id.sale_order_id.id)
+                ], limit=1, order='price_reduce_taxexcl')
+
             sum_so_margin += sale_order_line.item_margin
             sales_price = sale_order_line.price_reduce_taxexcl * sale_order_line.product_uom_qty
             if sale_order_line.order_id.shipping_cost_covered == 'indoteknik':