initial commit 2

2 files changed, 74 insertions, 0 deletions

diff --git a/addons/payment_fix_register_token/models/__init__.py b/addons/payment_fix_register_token/models/__init__.py
new file mode 100644
index 00000000..5f7620e1
--- /dev/null
+++ b/addons/payment_fix_register_token/models/__init__.py
@@ -0,0 +1,3 @@
+# -*- coding: utf-8 -*-
+
+from . import account_payment_register
diff --git a/addons/payment_fix_register_token/models/account_payment_register.py b/addons/payment_fix_register_token/models/account_payment_register.py
new file mode 100644
index 00000000..09359ea3
--- /dev/null
+++ b/addons/payment_fix_register_token/models/account_payment_register.py
@@ -0,0 +1,71 @@
+# -*- coding: utf-8 -*-
+
+from odoo import models, fields, api, _
+
+
+class AccountPaymentRegister(models.TransientModel):
+    _inherit = 'account.payment.register'
+
+    # == Business fields ==
+    payment_token_id = fields.Many2one(
+        comodel_name='payment.token',
+        string="Saved payment token",
+        store=True, readonly=False,
+        compute='_compute_payment_token_id',
+        domain='''[
+            (payment_method_code == 'electronic', '=', 1),
+            ('company_id', '=', company_id),
+            ('acquirer_id.capture_manually', '=', False),
+            ('acquirer_id.journal_id', '=', journal_id),
+            ('partner_id', 'in', suitable_payment_token_partner_ids),
+        ]''',
+        help="Note that tokens from acquirers set to only authorize transactions (instead of capturing the amount) are "
+             "not available.")
+
+    # == Display purpose fields ==
+    suitable_payment_token_partner_ids = fields.Many2many(
+        comodel_name='res.partner',
+        compute='_compute_suitable_payment_token_partner_ids')
+    payment_method_code = fields.Char(
+        related='payment_method_id.code')
+
+    # -------------------------------------------------------------------------
+    # COMPUTE METHODS
+    # -------------------------------------------------------------------------
+
+    @api.depends('can_edit_wizard')
+    def _compute_suitable_payment_token_partner_ids(self):
+        for wizard in self:
+            if wizard.can_edit_wizard:
+                lines = wizard._get_batches()[0]['lines']
+                partners = lines.partner_id
+                commercial_partners = partners.commercial_partner_id
+                children_partners = commercial_partners.child_ids
+                wizard.suitable_payment_token_partner_ids = (partners + commercial_partners + children_partners)._origin
+            else:
+                wizard.suitable_payment_token_partner_ids = False
+
+    @api.onchange('can_edit_wizard', 'payment_method_id', 'journal_id')
+    def _compute_payment_token_id(self):
+        for wizard in self:
+            if wizard.can_edit_wizard \
+                    and wizard.payment_method_id.code == 'electronic' \
+                    and wizard.journal_id \
+                    and wizard.suitable_payment_token_partner_ids:
+                wizard.payment_token_id = self.env['payment.token'].search([
+                    ('partner_id', 'in', wizard.suitable_payment_token_partner_ids.ids),
+                    ('acquirer_id.capture_manually', '=', False),
+                    ('acquirer_id.journal_id', '=', wizard.journal_id.id),
+                 ], limit=1)
+            else:
+                wizard.payment_token_id = False
+
+    # -------------------------------------------------------------------------
+    # BUSINESS METHODS
+    # -------------------------------------------------------------------------
+
+    def _create_payment_vals_from_wizard(self):
+        # OVERRIDE
+        payment_vals = super()._create_payment_vals_from_wizard()
+        payment_vals['payment_token_id'] = self.payment_token_id.id
+        return payment_vals