[IMP] stock_move_line_divide: Assign manufacturing customer to package

Author: aneavanzosc

stock_move_line_divide/__manifest__.py

@@ -1,5 +1,5 @@
 # Copyright 2022 Berezi Amubieta - AvanzOSC
-# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
+# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html)
 {
     "name": "Stock Move Line Divide",
     "version": "16.0.1.0.0",

stock_move_line_divide/models/stock_move_line.py

@@ -10,6 +10,10 @@ class StockMoveLine(models.Model):
 
     divide = fields.Integer(string="Divide in", default=1)
 
+    def _get_partner_from_mo(self):
+        production = self.move_id.production_id
+        return getattr(production, "manual_partner_id", False)
+
     def _get_default_dest_location(self):
         if self.location_dest_id != self.move_id.location_dest_id:
             return self.location_dest_id
@@ -42,8 +46,24 @@ def action_create_package(self, base_prefix=None):
             name = f"{prefix} - {next_seq:03}"
             pack_vals.update({"name": name})
 
-        package = self.env["stock.quant.package"].create(pack_vals)
-        return package
+        partner = self._get_partner_from_mo()
+        if partner:
+            pack_vals["partner_id"] = partner.id
+
+        return self.env["stock.quant.package"].create(pack_vals)
+
+    def _ensure_partner_on_packages(self, packages):
+        partner = self._get_partner_from_mo()
+        if partner and packages:
+            packages.filtered(lambda p: p and not p.partner_id).write(
+                {"partner_id": partner.id}
+            )
+
+    def write(self, vals):
+        res = super().write(vals)
+        if "result_package_id" in vals:
+            self._ensure_partner_on_packages(self.mapped("result_package_id"))
+        return res
 
     def action_divide(self):
         self.ensure_one()
@@ -84,9 +104,12 @@ def action_divide(self):
             self.result_package_id.name if self.result_package_id else self.reference
         )
 
+        created_packages = []
+
         first_qty = qty_quantities.pop(0)
         first_packaging_qty = packaging_qty_quantities.pop(0)
         first_package = self.action_create_package(base_prefix=base_prefix)
+        created_packages.append(first_package)
 
         self.write(
             {
@@ -102,6 +125,7 @@ def action_divide(self):
 
         for i, qty in enumerate(qty_quantities):
             package = self.action_create_package(base_prefix=base_prefix)
+            created_packages.append(package)
             new_vals = self.copy_data()[0]
             new_vals.update(
                 {
@@ -118,6 +142,10 @@ def action_divide(self):
                 new_vals
             )
 
+        self._ensure_partner_on_packages(
+            self.env["stock.quant.package"].browse([p.id for p in created_packages])
+        )
+
     @api.onchange("product_packaging_id")
     def _onchange_product_packaging_id(self):
         packaging = self.product_packaging_id
@@ -137,6 +165,16 @@ def _onchange_product_packaging_id(self):
             self.product_packaging_qty = 0
             self.qty_done = 1
 
+    @api.onchange("result_package_id")
+    def _onchange_result_package_id_set_partner(self):
+        for line in self:
+            pkg = line.result_package_id
+            if not pkg:
+                continue
+            partner = line._get_partner_from_mo()
+            if partner and not pkg.partner_id:
+                pkg.sudo().write({"partner_id": partner.id})
+
     @api.model_create_multi
     def create(self, vals_list):
         for vals in vals_list: