[ADD] stock_reserve_area

Author: mariadforgeflow

setup/stock_reserve_area/odoo/addons/stock_reserve_area

@@ -0,0 +1 @@
+../../../../stock_reserve_area

setup/stock_reserve_area/setup.py

@@ -0,0 +1,6 @@
+import setuptools
+
+setuptools.setup(
+    setup_requires=['setuptools-odoo'],
+    odoo_addon=True,
+)

stock_reserve_area/__init__.py

@@ -0,0 +1,3 @@
+from . import models
+from .hooks import post_init_hook
+from .hooks import pre_init_hook

stock_reserve_area/__manifest__.py

@@ -0,0 +1,25 @@
+# Copyright 2023 ForgeFlow S.L.
+# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
+{
+    "name": "Stock Reservation Area",
+    "summary": "Stock reservations on areas (group of locations)",
+    "version": "14.0.1.0.0",
+    "author": "ForgeFlow, Odoo Community Association (OCA)",
+    "category": "Warehouse",
+    "license": "AGPL-3",
+    "complexity": "normal",
+    "website": "https://github.com/OCA/stock-logistics-warehouse",
+    "depends": ["stock"],
+    "data": [
+        "security/ir.model.access.csv",
+        "security/stock_reserve_area_security.xml",
+        "views/stock_reserve_area_views.xml",
+        "views/stock_location_views.xml",
+        "views/stock_move_views.xml",
+        "views/stock_picking_views.xml",
+    ],
+    "auto_install": False,
+    "installable": True,
+    "post_init_hook": "post_init_hook",
+    "pre_init_hook": "pre_init_hook",
+}

stock_reserve_area/hooks.py

@@ -0,0 +1,80 @@
+# Copyright 2023 ForgeFlow SL.
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+
+import logging
+
+from odoo import SUPERUSER_ID, api
+from odoo.tools.sql import column_exists, table_exists
+
+_logger = logging.getLogger(__name__)
+
+
+def pre_init_hook(cr):
+    if not column_exists(cr, "stock_move", "area_reserved_availability"):
+        cr.execute(
+            """
+                ALTER TABLE "stock_move"
+                ADD COLUMN "area_reserved_availability" double precision DEFAULT 0
+            """
+        )
+        cr.execute(
+            """
+        ALTER TABLE "stock_move" ALTER COLUMN "area_reserved_availability" DROP DEFAULT
+        """
+        )
+
+    if not table_exists(cr, "stock_move_stock_reserve_area_rel"):
+        cr.execute(
+            """
+        CREATE TABLE stock_move_stock_reserve_area_rel
+        (stock_move_id INTEGER, stock_reserve_area_id INTEGER);
+        """
+        )
+
+
+def assign_reserve_area_ids_to_stock_move_query(cr):
+    query = """
+        INSERT INTO stock_move_stock_reserve_area_rel (stock_move_id, stock_reserve_area_id)
+        SELECT DISTINCT sm.id AS stock_move_id, srl.reserve_area_id AS stock_reserve_area_id
+        FROM stock_move sm
+        JOIN stock_location sl ON sm.location_id = sl.id
+        JOIN stock_reserve_area_stock_location_rel srl ON sl.id = srl.location_id
+        WHERE NOT EXISTS (
+        SELECT 1
+        FROM stock_move_stock_reserve_area_rel existing_rel
+        WHERE existing_rel.stock_move_id = sm.id
+        AND existing_rel.stock_reserve_area_id = srl.reserve_area_id
+        );
+    """
+    cr.execute(query)
+
+
+def post_init_hook(cr, registry):
+    """
+    This post-init-hook will create a Reserve Area for each existing WH.
+    """
+    _logger.info("Starting Post Init Hook")
+
+    env = api.Environment(cr, SUPERUSER_ID, dict())
+    warehouse_obj = env["stock.warehouse"]
+    warehouses = warehouse_obj.search([])
+    reserve_area_obj = env["stock.reserve.area"]
+    _logger.info("Creating a Reserve Area for each WH")
+    for warehouse_id in warehouses.ids:
+        warehouse = warehouse_obj.browse(warehouse_id)
+
+        all_locations = env["stock.location"].search(
+            [("id", "child_of", warehouse.view_location_id.id)]
+        )
+
+        reserve_area_obj.create(
+            {
+                "name": warehouse.name,
+                "location_ids": [(6, 0, all_locations.ids)],
+                "company_id": warehouse.company_id.id,
+            }
+        )
+
+    _logger.info("Starting assign_reserve_area_ids_to_stock_move_query")
+
+    assign_reserve_area_ids_to_stock_move_query(cr)

stock_reserve_area/models/__init__.py

@@ -0,0 +1,8 @@
+from . import stock_reserve_area
+from . import stock_move
+from . import stock_quant
+from . import stock_location
+from . import stock_picking
+from . import stock_warehouse
+from . import stock_move_line
+from . import stock_move_reserve_area_line

stock_reserve_area/models/stock_location.py

@@ -0,0 +1,47 @@
+# Copyright 2023 ForgeFlow SL.
+# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
+
+from odoo import api, fields, models
+
+
+class StockLocation(models.Model):
+    _inherit = "stock.location"
+
+    reserve_area_ids = fields.Many2many(
+        "stock.reserve.area",
+        relation="stock_reserve_area_stock_location_rel",
+        column1="location_id",
+        column2="reserve_area_id",
+        readonly=True,
+    )
+
+    def write(self, vals):
+        res = super().write(vals)
+        return res
+
+    @api.depends("reserve_area_ids")
+    def _update_impacted_moves(self, vals):
+        """If a location is moved outside/inside an area we have to check stock_moves"""
+        if vals.get("reserve_area_ids"):
+            new_reserve_areas = (
+                self.env["stock.reserve.area"].sudo().browse(vals["reserve_area_ids"])
+            )
+            moves_poss_impacted = self.search(
+                [
+                    "|",
+                    ("location_id", "=", self),
+                    ("location_dest_id", "=", self),
+                    ("state", "in", ("confirmed", "waiting", "partially_available")),
+                ]
+            )
+            for move in moves_poss_impacted:
+                for reserve_area in new_reserve_areas:
+                    if move._is_out_area(reserve_area):
+                        move.reserve_area_line_ids += self.env[
+                            "stock.move.reserve.area.line"
+                        ].create(
+                            {
+                                "move_id": move.id,
+                                "reserve_area_id": reserve_area.id,
+                            }
+                        )