From 0d5be566d028f7df2f47f1eb39d98c8220abcaf3 Mon Sep 17 00:00:00 2001
From: Lois Rilo <lois.rilo@forgeflow.com>
Date: Tue, 21 Sep 2021 16:08:09 +0200
Subject: [PATCH] [FIX] repair_stock_move: restrict lot/serial to consume.

Only reserve and consume the lot specified in repair line.
---
 repair_stock_move/models/stock_move.py | 51 ++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)

diff --git a/repair_stock_move/models/stock_move.py b/repair_stock_move/models/stock_move.py
index bd08333ab..09529a499 100644
--- a/repair_stock_move/models/stock_move.py
+++ b/repair_stock_move/models/stock_move.py
@@ -12,3 +12,54 @@ class StockMove(models.Model):
         string="Repair Line",
         ondelete="cascade",
     )
+
+    def _get_available_quantity(
+        self,
+        location_id,
+        lot_id=None,
+        package_id=None,
+        owner_id=None,
+        strict=False,
+        allow_negative=False,
+    ):
+        if self.repair_line_id and self.repair_line_id.lot_id:
+            lot_id = self.repair_line_id.lot_id
+
+        return super()._get_available_quantity(
+            location_id,
+            lot_id=lot_id,
+            package_id=package_id,
+            owner_id=owner_id,
+            strict=strict,
+            allow_negative=allow_negative,
+        )
+
+    def _update_reserved_quantity(
+        self,
+        need,
+        available_quantity,
+        location_id,
+        lot_id=None,
+        package_id=None,
+        owner_id=None,
+        strict=True,
+    ):
+        if self.repair_line_id and self.repair_line_id.lot_id:
+            lot_id = self.repair_line_id.lot_id
+        return super(StockMove, self)._update_reserved_quantity(
+            need,
+            available_quantity,
+            location_id,
+            lot_id=lot_id,
+            package_id=package_id,
+            owner_id=owner_id,
+            strict=strict,
+        )
+
+    def _prepare_move_line_vals(self, quantity=None, reserved_quant=None):
+        vals = super()._prepare_move_line_vals(
+            quantity=quantity, reserved_quant=reserved_quant
+        )
+        if self.repair_line_id and self.repair_line_id.lot_id:
+            vals["lot_id"] = self.repair_line_id.lot_id.id
+        return vals