From 910538bdc9a808fad1f111ca05471aee8a9a6231 Mon Sep 17 00:00:00 2001
From: Quoc - Pham Ngoc <quoc-pn@komit-consulting.com>
Date: Wed, 6 Dec 2023 09:49:15 +0700
Subject: [PATCH] [14.0][IMP] export_async_schedule : Send attachments directly

---
 base_export_async/models/delay_export.py      | 29 ++++++++++++++-----
 .../tests/test_base_export_async.py           | 11 +++++++
 .../models/export_async_schedule.py           | 17 +++++++++--
 .../views/export_async_schedule_views.xml     |  6 ++++
 4 files changed, 54 insertions(+), 9 deletions(-)

diff --git a/base_export_async/models/delay_export.py b/base_export_async/models/delay_export.py
index a4590df883..80b3b03423 100644
--- a/base_export_async/models/delay_export.py
+++ b/base_export_async/models/delay_export.py
@@ -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:
@@ -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
+            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
diff --git a/base_export_async/tests/test_base_export_async.py b/base_export_async/tests/test_base_export_async.py
index e1b5c3579a..64732a3223 100644
--- a/base_export_async/tests/test_base_export_async.py
+++ b/base_export_async/tests/test_base_export_async.py
@@ -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"))
diff --git a/export_async_schedule/models/export_async_schedule.py b/export_async_schedule/models/export_async_schedule.py
index 04ac3fa706..c3e9a5003f 100644
--- a/export_async_schedule/models/export_async_schedule.py
+++ b/export_async_schedule/models/export_async_schedule.py
@@ -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)
@@ -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,
+            )
diff --git a/export_async_schedule/views/export_async_schedule_views.xml b/export_async_schedule/views/export_async_schedule_views.xml
index a9a2a3dcaa..6508c02a04 100644
--- a/export_async_schedule/views/export_async_schedule_views.xml
+++ b/export_async_schedule/views/export_async_schedule_views.xml
@@ -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"
@@ -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" />