From 9ca0eb3e1462292d06474ed2660b353661553bb5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rapha=C3=ABl=20Valyi?= Date: Wed, 31 Jul 2024 13:34:18 +0000 Subject: [PATCH] [MIG] l10n_br_nfe: _render_qweb_* migration --- l10n_br_nfe/models/document.py | 4 +++- l10n_br_nfe/report/ir_actions_report.py | 13 ++++++------- l10n_br_nfe/tests/test_nfe_danfe.py | 2 +- 3 files changed, 10 insertions(+), 9 deletions(-) diff --git a/l10n_br_nfe/models/document.py b/l10n_br_nfe/models/document.py index 0fa79b76734e..1e7d59630ebb 100644 --- a/l10n_br_nfe/models/document.py +++ b/l10n_br_nfe/models/document.py @@ -1305,7 +1305,9 @@ def make_pdf(self): "type": "binary", } report = self.env.ref("l10n_br_nfe.report_danfe") - pdf_data = report._render_qweb_pdf(self.fiscal_line_ids.document_id.ids) + pdf_data = report._render_qweb_pdf( + "l10n_br_nfe.main_template_danfe", self.fiscal_line_ids.document_id.ids + ) attachment_data["datas"] = base64.b64encode(pdf_data[0]) file_pdf = self.file_report_id self.file_report_id = False diff --git a/l10n_br_nfe/report/ir_actions_report.py b/l10n_br_nfe/report/ir_actions_report.py index 17b37a4b76d1..b6e22f015ffa 100644 --- a/l10n_br_nfe/report/ir_actions_report.py +++ b/l10n_br_nfe/report/ir_actions_report.py @@ -37,18 +37,17 @@ def temp_xml_autorizacao(self, xml_string): new_root.append(protNFe_node) return etree.tostring(new_root) - def _render_qweb_html(self, res_ids, data=None): - if self.report_name == "main_template_danfe": + def _render_qweb_html(self, report_ref, res_ids, data=None): + if report_ref == "l10n_br_nfe.main_template_danfe": return - return super()._render_qweb_html(res_ids, data=data) + return super()._render_qweb_html(report_ref, res_ids, data=data) - def _render_qweb_pdf(self, res_ids, data=None): - if self.report_name not in ["main_template_danfe"]: - return super()._render_qweb_pdf(res_ids, data=data) + def _render_qweb_pdf(self, report_ref, res_ids, data=None): + if report_ref not in ["l10n_br_nfe.main_template_danfe"]: + return super()._render_qweb_pdf(report_ref, res_ids, data=data) nfe = self.env["l10n_br_fiscal.document"].search([("id", "in", res_ids)]) - return self._render_danfe(nfe) def _render_danfe(self, nfe): diff --git a/l10n_br_nfe/tests/test_nfe_danfe.py b/l10n_br_nfe/tests/test_nfe_danfe.py index 2a1d7f61ca37..85a1fee9a438 100644 --- a/l10n_br_nfe/tests/test_nfe_danfe.py +++ b/l10n_br_nfe/tests/test_nfe_danfe.py @@ -37,7 +37,7 @@ def test_generate_danfe_document_type_error(self): nfe.document_type_id = self.env.ref("l10n_br_fiscal.document_01") nfe.action_document_confirm() with self.assertRaises(UserError) as captured_exception: - danfe_report._render_qweb_pdf([nfe.id]) + danfe_report._render_qweb_pdf("l10n_br_nfe.main_template_danfe", [nfe.id]) self.assertEqual( captured_exception.exception.args[0], "You can only print a DANFE of a NFe(55).",