Merge 6ed1df2 into ebffb28

odoomrp · Apr 14, 2016 · c83bb99 · c83bb99
2 parents ebffb28 + 6ed1df2
commit c83bb99
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/stock_planning/wizard/wiz_stock_planning.py b/stock_planning/wizard/wiz_stock_planning.py
@@ -81,8 +81,12 @@ def calculate_stock_planning(self):
         for procurement in proc_obj._find_procurements_from_stock_planning(
             self.company, fdate, category=self.category,
                 template=self.template, product=self.product, periods=True):
-            product_datas = self._find_product_in_table(
-                product_datas, procurement.product_id, procurement.location_id)
+            if (not self.locations or
+                (self.locations and procurement.location_id.id in
+                 self.locations.ids)):
+                product_datas = self._find_product_in_table(
+                    product_datas, procurement.product_id,
+                    procurement.location_id)
         self._generate_stock_planning(product_datas)
         return {'name': _('Stock Planning'),
                 'type': 'ir.actions.act_window',