Public release from ruodoo-project: 19.0 - 2026-05-31 21:19:12 UTC
This commit is contained in:
CI Publish Bot
219
docx_report/controllers/main.py
Normal file
219
docx_report/controllers/main.py
Normal file
@ -0,0 +1,219 @@
|
||||
from json import dumps as json_dumps, loads as json_loads
|
||||
from werkzeug.urls import url_decode
|
||||
import re
|
||||
|
||||
from odoo.http import (
|
||||
content_disposition,
|
||||
request,
|
||||
route,
|
||||
serialize_exception as _serialize_exception,
|
||||
)
|
||||
from odoo.tools import html_escape
|
||||
from odoo.tools.safe_eval import safe_eval, time
|
||||
|
||||
from odoo.addons.web.controllers.report import ReportController
|
||||
|
||||
import logging
|
||||
_logger = logging.getLogger(__name__)
|
||||
|
||||
class DocxReportController(ReportController):
|
||||
|
||||
def _prepare_docx_report_context(self, docids, data):
|
||||
"""
|
||||
По сути дублирует сбор контекста и docids в части базового report_routes. Новая логика непосредственно в переопределенном report_routes.
|
||||
Возможно есть и более изящный способ.
|
||||
"""
|
||||
context = dict(request.env.context)
|
||||
payload = {}
|
||||
docids_list = [int(i) for i in docids.split(",")] if docids else []
|
||||
if data.get("options"):
|
||||
payload.update(json_loads(data.pop("options")))
|
||||
if data.get("context"):
|
||||
payload["context"] = json_loads(data["context"])
|
||||
if payload["context"].get("lang") and not payload.get("force_context_lang"):
|
||||
del payload["context"]["lang"]
|
||||
context.update(payload["context"])
|
||||
return context, payload, docids_list
|
||||
|
||||
def _get_docx_report_from_name(self, reportname):
|
||||
template = request.env["docx.template"].search([("report_name", "=", reportname)], limit=1)
|
||||
if template:
|
||||
return template.report_id
|
||||
return request.env["ir.actions.report"]._get_report_from_name(reportname)
|
||||
|
||||
@route()
|
||||
def report_routes(self, reportname, docids=None, converter=None, **data):
|
||||
"""
|
||||
Расширенный маршрут `/report/<converter>/<reportname>/<docids>`
|
||||
Добавлена поддержка:
|
||||
- converter=docx → генерация DOCX
|
||||
- converter=pdf → конвертация DOCX → PDF (Gotenberg)
|
||||
"""
|
||||
_logger.info("[ROUTE] report_routes() called: reportname=%s converter=%s docids=%s",
|
||||
reportname, converter, docids)
|
||||
|
||||
context, payload, docids_list = self._prepare_docx_report_context(docids, data)
|
||||
|
||||
report = self._get_docx_report_from_name(reportname)
|
||||
|
||||
# -----------------------
|
||||
# 1) DOCX генерация
|
||||
# -----------------------
|
||||
if converter == "docx":
|
||||
_logger.info("[ROUTE] DOCX branch for report=%s", reportname)
|
||||
|
||||
docx_content = report.with_context(context)._render_docx_docx(
|
||||
docids_list, data=payload
|
||||
)
|
||||
headers = [
|
||||
(
|
||||
"Content-Type",
|
||||
"application/vnd.openxmlformats-officedocument.wordprocessingml.document",
|
||||
),
|
||||
]
|
||||
return request.make_response(docx_content, headers=headers)
|
||||
|
||||
# -----------------------
|
||||
# 2) PDF генерация из DOCX
|
||||
# -----------------------
|
||||
if converter == "pdf" and "docx" in (report.report_type or ""):
|
||||
_logger.info("[ROUTE] PDF branch for report=%s docids=%s", reportname, docids_list)
|
||||
|
||||
pdf_result = report.with_context(context)._render_docx_pdf(
|
||||
docids_list, data=payload
|
||||
)
|
||||
|
||||
# Может быть либо bytes, либо (bytes, ext)
|
||||
if isinstance(pdf_result, tuple):
|
||||
pdf_content = pdf_result[0]
|
||||
else:
|
||||
pdf_content = pdf_result
|
||||
|
||||
headers = [
|
||||
("Content-Type", "application/pdf"),
|
||||
("Content-Length", len(pdf_content)),
|
||||
]
|
||||
return request.make_response(pdf_content, headers=headers)
|
||||
|
||||
return super().report_routes(
|
||||
reportname, docids, converter, **data
|
||||
)
|
||||
|
||||
def _get_docx_output_filename(self, report, docids, extension):
|
||||
"""
|
||||
Возвращает финальное имя файла для скачивания DOCX.
|
||||
|
||||
Приоритет:
|
||||
1) docx.template.filename_pattern (если существует docx.template)
|
||||
2) report.print_report_name (как fallback, если оно всё же задано)
|
||||
3) report.name
|
||||
|
||||
Ожидается, что выражение возвращает строку БЕЗ расширения (.docx).
|
||||
"""
|
||||
filename = "%s.%s" % (report.name, extension)
|
||||
|
||||
template = request.env["docx.template"].sudo().search(
|
||||
[("report_id", "=", report.id)], limit=1
|
||||
)
|
||||
|
||||
pattern = None
|
||||
if template and template.filename_pattern:
|
||||
pattern = template.filename_pattern
|
||||
elif report.print_report_name:
|
||||
pattern = report.print_report_name
|
||||
|
||||
if not pattern or not docids:
|
||||
return filename
|
||||
|
||||
ids = [int(x) for x in docids.split(",") if x.isdigit()]
|
||||
if len(ids) != 1:
|
||||
return filename
|
||||
|
||||
obj = request.env[report.model].browse(ids)
|
||||
|
||||
try:
|
||||
report_name = safe_eval(
|
||||
pattern,
|
||||
{"object": obj, "time": time},
|
||||
)
|
||||
except Exception as e:
|
||||
return filename
|
||||
|
||||
if not isinstance(report_name, str):
|
||||
return filename
|
||||
|
||||
clean = report_name.strip()
|
||||
if not clean:
|
||||
return filename
|
||||
|
||||
clean = re.sub(r"\.(docx?|pdf)$", "", clean, flags=re.IGNORECASE)
|
||||
|
||||
return "%s.%s" % (clean, extension)
|
||||
|
||||
@route()
|
||||
def report_download(self, data, token, context=None):
|
||||
"""
|
||||
Обрабатывает запрос на скачивание файла отчета.
|
||||
Расширен для:
|
||||
- docx-docx (DOCX)
|
||||
- docx-pdf (PDF из DOCX)
|
||||
"""
|
||||
_logger.info("[REPORT DOWNLOAD] data=%s, token=%s", data, token)
|
||||
|
||||
requestcontent = json_loads(data)
|
||||
url, type_ = requestcontent[0], requestcontent[1]
|
||||
|
||||
try:
|
||||
if type_ in ["docx-docx", "docx-pdf"]:
|
||||
converter = "docx" if type_ == "docx-docx" else "pdf"
|
||||
extension = "docx" if type_ == "docx-docx" else "pdf"
|
||||
|
||||
pattern = "/report/%s/" % ("docx" if type_ == "docx-docx" else "pdf")
|
||||
reportname = url.split(pattern)[1].split("?")[0]
|
||||
|
||||
docids = None
|
||||
if "/" in reportname:
|
||||
reportname, docids = reportname.split("/")
|
||||
|
||||
_logger.info(
|
||||
"[REPORT DOWNLOAD] type=%s converter=%s extension=%s reportname=%s docids=%s",
|
||||
type_, converter, extension, reportname, docids,
|
||||
)
|
||||
|
||||
if docids:
|
||||
response = self.report_routes(
|
||||
reportname, docids=docids, converter=converter, context=context
|
||||
)
|
||||
|
||||
else:
|
||||
query_data = {}
|
||||
if "?" in url:
|
||||
query_data = dict(url_decode(url.split("?")[1]).items())
|
||||
|
||||
if "context" in query_data:
|
||||
base_ctx = json_loads(context or "{}") if context else {}
|
||||
data_context = json_loads(query_data.pop("context"))
|
||||
context = json_dumps({**base_ctx, **data_context})
|
||||
|
||||
response = self.report_routes(
|
||||
reportname, converter=converter, context=context, **query_data
|
||||
)
|
||||
|
||||
report = self._get_docx_report_from_name(reportname)
|
||||
filename = self._get_docx_output_filename(report, docids, extension)
|
||||
|
||||
response.headers.add(
|
||||
"Content-Disposition", content_disposition(filename)
|
||||
)
|
||||
return response
|
||||
|
||||
else:
|
||||
return super().report_download(
|
||||
data, context=context
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
_logger.exception("[REPORT DOWNLOAD] ERROR: %s", e)
|
||||
se = _serialize_exception(e)
|
||||
error = {"code": 200, "message": "Odoo Server Error", "data": se}
|
||||
return request.make_response(html_escape(json_dumps(error)))
|
||||
Reference in New Issue
Block a user