l10n-switzerland-flectra/l10n_ch_pain_base/models/account_move_line.py

# copyright 2016 Akretion (Alexis de Lattre <alexis.delattre@akretion.com>)
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).

from flectra import models, api


class AccountMoveLine(models.Model):
    _inherit = 'account.move.line'

    def _prepare_payment_line_vals(self, payment_order):
        vals = super(AccountMoveLine, self)._prepare_payment_line_vals(
            payment_order)
        if self.invoice_id and self.invoice_id.reference_type in ['isr', 'QRR', 'SCOR']:
            if self.invoice_id.reference_type == 'isr':
                vals['local_instrument'] = 'CH01'
            if vals['communication']:
                vals['communication'] = vals['communication'].replace(' ', '')
        return vals