Files
suite/account_invoice_change_analytic/wizard/invoice_change.py

61 lines
2.2 KiB
Python

from odoo import api, fields, models, _
class InvoiceChangeWizard(models.TransientModel):
_inherit = 'account.invoice.change'
analytic_account_id = fields.Many2one('account.analytic.account', string='Analytic Account')
set_analytic_account_id = fields.Boolean()
update_tags = fields.Selection(selection=[
('no', 'Do not update tags'),
('add', 'Add to tags'),
('set', 'Set tags'),
], default='no', string='Update Analytic Tags')
analytic_tag_ids = fields.Many2many('account.analytic.tag', string='Analytic Tags')
def _analytic_account_id(self, invoice):
analytics = invoice.invoice_line_ids.mapped('analytic_account_id')
if len(analytics):
return analytics[0].id
return False
@api.model
def default_get(self, fields):
rec = super(InvoiceChangeWizard, self).default_get(fields)
context = dict(self._context or {})
invoices = self.env['account.move'].browse(context.get('active_ids'))
if len(invoices) == 1:
rec.update({
'set_analytic_account_id': True,
'analytic_account_id': self._analytic_account_id(invoices),
'analytic_tag_ids': [(6, 0, invoices.invoice_line_ids.analytic_tag_ids.ids)],
})
return rec
def affect_change(self):
res = super(InvoiceChangeWizard, self).affect_change()
self._affect_analytic_change()
return res
def _prepare_analytic_values(self):
vals = {}
if self.set_analytic_account_id:
vals['analytic_account_id'] = self.analytic_account_id.id
tag_commands = []
if self.update_tags == 'add':
tag_commands = [(4, tag.id, 0) for tag in self.analytic_tag_ids]
if self.update_tags == 'set':
tag_commands = [(6, 0, self.analytic_tag_ids.ids)]
if tag_commands:
vals['analytic_tag_ids'] = tag_commands
if vals:
vals['analytic_line_ids'] = [(5, 0, 0)]
return vals
def _affect_analytic_change(self):
lines_to_affect = self.move_ids.invoice_line_ids
vals = self._prepare_analytic_values()
if vals:
lines_to_affect.write(vals)
lines_to_affect.create_analytic_lines()