60 lines
2.5 KiB
Python
60 lines
2.5 KiB
Python
# Copyright 2024 Quartile (https://www.quartile.co)
|
|
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
|
|
|
|
import json
|
|
import zipfile
|
|
from datetime import datetime
|
|
from io import BytesIO
|
|
|
|
from odoo import http, tools
|
|
from odoo.http import content_disposition, request
|
|
from odoo.tools.safe_eval import safe_eval
|
|
|
|
from odoo.addons.web.controllers.report import ReportController
|
|
|
|
|
|
class ExtendedReportController(ReportController):
|
|
@http.route()
|
|
def report_routes(self, reportname, docids=None, converter=None, **data):
|
|
report = request.env["ir.actions.report"]._get_report_from_name(reportname)
|
|
report_name = report.report_file
|
|
doc_ids = []
|
|
if converter == "zip":
|
|
if docids:
|
|
doc_ids = [int(i) for i in docids.split(",")]
|
|
context = dict(request.env.context)
|
|
if data.get("options"):
|
|
data.update(json.loads(data.pop("options")))
|
|
if data.get("context"):
|
|
data["context"] = json.loads(data["context"])
|
|
if data["context"].get("lang"):
|
|
del data["context"]["lang"]
|
|
context.update(data["context"])
|
|
attachments = []
|
|
for doc_id in doc_ids:
|
|
pdf_content, _ = report.with_context(**context)._render_qweb_pdf(
|
|
reportname, res_ids=[doc_id], data=data
|
|
)
|
|
if report.print_report_name:
|
|
obj = request.env[report.model].browse(doc_id)
|
|
report_name = safe_eval(
|
|
report.print_report_name,
|
|
{"object": obj, "time": tools.safe_eval.time},
|
|
)
|
|
report_name = report_name.replace("/", "_")
|
|
pdf_name = f"{report_name}.pdf"
|
|
attachments.append((pdf_name, pdf_content))
|
|
# Generate the ZIP file
|
|
zip_filename = f"{datetime.now().strftime('%Y%m%d_%H%M%S')}.zip"
|
|
bitIO = BytesIO()
|
|
with zipfile.ZipFile(bitIO, "w", zipfile.ZIP_DEFLATED) as zip_file:
|
|
for pdf_name, pdf_content in attachments:
|
|
zip_file.writestr(pdf_name, pdf_content)
|
|
zip_content = bitIO.getvalue()
|
|
headers = [
|
|
("Content-Type", "application/zip"),
|
|
("Content-Disposition", content_disposition(zip_filename)),
|
|
]
|
|
return request.make_response(zip_content, headers=headers)
|
|
return super().report_routes(reportname, docids, converter, **data)
|