account-financial-tools/account_spread_cost_revenue/models/account_invoice_line.py

# Copyright 2016-2019 Onestein (<https://www.onestein.eu>)
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).

from odoo import _, api, fields, models
from odoo.exceptions import UserError


class AccountInvoiceLine(models.Model):
    _inherit = 'account.invoice.line'

    spread_id = fields.Many2one(
        'account.spread',
        string='Spread Board',
        copy=False)
    spread_check = fields.Selection([
        ('linked', 'Linked'),
        ('unlinked', 'Unlinked'),
        ('unavailable', 'Unavailable')
    ], compute='_compute_spread_check')

    @api.depends('spread_id', 'invoice_id.state')
    def _compute_spread_check(self):
        for line in self:
            if line.spread_id:
                line.spread_check = 'linked'
            elif line.invoice_id.state == 'draft':
                line.spread_check = 'unlinked'
            else:
                line.spread_check = 'unavailable'

    @api.multi
    def spread_details(self):
        """Button on the invoice lines tree view of the invoice
        form to show the spread form view."""
        if not self:
            # In case the widget clicked before the creation of the line
            return

        if self.spread_id:
            return {
                'name': _('Spread Details'),
                'view_type': 'form',
                'view_mode': 'form',
                'res_model': 'account.spread',
                'type': 'ir.actions.act_window',
                'target': 'current',
                'readonly': False,
                'res_id': self.spread_id.id,
            }

        # In case no spread board is linked to the invoice line
        # open the wizard to link them
        company = self.invoice_id.company_id
        ctx = dict(
            self.env.context,
            default_invoice_line_id=self.id,
            default_company_id=company.id,
            allow_spread_planning=company.allow_spread_planning,
        )
        return {
            'name': _('Link Invoice Line with Spread Board'),
            'view_type': 'form',
            'view_mode': 'form',
            'res_model': 'account.spread.invoice.line.link.wizard',
            'type': 'ir.actions.act_window',
            'target': 'new',
            'context': ctx,
        }

    def create_auto_spread(self):
        """ Create auto spread table for each invoice line, when needed """

        def _filter_line(aline, iline):
            """ Find matching template auto line with invoice line """
            if aline.product_id and iline.product_id != aline.product_id:
                return False
            if aline.account_id and iline.account_id != aline.account_id:
                return False
            if aline.analytic_account_id and \
                    iline.account_analytic_id != aline.analytic_account_id:
                return False
            return True

        for line in self:
            if line.spread_check == 'linked':
                continue
            spread_type = (
                'sale' if line.invoice_type in ['out_invoice', 'out_refund']
                else 'purchase')
            spread_auto = self.env['account.spread.template.auto'].search(
                [('template_id.auto_spread', '=', True),
                 ('template_id.spread_type', '=', spread_type)])
            matched = spread_auto.filtered(lambda a, i=line: _filter_line(a, i))
            template = matched.mapped('template_id')
            if not template:
                continue
            elif len(template) > 1:
                raise UserError(
                    _('Too many auto spread templates (%s) matched with the '
                      'invoice line, %s') % (len(template), line.display_name))
            # Found auto spread template for this invoice line, create it
            wizard = self.env['account.spread.invoice.line.link.wizard'].new({
                'invoice_line_id': line.id,
                'company_id': line.company_id.id,
                'spread_action_type': 'template',
                'template_id': template.id,
            })
            wizard.confirm()