Odoo18-Base/addons/account/wizard/account_move_send_wizard.py
2025-01-06 10:57:38 +07:00

300 lines
12 KiB
Python

from odoo import api, fields, models
from odoo.tools.misc import get_lang
class AccountMoveSendWizard(models.TransientModel):
"""Wizard that handles the sending a single invoice."""
_name = 'account.move.send.wizard'
_inherit = ['account.move.send']
_description = "Account Move Send Wizard"
move_id = fields.Many2one(comodel_name='account.move', required=True)
company_id = fields.Many2one(comodel_name='res.company', related='move_id.company_id')
alerts = fields.Json(compute='_compute_alerts')
sending_methods = fields.Json(
compute='_compute_sending_methods',
inverse='_inverse_sending_methods',
)
sending_method_checkboxes = fields.Json(
compute='_compute_sending_method_checkboxes',
precompute=True,
readonly=False,
store=True,
)
extra_edis = fields.Json(
compute='_compute_extra_edis',
inverse='_inverse_extra_edis',
)
extra_edi_checkboxes = fields.Json(
compute='_compute_extra_edi_checkboxes',
precompute=True,
readonly=False,
store=True,
)
invoice_edi_format = fields.Selection(
selection=lambda self: self.env['res.partner']._fields['invoice_edi_format'].selection,
compute='_compute_invoice_edi_format',
)
pdf_report_id = fields.Many2one(
comodel_name='ir.actions.report',
string="Invoice template",
domain="[('is_invoice_report', '=', True)]",
compute='_compute_pdf_report_id',
readonly=False,
store=True,
)
display_pdf_report_id = fields.Boolean(compute='_compute_display_pdf_report_id')
is_download_only = fields.Boolean(compute='_compute_is_download_only')
# MAIL
mail_template_id = fields.Many2one(
comodel_name='mail.template',
string="Email template",
domain="[('model', '=', 'account.move')]",
compute='_compute_mail_template_id',
readonly=False,
store=True,
)
mail_lang = fields.Char(compute='_compute_mail_lang')
mail_partner_ids = fields.Many2many(
comodel_name='res.partner',
string="Recipients",
compute='_compute_mail_subject_body_partners',
store=True,
readonly=False,
)
mail_subject = fields.Char(
string="Subject",
compute='_compute_mail_subject_body_partners',
store=True,
readonly=False,
)
mail_body = fields.Html(
string="Contents",
sanitize_style=True,
compute='_compute_mail_subject_body_partners',
store=True,
readonly=False,
)
mail_attachments_widget = fields.Json(
compute='_compute_mail_attachments_widget',
store=True,
readonly=False,
)
# -------------------------------------------------------------------------
# DEFAULTS
# -------------------------------------------------------------------------
@api.model
def default_get(self, fields_list):
# EXTENDS 'base'
results = super().default_get(fields_list)
if 'move_id' in fields_list and 'move_id' not in results:
move_id = self._context.get('active_ids', [])[0]
results['move_id'] = move_id
return results
# -------------------------------------------------------------------------
# COMPUTE METHODS
# -------------------------------------------------------------------------
@api.depends('sending_methods', 'extra_edis')
def _compute_alerts(self):
for wizard in self:
move_data = {
wizard.move_id: {
'sending_methods': wizard.sending_methods or {},
'invoice_edi_format': wizard.invoice_edi_format,
'extra_edis': wizard.extra_edis or {},
}
}
wizard.alerts = self._get_alerts(wizard.move_id, move_data)
@api.depends('sending_method_checkboxes')
def _compute_sending_methods(self):
for wizard in self:
wizard.sending_methods = self._get_selected_checkboxes(wizard.sending_method_checkboxes)
def _inverse_sending_methods(self):
for wizard in self:
wizard.sending_method_checkboxes = {method_key: {'checked': True} for method_key in wizard.sending_methods or {}}
@api.depends('move_id')
def _compute_sending_method_checkboxes(self):
""" Select one applicable sending method given the following priority
1. preferred method set on partner,
2. email,
3. manual.
"""
methods = self.env['ir.model.fields'].get_field_selection('res.partner', 'invoice_sending_method')
for wizard in self:
preferred_method = self._get_default_sending_method(wizard.move_id)
need_fallback = not self._is_applicable_to_move(preferred_method, wizard.move_id)
fallback_method = need_fallback and ('email' if self._is_applicable_to_move('email', wizard.move_id) else 'manual')
wizard.sending_method_checkboxes = {
method_key: {
'checked': method_key == preferred_method if not need_fallback else method_key == fallback_method,
'label': method_label,
}
for method_key, method_label in methods if self._is_applicable_to_company(method_key, wizard.company_id)
}
@api.depends('extra_edi_checkboxes')
def _compute_extra_edis(self):
for wizard in self:
wizard.extra_edis = self._get_selected_checkboxes(wizard.extra_edi_checkboxes)
def _inverse_extra_edis(self):
for wizard in self:
wizard.extra_edi_checkboxes = {method_key: {'checked': True} for method_key in wizard.extra_edis or {}}
@api.depends('move_id')
def _compute_extra_edi_checkboxes(self):
all_extra_edis = self._get_all_extra_edis()
for wizard in self:
wizard.extra_edi_checkboxes = {
edi_key: {'checked': True, 'label': all_extra_edis[edi_key]['label'], 'help': all_extra_edis[edi_key].get('help')}
for edi_key in self._get_default_extra_edis(wizard.move_id)
}
def _compute_invoice_edi_format(self):
for wizard in self:
wizard.invoice_edi_format = self._get_default_invoice_edi_format(wizard.move_id)
@api.depends('move_id')
def _compute_pdf_report_id(self):
for wizard in self:
wizard.pdf_report_id = self._get_default_pdf_report_id(wizard.move_id)
@api.depends('move_id')
def _compute_display_pdf_report_id(self):
# show pdf template menu if there are more than 1 template available and there is at least one move that needs a pdf
available_templates_count = self.env['ir.actions.report'].search_count([('is_invoice_report', '=', True)], limit=2)
for wizard in self:
wizard.display_pdf_report_id = available_templates_count > 1 and not wizard.move_id.invoice_pdf_report_id
@api.depends('sending_methods')
def _compute_is_download_only(self):
for wizard in self:
wizard.is_download_only = wizard.sending_methods == ['manual']
@api.depends('move_id')
def _compute_mail_template_id(self):
for wizard in self:
wizard.mail_template_id = self._get_default_mail_template_id(wizard.move_id)
@api.depends('mail_template_id')
def _compute_mail_lang(self):
for wizard in self:
wizard.mail_lang = self._get_default_mail_lang(wizard.move_id, wizard.mail_template_id) if wizard.mail_template_id else get_lang(self.env).code
@api.depends('mail_template_id', 'mail_lang')
def _compute_mail_subject_body_partners(self):
for wizard in self:
if wizard.mail_template_id:
wizard.mail_subject = self._get_default_mail_subject(wizard.move_id, wizard.mail_template_id, wizard.mail_lang)
wizard.mail_body = self._get_default_mail_body(wizard.move_id, wizard.mail_template_id, wizard.mail_lang)
wizard.mail_partner_ids = self._get_default_mail_partner_ids(wizard.move_id, wizard.mail_template_id, wizard.mail_lang)
else:
wizard.mail_subject = wizard.mail_body = wizard.mail_partner_ids = None
@api.depends('mail_template_id', 'sending_methods', 'extra_edis')
def _compute_mail_attachments_widget(self):
for wizard in self:
manual_attachments_data = [x for x in wizard.mail_attachments_widget or [] if x.get('manual')]
wizard.mail_attachments_widget = (
self._get_default_mail_attachments_widget(
wizard.move_id,
wizard.mail_template_id,
extra_edis=wizard.extra_edis or {},
pdf_report=wizard.pdf_report_id,
)
+ manual_attachments_data
)
# -------------------------------------------------------------------------
# CONSTRAINS
# -------------------------------------------------------------------------
@api.constrains('move_id')
def _check_move_id_constrains(self):
for wizard in self:
self._check_move_constrains(wizard.move_id)
# -------------------------------------------------------------------------
# HELPERS
# -------------------------------------------------------------------------
@api.model
def _get_selected_checkboxes(self, json_checkboxes):
if not json_checkboxes:
return {}
return [checkbox_key for checkbox_key, checkbox_vals in json_checkboxes.items() if checkbox_vals['checked']]
# -------------------------------------------------------------------------
# BUSINESS METHODS
# -------------------------------------------------------------------------
def _get_sending_settings(self):
self.ensure_one()
send_settings = {
'sending_methods': self.sending_methods or [],
'invoice_edi_format': self.invoice_edi_format,
'extra_edis': self.extra_edis or [],
'pdf_report': self.pdf_report_id,
'author_user_id': self.env.user.partner_id.id,
'author_partner_id': self.env.user.id,
}
if self.sending_methods and 'email' in self.sending_methods:
send_settings.update({
'mail_template': self.mail_template_id,
'mail_lang': self.mail_lang,
'mail_body': self.mail_body,
'mail_subject': self.mail_subject,
'mail_partner_ids': self.mail_partner_ids.ids,
'mail_attachments_widget': self.mail_attachments_widget,
})
return send_settings
def _update_preferred_settings(self):
"""If the partner's settings are not set, we use them as partner's default."""
self.ensure_one()
if (
self.sending_methods
and len(self.sending_methods) == 1
and not self.move_id.partner_id.with_company(self.company_id).invoice_sending_method
):
self.move_id.partner_id.with_company(self.company_id).invoice_sending_method = self.sending_methods[0]
if not self.move_id.partner_id.invoice_template_pdf_report_id and self.pdf_report_id != self._get_default_pdf_report_id(self.move_id):
self.move_id.partner_id.invoice_template_pdf_report_id = self.pdf_report_id
# -------------------------------------------------------------------------
# BUSINESS ACTIONS
# -------------------------------------------------------------------------
@api.model
def _action_download(self, attachments):
""" Download the PDF attachment, or a zip of attachments if there are more than one. """
return {
'type': 'ir.actions.act_url',
'url': f'/account/download_invoice_attachments/{",".join(map(str, attachments.ids))}',
'close': True,
}
def action_send_and_print(self, allow_fallback_pdf=False):
""" Create invoice documents and send them."""
self.ensure_one()
if self.alerts:
self._raise_danger_alerts(self.alerts)
self._update_preferred_settings()
attachments = self._generate_and_send_invoices(
self.move_id,
**self._get_sending_settings(),
allow_fallback_pdf=allow_fallback_pdf,
)
if attachments and self.sending_methods and 'manual' in self.sending_methods:
return self._action_download(attachments)
else:
return {'type': 'ir.actions.act_window_close'}