[MIG] purchase_stock_operating_unit: Migration to 18.0

Author: BhaveshHeliconia

purchase_stock_operating_unit/README.rst

@@ -69,6 +69,9 @@ Contributors
 ------------
 
 - Saran Lim. <saranl@ecosoft.co.th>
+- `Heliconia Solutions Pvt. Ltd. <https://www.heliconia.io>`__
+
+  - Bhavesh Heliconia
 
 Maintainers
 -----------

purchase_stock_operating_unit/__manifest__.py

@@ -4,7 +4,7 @@
 {
     "name": "Purchase Stock Operating Unit",
     "summary": "Copies the operating unit of purchase picking to the stock picking",
-    "version": "15.0.1.0.0",
+    "version": "18.0.1.0.0",
     "author": "Ecosoft, Odoo Community Association (OCA)",
     "website": "https://github.com/OCA/operating-unit",
     "category": "Purchase Management",

purchase_stock_operating_unit/models/__init__.py

@@ -1,4 +1,4 @@
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
 
-from . import purchase_order_line
 from . import purchase
+from . import stock_rule

purchase_stock_operating_unit/models/purchase.py

@@ -1,24 +1,63 @@
 # Copyright 2023 Ecosoft Co., Ltd. (http://ecosoft.co.th)
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
 
-from odoo import _, api, models
+from odoo import api, fields, models
 from odoo.exceptions import UserError
 
 
 class PurchaseOrder(models.Model):
     _inherit = "purchase.order"
 
+    picking_type_id = fields.Many2one(
+        compute="_compute_picking_type_id", store=True, readonly=False
+    )
+    operating_unit_id = fields.Many2one(
+        compute="_compute_operating_unit_id",
+        store=True,
+        readonly=False,
+    )
+
+    @api.depends("operating_unit_id")
+    def _compute_picking_type_id(self):
+        for purchase in self:
+            if purchase.operating_unit_id:
+                purchase.picking_type_id = self.env["stock.picking.type"].search(
+                    [
+                        (
+                            "warehouse_id.operating_unit_id",
+                            "=",
+                            purchase.operating_unit_id.id,
+                        ),
+                        ("code", "=", "incoming"),
+                    ],
+                    limit=1,
+                )
+
+    @api.depends("picking_type_id")
+    def _compute_operating_unit_id(self):
+        for purchase in self:
+            if purchase.picking_type_id:
+                purchase.operating_unit_id = (
+                    purchase.picking_type_id.warehouse_id.operating_unit_id
+                )
+
     @api.constrains("operating_unit_id", "picking_type_id")
     def _check_operating_unit_picking_type(self):
         for rec in self:
             if (
                 rec.operating_unit_id
+                and rec.picking_type_id.warehouse_id.operating_unit_id
                 and rec.operating_unit_id
                 != rec.picking_type_id.warehouse_id.operating_unit_id
             ):
                 raise UserError(
-                    _(
+                    self.env._(
                         "Configuration error. The Operating Unit in "
                         "the Purchase and Deliver To must be the same."
                     )
                 )
+
+    def _prepare_picking(self):
+        vals = super()._prepare_picking()
+        vals["operating_unit_id"] = self.operating_unit_id.id
+        return vals

purchase_stock_operating_unit/models/purchase_order_line.py

@@ -0,0 +1,14 @@
+# Copyright 2023 Ecosoft Co., Ltd. (http://ecosoft.co.th)
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+
+from odoo import models
+
+
+class StockRule(models.Model):
+    _inherit = "stock.rule"
+
+    def _prepare_purchase_order(self, company_id, origins, values):
+        res = super()._prepare_purchase_order(company_id, origins, values)
+        if self.operating_unit_id:
+            res["operating_unit_id"] = self.operating_unit_id.id
+        return res

purchase_stock_operating_unit/models/stock_rule.py

@@ -1 +1,3 @@
 - Saran Lim. \<<saranl@ecosoft.co.th>\>
+- [Heliconia Solutions Pvt. Ltd.](https://www.heliconia.io)
+  - Bhavesh Heliconia

purchase_stock_operating_unit/readme/CONTRIBUTORS.md

@@ -416,6 +416,10 @@ <h2><a class="toc-backref" href="#toc-entry-4">Authors</a></h2>
 <h2><a class="toc-backref" href="#toc-entry-5">Contributors</a></h2>
 <ul class="simple">
 <li>Saran Lim. &lt;<a class="reference external" href="mailto:saranl&#64;ecosoft.co.th">saranl&#64;ecosoft.co.th</a>&gt;</li>
+<li><a class="reference external" href="https://www.heliconia.io">Heliconia Solutions Pvt. Ltd.</a><ul>
+<li>Bhavesh Heliconia</li>
+</ul>
+</li>
 </ul>
 </div>
 <div class="section" id="maintainers">

purchase_stock_operating_unit/static/description/index.html

@@ -1,39 +1,63 @@
 # Copyright 2023 Ecosoft Co., Ltd. (http://ecosoft.co.th)
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
 
-from odoo.exceptions import UserError
-from odoo.tests import Form
+from odoo.models import Command
 
 from odoo.addons.purchase_operating_unit.tests.test_purchase_operating_unit import (
     TestPurchaseOperatingUnit,
 )
 
 
 class TestPurchaseStockOperatingUnit(TestPurchaseOperatingUnit):
-    def setUp(self):
-        super().setUp()
-        self.warehouse_b2b = self.env.ref("stock_operating_unit.stock_warehouse_b2b")
-        self.picking_type2 = self.env["stock.picking.type"].search(
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+        cls.warehouse_b2b = cls.env.ref("stock_operating_unit.stock_warehouse_b2b")
+        cls.picking_type2 = cls.env["stock.picking.type"].search(
             [
                 ("code", "=", "incoming"),
-                ("warehouse_id", "=", self.warehouse_b2b.id),
+                ("warehouse_id", "=", cls.warehouse_b2b.id),
             ],
             limit=1,
         )
-        # Add permission b2b operating unit in user1
-        self.b2b = self.env.ref("operating_unit.b2b_operating_unit")
-        user = self.env["res.users"].browse(self.user1_id)
-        user.operating_unit_ids = [(4, self.b2b.id)]
+        cls.user1.write(
+            {
+                "operating_unit_ids": [
+                    Command.link(cls.b2b.id),
+                ],
+            }
+        )
 
     def test_01_purchase_stock_operating_unit(self):
         self.assertEqual(self.purchase1.state, "purchase")
         self.assertEqual(
             self.purchase1.picking_ids.operating_unit_id,
             self.purchase1.picking_type_id.warehouse_id.operating_unit_id,
         )
+        # Ensure that in case of picking type changes,
+        # the operating unit is also updated.
         self.purchase1.button_cancel()
         self.purchase1.button_draft()
-        # Check change picking type is not equal operating unit, it should error
-        with self.assertRaises(UserError):
-            with Form(self.purchase1) as po:
-                po.picking_type_id = self.picking_type2
+        self.purchase1.picking_type_id = self.picking_type2
+        self.assertEqual(
+            self.purchase1.operating_unit_id,
+            self.purchase1.picking_type_id.warehouse_id.operating_unit_id,
+        )
+
+    def test_02_compute_picking_type_id(self):
+        self.purchase1.operating_unit_id = self.b2b
+        self.purchase1._compute_picking_type_id()
+        expected_picking_type = self.env["stock.picking.type"].search(
+            [
+                ("warehouse_id.operating_unit_id", "=", self.b2b.id),
+                ("code", "=", "incoming"),
+            ],
+            limit=1,
+        )
+        self.assertEqual(self.purchase1.picking_type_id, expected_picking_type)
+
+    def test_03_prepare_picking(self):
+        picking_vals = self.purchase1._prepare_picking()
+        self.assertEqual(
+            picking_vals.get("operating_unit_id"), self.purchase1.operating_unit_id.id
+        )