Files
suite/account_invoice_change/wizard/invoice_change.py

78 lines
3.1 KiB
Python

from odoo import api, fields, models, _
from odoo.exceptions import UserError
class InvoiceChangeWizard(models.TransientModel):
_name = 'account.invoice.change'
_description = 'Invoice Change'
move_ids = fields.Many2many('account.move', string='Invoice', readonly=True, required=True)
is_single_move = fields.Boolean(compute='_compute_move_company_id')
move_company_id = fields.Many2one('res.company', readonly=True, compute='_compute_move_company_id')
invoice_user_id = fields.Many2one('res.users', string='Salesperson')
set_invoice_user_id = fields.Boolean()
date = fields.Date(string='Accounting Date')
set_date = fields.Boolean()
invoice_date = fields.Date(string='Invoice Date')
set_invoice_date = fields.Boolean()
@api.model
def default_get(self, fields):
rec = super(InvoiceChangeWizard, self).default_get(fields)
context = dict(self._context or {})
active_model = context.get('active_model')
active_ids = context.get('active_ids')
# Checks on context parameters
if not active_model or not active_ids:
raise UserError(
_("Programmation error: wizard action executed without active_model or active_ids in context."))
if active_model != 'account.move':
raise UserError(_(
"Programmation error: the expected model for this action is 'account.move'. The provided one is '%d'.") % active_model)
# Checks on received invoice records
invoices = self.env[active_model].browse(active_ids)
if invoices.filtered(lambda i: not i.is_invoice()):
raise UserError(_('Only invoices can be updated.'))
if invoices.filtered(lambda i: i.state in ('draft', 'cancel')):
raise UserError(_('Only posted invoices can be updated.'))
if len(invoices.company_id) > 1:
raise UserError(_('Selected invoices must be in the same company.'))
rec['move_ids'] = [(6, 0, invoices.ids)]
if len(invoices) == 1:
rec.update({
'set_invoice_user_id': True,
'set_date': True,
'set_invoice_date': True,
'invoice_user_id': invoices.invoice_user_id.id,
'date': invoices.date,
'invoice_date': invoices.invoice_date,
})
return rec
@api.depends('move_ids')
def _compute_move_company_id(self):
for wiz in self:
company = wiz.move_ids.company_id
company.ensure_one()
wiz.move_company_id = company
wiz.is_single_move = len(wiz.move_ids) == 1
def _new_move_vals(self):
vals = {}
if self.set_invoice_user_id:
vals['invoice_user_id'] = self.invoice_user_id.id
if self.set_date:
vals['date'] = self.date
if self.set_invoice_date:
vals['invoice_date'] = self.invoice_date
return vals
def affect_change(self):
self.ensure_one()
vals = self._new_move_vals()
if vals:
self.move_ids.write(vals)
return True