[12.0][ADD] - Add new module Contract Transmit Method

contract_transmit_method/models/__init__.py

@@ -0,0 +1,2 @@
+from . import contract_abstract
+from . import contract_contract

contract_transmit_method/models/contract_abstract.py

@@ -0,0 +1,30 @@
+# Copyright 2020 ACSONE SA/NV
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+
+from odoo import api, fields, models
+
+
+class ContractAbstractContract(models.AbstractModel):
+
+    _inherit = "contract.abstract.contract"
+
+    transmit_method_id = fields.Many2one(
+        comodel_name="transmit.method",
+        string="Transmission Method",
+        track_visibility="onchange",
+        ondelete="restrict",
+    )
+
+    @api.onchange("partner_id", "company_id")
+    def onchange_partner_transmit_method(self):
+        if self.partner_id and self.contract_type:
+            if self.contract_type == "sale":
+                self.transmit_method_id = (
+                    self.partner_id.customer_invoice_transmit_method_id.id
+                    or False
+                )
+            else:
+                self.transmit_method_id = (
+                    self.partner_id.supplier_invoice_transmit_method_id.id
+                    or False
+                )

contract_transmit_method/models/contract_contract.py

@@ -0,0 +1,16 @@
+# Copyright 2020 ACSONE SA/NV
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+
+from odoo import api, models
+
+
+class ContractContract(models.Model):
+
+    _inherit = "contract.contract"
+
+    @api.multi
+    def _prepare_invoice(self, date_invoice, journal=None):
+        invoice_vals = super()._prepare_invoice(date_invoice, journal=journal)
+        if self.transmit_method_id:
+            invoice_vals["transmit_method_id"] = self.transmit_method_id.id
+        return invoice_vals