Skip to content

Commit

Permalink
[14.0][IMP] export_async_schedule : Send attachments directly
Browse files Browse the repository at this point in the history
  • Loading branch information
quoc-pn committed Dec 6, 2023
1 parent 306093d commit 910538b
Show file tree
Hide file tree
Showing 4 changed files with 54 additions and 9 deletions.
29 changes: 22 additions & 7 deletions base_export_async/models/delay_export.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,9 @@ def _get_file_content(self, params):
return xls.from_data(columns_headers, import_data)

@api.model
def export(self, params):
def export(
self, params, mail_template=None, is_export_file_attached_to_email=False
):
"""Delayed export of a file sent by email
The ``params`` is a dict of parameters, contains:
Expand Down Expand Up @@ -124,14 +126,27 @@ def export(self, params):
"model_description": model_description,
}
)
email_values = {
"email_from": email_from,
"reply_to": email_from,
"recipient_ids": [(6, 0, users.mapped("partner_id").ids)],
}

email_template = (
mail_template
if mail_template
else self.env.ref("base_export_async.delay_export_mail_template")
)

self.env.ref("base_export_async.delay_export_mail_template").send_mail(
if is_export_file_attached_to_email:
attachment_ids = attachment.ids
if email_template and email_template.attachment_ids:
attachment_ids += email_template.attachment_ids.ids

Check warning on line 144 in base_export_async/models/delay_export.py

View check run for this annotation

Codecov / codecov/patch

base_export_async/models/delay_export.py#L144

Added line #L144 was not covered by tests
attachment_ids = [(4, aid) for aid in attachment_ids]
email_values.update({"attachment_ids": attachment_ids})
email_template.send_mail(
export_record.id,
email_values={
"email_from": email_from,
"reply_to": email_from,
"recipient_ids": [(6, 0, users.mapped("partner_id").ids)],
},
email_values=email_values,
)

@api.model
Expand Down
11 changes: 11 additions & 0 deletions base_export_async/tests/test_base_export_async.py
Original file line number Diff line number Diff line change
Expand Up @@ -81,6 +81,17 @@ def test_export_xls(self):
self.assertEqual(len(new_mail), 1)
self.assertEqual(new_attachment.name, "res.partner.xls")

def test_export_mail_attachments(self):
"""Check that the export generate an email and attached the generated attachment"""
params = json.loads(data_csv.get("data"))
mails = self.env["mail.mail"].search([])
self.delay_export_obj.export(
params, mail_template=None, is_export_file_attached_to_email=True
)
new_mail = self.env["mail.mail"].search([]) - mails
self.assertEqual(len(new_mail.attachment_ids), 1)
self.assertEqual(new_mail.attachment_ids.name, "res.partner.csv")

def test_cron_delete(self):
"""Check that cron delete attachment after TTL"""
params = json.loads(data_csv.get("data"))
Expand Down
17 changes: 15 additions & 2 deletions export_async_schedule/models/export_async_schedule.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,16 @@ class ExportAsyncSchedule(models.Model):
default=lambda self: self.env.lang,
help="Exports will be translated in this language.",
)

mail_template_id = fields.Many2one(
"mail.template",
string="Email Template",
default=lambda self: self.env.ref(
"base_export_async.delay_export_mail_template"
),
)
is_export_file_attached_to_email = fields.Boolean(
"Export file attached to email", default=False
)
# Scheduling
next_execution = fields.Datetime(default=fields.Datetime.now, required=True)
interval = fields.Integer(default=1, required=True)
Expand Down Expand Up @@ -149,4 +158,8 @@ def action_export(self):
for record in self:
record = record.with_context(lang=record.lang)
params = record._prepare_export_params()
record.env["delay.export"].with_delay().export(params)
record.env["delay.export"].with_delay().export(
params,
mail_template=record.mail_template_id,
is_export_file_attached_to_email=record.is_export_file_attached_to_email,
)
6 changes: 6 additions & 0 deletions export_async_schedule/views/export_async_schedule_views.xml
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,11 @@
name="model_id"
options="{'no_open': True, 'no_create_edit': True}"
/>
<field
name="mail_template_id"
domain="[('model_id','=',model_id)]"
context="{'default_model_id': model_id}"
/>
<field name="model_name" invisible="1" />
<field
name="user_ids"
Expand All @@ -61,6 +66,7 @@
/>
<field name="export_format" />
<field name="import_compat" />
<field name="is_export_file_attached_to_email" />
</group>
<group name="scheduling" string="Scheduling">
<field name="next_execution" />
Expand Down

0 comments on commit 910538b

Please sign in to comment.