[UPD] edi_purchase_oca: Update for edi configuration

edi_purchase_oca/__init__.py

@@ -1 +1,2 @@
 from . import models
+from . import components

edi_purchase_oca/__manifest__.py

@@ -10,6 +10,11 @@
     "author": "ForgeFlow, Odoo Community Association (OCA)",
     "website": "https://github.com/OCA/edi",
     "depends": ["purchase", "edi_oca", "component_event"],
-    "data": ["views/purchase_order_views.xml", "views/edi_exchange_record_views.xml"],
+    "data": [
+        "views/purchase_order_views.xml",
+        "views/edi_exchange_record_views.xml",
+        "views/res_partner_view.xml",
+        "data/edi_configuration.xml",
+    ],
     "demo": [],
 }

edi_purchase_oca/components/__init__.py

@@ -0,0 +1 @@
+from . import listener_purchase_order

edi_purchase_oca/components/listener_purchase_order.py

@@ -0,0 +1,41 @@
+# Copyright 2024 Camptocamp SA
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+
+from odoo.addons.component.core import Component
+
+
+class EDIConfigPurchaseListener(Component):
+    _name = "edi.config.consumer.listener.purchase.order"
+    _inherit = "base.event.listener"
+    _apply_on = ["purchase.order"]
+
+    def on_button_confirm_purchase_order(self, record):
+        trigger = "on_button_confirm_purchase_order"
+        for rec in record:
+            confs = self.env["edi.configuration"].edi_get_conf(
+                trigger, rec._name, rec.partner_id
+            )
+            for conf in confs:
+                conf.edi_exec_snippet_do(rec)
+
+    def on_record_write(self, record, fields=None, **kwargs):
+        trigger = "on_record_write"
+        if kwargs.get("vals", False):
+            for rec in record:
+                confs = self.env["edi.configuration"].edi_get_conf(
+                    trigger, rec._name, rec.partner_id
+                )
+                for conf in confs:
+                    conf.edi_exec_snippet_do(rec, **kwargs)
+
+    def on_record_create(self, record, fields=None, **kwargs):
+        trigger = "on_record_create"
+        val_list = kwargs.get("vals", False)
+        if val_list:
+            for rec, vals in zip(record, val_list):
+                kwargs["vals"] = {rec.id: vals}
+                confs = self.env["edi.configuration"].edi_get_conf(
+                    trigger, rec._name, rec.partner_id
+                )
+                for conf in confs:
+                    conf.edi_exec_snippet_do(rec, **kwargs)

edi_purchase_oca/data/edi_configuration.xml

@@ -6,14 +6,14 @@
         <field name="trigger">on_button_confirm_purchase_order</field>
         <field name="model" ref="purchase.model_purchase_order" />
         <field name="snippet_before_do">
-            result={
-                "snippet_var_do": {}
-            }
+result={
+    "snippet_var_do": {}
+}
         </field>
         <field name="snippet_do" />
     </record>
     <record id="edi_conf_send_via_email" model="edi.configuration">
-        <field name="name">Send EDI Quoctation Config</field>
+        <field name="name">Send EDI Quotation Config</field>
         <field name="code">send_via_email_rfq</field>
         <field name="trigger">send_via_email_rfq</field>
         <field

edi_purchase_oca/models/__init__.py

@@ -1 +1,3 @@
 from . import purchase_order
+from . import res_partner
+from . import edi_configuration

edi_purchase_oca/models/edi_configuration.py

@@ -0,0 +1,15 @@
+# Copyright 2024 Camptocamp SA
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+
+from odoo import fields, models
+
+
+class EdiConfiguration(models.Model):
+    _inherit = "edi.configuration"
+
+    trigger = fields.Selection(
+        selection_add=[
+            ("on_button_confirm_purchase_order", "On Button Confirm Purchase Order"),
+            ("send_via_email_rfq", "Send via Email RFQ"),
+        ],
+    )

edi_purchase_oca/models/purchase_order.py

@@ -6,13 +6,16 @@
 
 class PurchaseOrder(models.Model):
     _name = "purchase.order"
-    _inherit = ["purchase.order", "edi.exchange.consumer.mixin"]
+    _inherit = [
+        "purchase.order",
+        "edi.exchange.consumer.mixin",
+        "edi.configuration.mixin",
+    ]
 
     def button_confirm(self):
-        result = super().button_confirm()
-        if self:
-            self._event("on_button_confirm_purchase_order").notify(self)
-        return result
+        res = super().button_confirm()
+        self._event("on_button_confirm_purchase_order").notify(self)
+        return res
 
     def button_cancel(self):
         result = super().button_cancel()

edi_purchase_oca/models/res_partner.py

@@ -14,6 +14,5 @@ class ResPartner(models.Model):
         relation="res_partner_edi_configuration_rel",
         column1="partner_id",
         column2="conf_id",
-        # TODO: Domain for Purchase model
-        domain="[('model_name', '=', 'purchase.order')]"
+        domain="[('model_name', '=', 'purchase.order')]",
     )

edi_purchase_oca/tests/__init__.py

@@ -0,0 +1 @@
+from . import test_edi_configuration

edi_purchase_oca/tests/test_edi_configuration.py

@@ -0,0 +1,85 @@
+# Copyright 2024 CamptoCamp SA,
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+
+import mock
+
+from odoo.exceptions import UserError
+
+from odoo.addons.edi_oca.tests.common import EDIBackendCommonComponentRegistryTestCase
+
+
+class TestsPurchaseEDIConfiguration(EDIBackendCommonComponentRegistryTestCase):
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+        cls._load_module_components(cls, "edi_purchase_oca")
+        cls.edi_configuration = cls.env["edi.configuration"]
+        cls.purchase_order = cls.env["purchase.order"]
+        cls.product = cls.env["product.product"].create(
+            {
+                "name": "Product 1",
+                "default_code": "1234567",
+            }
+        )
+
+    def setUp(self):
+        super().setUp()
+        self.create_config = self.edi_configuration.create(
+            {
+                "name": "Create Config",
+                "active": True,
+                "code": "create_config",
+                "backend_id": self.backend.id,
+                "type_id": self.exchange_type_out.id,
+                "trigger": "on_record_create",
+                "model": self.env["ir.model"]._get_id("purchase.order"),
+            }
+        )
+
+    def test_edi_configuration_snippet_before_do(self):
+        self.create_config.snippet_before_do = "record.button_cancel()"
+        order = self.purchase_order.create(
+            {
+                "partner_id": self.partner.id,
+                "order_line": [
+                    (
+                        0,
+                        0,
+                        {
+                            "product_id": self.product.id,
+                            "product_qty": 10,
+                            "price_unit": 100.0,
+                        },
+                    )
+                ],
+            }
+        )
+        self.assertTrue(order)
+        self.assertEqual(order.state, "cancel")
+
+    def test_edi_configuration_snippet_do(self):
+        self.create_config.snippet_do = "record._edi_send_via_edi(conf.type_id)"
+        with mock.patch.object(
+            type(self.backend), "exchange_generate", return_value=True
+        ):
+
+            with self.assertRaises(UserError) as err:
+                self.purchase_order.create(
+                    {
+                        "partner_id": self.partner.id,
+                        "order_line": [
+                            (
+                                0,
+                                0,
+                                {
+                                    "product_id": self.product.id,
+                                    "product_qty": 10,
+                                    "price_unit": 100.0,
+                                },
+                            )
+                        ],
+                    }
+                )
+            self.assertRegex(
+                err.exception.args[0], r"Record ID=\d+ has no file to send!"
+            )

edi_purchase_oca/views/res_partner_view.xml

@@ -1,4 +1,4 @@
-<?xml version="1.0" encoding="utf-8"?>
+<?xml version="1.0" encoding="utf-8" ?>
 <odoo>
     <record id="view_partner_form_inherit" model="ir.ui.view">
         <field name="name">res.partner.form.inherit.sales_purchases</field>
@@ -7,8 +7,10 @@
         <field name="arch" type="xml">
             <xpath expr="//page[@name='sales_purchases']" position="inside">
                 <group string="Edi Configuration">
-                    <field name="edi_purchase_conf_ids"
-                        context="{'default_model_name': 'purchase.order', 'default_partner_ids': active_ids}" />
+                    <field
+                        name="edi_purchase_conf_ids"
+                        context="{'default_model_name': 'purchase.order', 'default_partner_ids': active_ids}"
+                    />
                 </group>
             </xpath>
         </field>