[FIX] website_sale_product_assortment: dont restrict other assortment…

… products
OCA · Oct 15, 2024 · 6229f85 · 6229f85
1 parent 0041f97
commit 6229f85
Showing 1 changed file with 15 additions and 6 deletions.
diff --git a/website_sale_product_assortment/models/product_template.py b/website_sale_product_assortment/models/product_template.py
@@ -23,14 +23,23 @@ def get_product_assortment_restriction_info(self, product_ids):
                 ]
             )
         )
-        assortment_dict = {}
+
+        allowed_product_ids = set()
         for assortment in assortments:
             if partner & assortment.with_context(active_test=False).all_partner_ids:
-                allowed_product_ids = assortment.all_product_ids.ids
-                for product in product_ids:
-                    if product not in allowed_product_ids:
-                        assortment_dict.setdefault(product, self.env["ir.filters"])
-                        assortment_dict[product] |= assortment
+                allowed_product_ids.update(assortment.all_product_ids.ids)
+
+        assortment_dict = {}
+        for product in product_ids:
+            if product not in allowed_product_ids:
+                assortment_dict.setdefault(product, self.env["ir.filters"])
+                for assortment in assortments:
+                    if (
+                        partner
+                        & assortment.with_context(active_test=False).all_partner_ids
+                    ):
+                        if product not in assortment.all_product_ids.ids:
+                            assortment_dict[product] |= assortment
         return assortment_dict
 
     def _get_combination_info(