diff --git a/report_async/models/ir_report.py b/report_async/models/ir_report.py index e70988bdd8..ad8df751f8 100644 --- a/report_async/models/ir_report.py +++ b/report_async/models/ir_report.py @@ -3,26 +3,24 @@ from odoo import api, models - # Define all supported report_type -REPORT_TYPES = ['qweb-pdf', 'qweb-text', - 'qweb-xml', 'csv', - 'excel', 'xlsx'] +REPORT_TYPES = ["qweb-pdf", "qweb-text", "qweb-xml", "csv", "excel", "xlsx"] class Report(models.Model): - _inherit = 'ir.actions.report' + _inherit = "ir.actions.report" @api.noguess def report_action(self, docids, data=None, config=True): - res = super(Report, self).report_action(docids, data=data, - config=config) - if res['context'].get('async_process', False): - rpt_async_id = res['context']['active_id'] - report_async = self.env['report.async'].browse(rpt_async_id) - if res['report_type'] in REPORT_TYPES: - report_async.with_delay().run_report( - res['context'].get('active_ids', []), data, - self.id, self._uid) + res = super(Report, self).report_action(docids, data=data, config=config) + if res["context"].get("async_process", False): + rpt_async_id = res["context"]["active_id"] + report_async = self.env["report.async"].browse(rpt_async_id) + if res["report_type"] in REPORT_TYPES: + report_async.with_delay( + eta=res["context"].get("eta", False) + ).run_report( + res["context"].get("active_ids", []), data, self.id, self._uid + ) return {} return res diff --git a/report_async/models/report_async.py b/report_async/models/report_async.py index 376acc71c2..b5925abc53 100644 --- a/report_async/models/report_async.py +++ b/report_async/models/report_async.py @@ -2,6 +2,7 @@ # License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html) import base64 +from datetime import datetime, timedelta from odoo import api, fields, models, _ from odoo.tools.safe_eval import safe_eval from odoo.exceptions import UserError @@ -70,6 +71,8 @@ class ReportAsync(models.Model): help="List all files created by this report background process", ) + schedule_time = fields.Char(string='Schedule time') + @api.multi def _compute_job(self): for rec in self: @@ -110,6 +113,8 @@ def run_async(self): result = action.read()[0] ctx = safe_eval(result.get('context', {})) ctx.update({'async_process': True}) + if self.schedule_time: + ctx.update({'eta': self._get_next_schedule_time()}) result['context'] = ctx return result @@ -160,3 +165,11 @@ def _send_email(self, attachment): template.send_mail(attachment.id, notif_layout='mail.mail_notification_light', force_send=False) + + def _get_next_schedule_time(self): + target_time = datetime.strptime(self.schedule_time, "%H:%M").time() + now = fields.Datetime.now() + target_datetime = datetime.combine(now.date(), target_time) + if now.time() > target_time: + target_datetime += timedelta(days=1) + return target_datetime diff --git a/report_async/views/report_async.xml b/report_async/views/report_async.xml index f6d5f58f14..cbf6deaff0 100644 --- a/report_async/views/report_async.xml +++ b/report_async/views/report_async.xml @@ -73,6 +73,8 @@ +