flectra/addons/l10n_ae_extend/models/sale.py

# Part of Flectra. See LICENSE file for full copyright and licensing details.

from flectra import fields, models, api


class SaleOrder(models.Model):
    _inherit = 'sale.order'

    @api.multi
    def _default_config_type(self):
        return self.vat_config_type.search([
            ('journal_id.type', '=', 'sale')], limit=1)

    vat_config_type = fields.Many2one(
        'vat.config.type', 'VAT Type', default=_default_config_type,
        readonly=True, states={'draft': [('readonly', False)]})

    @api.multi
    def _prepare_invoice(self):
        invoice_vals = super(SaleOrder, self)._prepare_invoice()
        invoice_vals.update({
            'vat_config_type': self.vat_config_type.id,
            'journal_id': self.vat_config_type.journal_id.id,
            'account_id':
                self.vat_config_type.journal_id.default_debit_account_id.id,
            })
        return invoice_vals