[16.0][IMP] website_sale_product_assortment: Implement restrictions on the cart in _cart_update and shop_payment

- Add SaleOrder._cart_update validation to prevent adding restricted products
- Add WebsiteSale.shop_payment validation to block checkout with restricted products
- Add unit tests for both validation methods
- Add Spanish translations for error messages

Author: SuShanVoong

website_sale_product_assortment/controllers/website_sale.py

@@ -2,6 +2,8 @@
 # License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
 from werkzeug.exceptions import NotFound
 
+from odoo import _, http
+from odoo.exceptions import UserError
 from odoo.http import request, route
 
 from odoo.addons.website_sale.controllers.main import WebsiteSale
@@ -81,3 +83,25 @@ def _get_search_options(
                 ("product_variant_ids", "in", list(allowed_product_ids))
             ]
         return res
+
+    @http.route()
+    def shop_payment(self, **post):
+        order = request.website.sale_get_order()
+        restricted_product_names = []
+        for line in order.order_line.filtered(lambda l: not l.display_type):
+            if (
+                not line.product_id._show_quick_add_accesory_assortments()
+                and line.product_id.purchase_ok
+            ):
+                restricted_product_names.append(line.product_id.name)
+
+        if restricted_product_names:
+            restricted_products_str = ", ".join(restricted_product_names)
+            raise UserError(
+                _(
+                    "The following products cannot be added to the cart because "
+                    "they are restricted: %s. Please remove the product to continue."
+                )
+                % restricted_products_str
+            )
+        return super().shop_payment(**post)

website_sale_product_assortment/i18n/es.po

@@ -92,6 +92,14 @@ msgstr "No aplicar restricción"
 msgid "Filters"
 msgstr "Filtros"
 
+#. module: website_sale_product_assortment
+#. odoo-python
+#: code:addons/website_sale_product_assortment/models/sale_order.py:0
+#, python-format
+msgid "It cannot be added to the cart because the product is restricted."
+msgstr ""
+"No se puede añadir al carrito porque el producto está restringido."
+
 #. module: website_sale_product_assortment
 #: model:ir.model.fields,help:website_sale_product_assortment.field_ir_filters__message_unavailable
 msgid ""
@@ -119,6 +127,17 @@ msgstr "No disponible"
 msgid "Product Template"
 msgstr "Plantilla de producto"
 
+#. module: website_sale_product_assortment
+#. odoo-python
+#: code:addons/website_sale_product_assortment/controllers/website_sale.py:0
+#, python-format
+msgid ""
+"The following products cannot be added to the cart because they are "
+"restricted: %s. Please remove the product to continue."
+msgstr ""
+"Los siguientes productos no se pueden añadir al carrito porque están "
+"restringidos: %s. Por favor, elimínelos para continuar."
+
 #. module: website_sale_product_assortment
 #. openerp-web
 #: code:addons/website_sale_product_assortment/static/src/xml/website_sale_product_assortment.xml:0

website_sale_product_assortment/i18n/website_sale_product_assortment.pot

@@ -70,6 +70,13 @@ msgstr ""
 msgid "Filters"
 msgstr ""
 
+#. module: website_sale_product_assortment
+#. odoo-python
+#: code:addons/website_sale_product_assortment/models/sale_order.py:0
+#, python-format
+msgid "It cannot be added to the cart because the product is restricted."
+msgstr ""
+
 #. module: website_sale_product_assortment
 #: model:ir.model.fields,help:website_sale_product_assortment.field_ir_filters__message_unavailable
 msgid ""
@@ -95,6 +102,15 @@ msgstr ""
 msgid "Product"
 msgstr ""
 
+#. module: website_sale_product_assortment
+#. odoo-python
+#: code:addons/website_sale_product_assortment/controllers/website_sale.py:0
+#, python-format
+msgid ""
+"The following products cannot be added to the cart because they are "
+"restricted: %s. Please remove the product to continue."
+msgstr ""
+
 #. module: website_sale_product_assortment
 #. odoo-javascript
 #: code:addons/website_sale_product_assortment/static/src/xml/website_sale_product_assortment.xml:0

website_sale_product_assortment/models/__init__.py

@@ -2,3 +2,4 @@
 from . import product_template
 from . import website_snippet_filter
 from . import product_product
+from . import sale_order

website_sale_product_assortment/models/product_product.py

@@ -5,6 +5,7 @@ class ProductProduct(models.Model):
     _inherit = "product.product"
 
     def _show_quick_add_accesory_assortments(self):
+        self.ensure_one()
         res = self._website_show_quick_add()
         if not res:
             return res

website_sale_product_assortment/models/product_template.py

@@ -95,21 +95,18 @@ def _get_allowed_show_products(self, products):
         if not restricted_products_dict:
             return products
 
-        allowed_show_products = products.filtered(
+        return products.filtered(
             lambda p: p.id not in restricted_products_dict
             or not any(
                 assortment.website_availability == "no_show"
                 for assortment in restricted_products_dict[p.id]
             )
         )
-        return allowed_show_products
 
     def _get_website_accessory_product(self):
-        res = super()._get_website_accessory_product()
-        visible_products = self._get_allowed_show_products(res)
-        return visible_products
+        return self._get_allowed_show_products(super()._get_website_accessory_product())
 
     def _get_website_alternative_product(self):
-        res = super()._get_website_alternative_product()
-        visible_products = self._get_allowed_show_products(res)
-        return visible_products
+        return self._get_allowed_show_products(
+            super()._get_website_alternative_product()
+        )

website_sale_product_assortment/models/sale_order.py

@@ -0,0 +1,22 @@
+from odoo import _, models
+from odoo.exceptions import UserError
+
+
+class SaleOrder(models.Model):
+    _inherit = "sale.order"
+
+    def _cart_update(self, product_id, line_id=None, add_qty=0, set_qty=0, **kwargs):
+        product = self.env["product.product"].browse(product_id)
+        if not product._show_quick_add_accesory_assortments() and not (
+            add_qty == 0 or (not add_qty and set_qty == 0)
+        ):
+            raise UserError(
+                _("It cannot be added to the cart because the product is restricted.")
+            )
+        return super()._cart_update(
+            product_id,
+            line_id=line_id,
+            add_qty=add_qty,
+            set_qty=set_qty,
+            **kwargs,
+        )

website_sale_product_assortment/tests/test_ui.py

@@ -1,5 +1,8 @@
 # Copyright 2021 Tecnativa - Carlos Roca
 # License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
+from unittest.mock import patch
+
+from odoo.exceptions import UserError
 from odoo.tests import tagged
 from odoo.tests.common import HttpCase
 
@@ -25,6 +28,20 @@ def setUp(self):
             }
         )
 
+        # Configuración adicional para tests unitarios
+        self.partner = self.env["res.partner"].create(
+            {
+                "name": "Test Partner",
+                "email": "test@example.com",
+            }
+        )
+
+        self.sale_order = self.env["sale.order"].create(
+            {
+                "partner_id": self.partner.id,
+            }
+        )
+
     def test_01_ui_no_restriction(self):
         self.env["ir.filters"].create(
             {
@@ -98,3 +115,31 @@ def test_04_ui_no_restriction_no_show(self):
         self.start_tour(
             "/shop", "test_assortment_with_no_restriction_no_show", login="admin"
         )
+
+    def test_05_ui_cart_update_restricted_product(self):
+        product_patch_path = (
+            "odoo.addons.website_sale_product_assortment.models.product_product."
+            "ProductProduct._show_quick_add_accesory_assortments"
+        )
+        with patch(product_patch_path, return_value=False):
+            with self.assertRaises(UserError) as cm:
+                self.sale_order._cart_update(
+                    product_id=self.product.product_variant_id.id,
+                    add_qty=1,
+                    set_qty=1,
+                )
+            self.assertIn(
+                "It cannot be added to the cart because the product is restricted.",
+                str(cm.exception),
+            )
+
+    def test_06_ui_cart_update_allowed_product(self):
+        product_patch_path = (
+            "odoo.addons.website_sale_product_assortment.models.product_product."
+            "ProductProduct._show_quick_add_accesory_assortments"
+        )
+        with patch(product_patch_path, return_value=True):
+            result = self.sale_order._cart_update(
+                product_id=self.product.product_variant_id.id, add_qty=1
+            )
+            self.assertIsInstance(result, dict)