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()