Skip to content

Commit bb43e92

Browse files
authored
fix(report): use pagina for ENS in footer (#9345)
1 parent 0225627 commit bb43e92

File tree

2 files changed

+23
-6
lines changed

2 files changed

+23
-6
lines changed

api/CHANGELOG.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ All notable changes to the **Prowler API** are documented in this file.
1212
## [1.15.1] (Prowler v5.14.1)
1313

1414
### Fixed
15-
- Fix typo in PDF reporting [(#9322)](https://github.com/prowler-cloud/prowler/pull/9322)
15+
- Fix typo in PDF reporting [(#9345)](https://github.com/prowler-cloud/prowler/pull/9345)
1616
- Fix IaC provider initialization failure when mutelist processor is configured [(#9331)](https://github.com/prowler-cloud/prowler/pull/9331)
1717

1818
---

api/src/backend/tasks/jobs/report.py

Lines changed: 22 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import io
22
import os
33
from collections import defaultdict
4+
from functools import partial
45
from pathlib import Path
56
from shutil import rmtree
67

@@ -772,7 +773,9 @@ def _create_section_score_chart(
772773
return buffer
773774

774775

775-
def _add_pdf_footer(canvas_obj: canvas.Canvas, doc: SimpleDocTemplate) -> None:
776+
def _add_pdf_footer(
777+
canvas_obj: canvas.Canvas, doc: SimpleDocTemplate, compliance_name: str
778+
) -> None:
776779
"""
777780
Add footer with page number and branding to each page of the PDF.
778781
@@ -782,7 +785,9 @@ def _add_pdf_footer(canvas_obj: canvas.Canvas, doc: SimpleDocTemplate) -> None:
782785
"""
783786
canvas_obj.saveState()
784787
width, height = doc.pagesize
785-
page_num_text = f"Page {doc.page}"
788+
page_num_text = (
789+
f"{'Página' if 'ens' in compliance_name.lower() else 'Page'} {doc.page}"
790+
)
786791
canvas_obj.setFont("PlusJakartaSans", 9)
787792
canvas_obj.setFillColorRGB(0.4, 0.4, 0.4)
788793
canvas_obj.drawString(30, 20, page_num_text)
@@ -1595,7 +1600,11 @@ def get_weight_for_requirement(requirement_dict):
15951600
elements.append(PageBreak())
15961601

15971602
# Build the PDF
1598-
doc.build(elements, onFirstPage=_add_pdf_footer, onLaterPages=_add_pdf_footer)
1603+
doc.build(
1604+
elements,
1605+
onFirstPage=partial(_add_pdf_footer, compliance_name=compliance_name),
1606+
onLaterPages=partial(_add_pdf_footer, compliance_name=compliance_name),
1607+
)
15991608
except Exception as e:
16001609
tb_lineno = e.__traceback__.tb_lineno if e.__traceback__ else "unknown"
16011610
logger.info(f"Error building the document, line {tb_lineno} -- {e}")
@@ -2818,7 +2827,11 @@ def generate_ens_report(
28182827

28192828
# Build the PDF
28202829
logger.info("Building PDF...")
2821-
doc.build(elements, onFirstPage=_add_pdf_footer, onLaterPages=_add_pdf_footer)
2830+
doc.build(
2831+
elements,
2832+
onFirstPage=partial(_add_pdf_footer, compliance_name=compliance_name),
2833+
onLaterPages=partial(_add_pdf_footer, compliance_name=compliance_name),
2834+
)
28222835
except Exception as e:
28232836
tb_lineno = e.__traceback__.tb_lineno if e.__traceback__ else "unknown"
28242837
logger.error(f"Error building ENS report, line {tb_lineno} -- {e}")
@@ -3365,7 +3378,11 @@ def generate_nis2_report(
33653378

33663379
# Build the PDF
33673380
logger.info("Building NIS2 PDF...")
3368-
doc.build(elements, onFirstPage=_add_pdf_footer, onLaterPages=_add_pdf_footer)
3381+
doc.build(
3382+
elements,
3383+
onFirstPage=partial(_add_pdf_footer, compliance_name=compliance_name),
3384+
onLaterPages=partial(_add_pdf_footer, compliance_name=compliance_name),
3385+
)
33693386
logger.info(f"NIS2 report successfully generated at {output_path}")
33703387

33713388
except Exception as e:

0 commit comments

Comments
 (0)