11import io
22import os
33from collections import defaultdict
4+ from functools import partial
45from pathlib import Path
56from 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