From ba9cdd8f27bfa17772d2a367866e5e3121f515d0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?S=C3=A9bastien=20Alix?= <sebastien.alix@camptocamp.com>
Date: Wed, 8 Jan 2025 11:43:52 +0100
Subject: [PATCH] stock_move_line_change_lot: add hooks before/after moves
 reservation

---
 .../models/stock_move_line.py                    | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/stock_move_line_change_lot/models/stock_move_line.py b/stock_move_line_change_lot/models/stock_move_line.py
index 634a55adce83..1230591fb681 100644
--- a/stock_move_line_change_lot/models/stock_move_line.py
+++ b/stock_move_line_change_lot/models/stock_move_line.py
@@ -2,6 +2,8 @@
 # Copyright 2024 Jacques-Etienne Baudoux (BCIM) <je@bcim.be>
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
 
+from collections import defaultdict
+
 from odoo import _, models
 from odoo.exceptions import UserError
 from odoo.tools.float_utils import float_compare
@@ -69,6 +71,7 @@ def write(self, vals):
         res = True
         already_processed = self.browse()
         to_reassign_moves = self.env["stock.move"]
+        moves_by_previous_lot = defaultdict(self.env["stock.move"].browse)
         lot = self.env["stock.lot"].browse(vals["lot_id"])
         for move_line in self:
             if move_line.move_id._should_bypass_reservation(move_line.location_id):
@@ -87,6 +90,7 @@ def write(self, vals):
             package = move_line.package_id.browse(
                 vals.get("package_id", move_line.package_id)
             )
+            moves_by_previous_lot[move_line.lot_id] |= move_line.move_id
 
             available_quantity = 0
             # Collect new lot inside or outside a package (strict=False)
@@ -161,6 +165,18 @@ def write(self, vals):
         res &= super(StockMoveLine, self - already_processed).write(vals)
 
         if to_reassign_moves:
+            for previous_lot, moves in moves_by_previous_lot.items():
+                moves &= to_reassign_moves
+                self._hook_change_lot_before_assign(previous_lot, lot, moves)
             to_reassign_moves._action_assign()
+            for previous_lot, moves in moves_by_previous_lot.items():
+                moves &= to_reassign_moves
+                self._hook_change_lot_after_assign(previous_lot, lot, moves)
 
         return res
+
+    def _hook_change_lot_before_assign(self, previous_lot, lot, moves):
+        pass
+
+    def _hook_change_lot_after_assign(self, previous_lot, lot, moves):
+        pass