mirror of https://github.com/OCA/social.git
60 lines
2.4 KiB
Python
60 lines
2.4 KiB
Python
# Copyright 2021 ACSONE SA/NV (<http://acsone.eu>)
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
|
|
import base64
|
|
|
|
from odoo import _, exceptions, fields, models
|
|
|
|
|
|
class MailTemplate(models.Model):
|
|
_inherit = "mail.template"
|
|
|
|
template_report_ids = fields.One2many(
|
|
comodel_name="mail.template.report", inverse_name="mail_template_id",
|
|
)
|
|
|
|
# pylint: disable=redefined-outer-name
|
|
def generate_email(self, res_ids, fields=None):
|
|
"""
|
|
Inherit to generate attachments.
|
|
Inspired from original mail.template,generate_email(...) from Odoo.
|
|
:param res_ids: int or list of int
|
|
:param fields:
|
|
:return: dict
|
|
"""
|
|
self.ensure_one()
|
|
multi_mode = True
|
|
results = super(MailTemplate, self).generate_email(res_ids, fields=fields)
|
|
if not self.template_report_ids:
|
|
return results
|
|
if isinstance(res_ids, int):
|
|
multi_mode = False
|
|
results = {res_ids: results}
|
|
# Generate attachments (inspired from Odoo); Just add new attachments
|
|
# into 'attachments' key
|
|
for res_id, values in results.items():
|
|
attachments = values.setdefault("attachments", [])
|
|
for template_report in self.template_report_ids:
|
|
report_name = self._render_template(
|
|
template_report.report_name, template_report.model, res_id
|
|
)
|
|
report = template_report.report_template_id
|
|
report_service = report.report_name
|
|
|
|
if report.report_type in ["qweb-html", "qweb-pdf"]:
|
|
result, report_format = report.render_qweb_pdf([res_id])
|
|
else:
|
|
res = report.render([res_id])
|
|
if not res:
|
|
raise exceptions.UserError(
|
|
_("Unsupported report type %s found.") % report.report_type
|
|
)
|
|
result, report_format = res
|
|
result = base64.b64encode(result)
|
|
if not report_name:
|
|
report_name = "report." + report_service
|
|
ext = "." + report_format
|
|
if not report_name.endswith(ext):
|
|
report_name += ext
|
|
attachments.append((report_name, result))
|
|
return multi_mode and results or results[res_ids]
|