[FIX] stock_orderpoint_origin: support multiwarehouse case

foodles-tech · Mar 22, 2022 · d022a47 · d022a47
1 parent 2337653
commit d022a47
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/stock_orderpoint_origin/models/procurement_group.py b/stock_orderpoint_origin/models/procurement_group.py
@@ -15,9 +15,12 @@ def run(self, procurements, raise_user_error=True):
         Forecast = self.env["report.stock.report_product_product_replenishment"]
         new_procurements = []
         for procurement in procurements:
+            ForecastWH = Forecast
+            warehouse = procurement.values.get("warehouse_id")
+            if warehouse:
+                ForecastWH = Forecast.with_context(warehouse=warehouse.id)
             product = procurement.product_id
-            # TODO: set warehouse_id in context?
-            data = Forecast._get_report_data(product_variant_ids=[product.id])
+            data = ForecastWH._get_report_data(product_variant_ids=[product.id])
             source_docs = set()  # Avoid duplicate sources
             for line in data["lines"]:
                 if not line["document_in"] and line["document_out"]: