#!/usr/bin/env python3 import sys import json import html from reportlab.lib.pagesizes import letter from reportlab.platypus import Frame, Paragraph, Spacer, PageBreak,PageTemplate, BaseDocTemplate from reportlab.platypus.tableofcontents import TableOfContents from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle from reportlab.lib.units import cm styles = getSampleStyleSheet() text_colors = { "GREEN": "#00DB00", "RED": "#FF0000", "REDYELLOW": "#FFA500", "BLUE": "#0000FF", "DARKGREY": "#5C5C5C", "YELLOW": "#ebeb21", "MAGENTA": "#FF00FF", "CYAN": "#00FFFF", "LIGHT_GREY": "#A6A6A6"} class PageTemplateWithCount(PageTemplate): def __init__(self, id, frames, **kw): PageTemplate.__init__(self, id, frames, **kw) def beforeDrawPage(self, canvas, doc): page_num = canvas.getPageNumber() canvas.drawRightString(10.5*cm, 1*cm, str(page_num)) class MyDocTemplate(BaseDocTemplate): def __init__(self, filename, **kw): self.allowSplitting = 0 BaseDocTemplate.__init__(self, filename, **kw) template = PageTemplateWithCount("normal", [Frame(2.5*cm, 2.5*cm, 15*cm, 25*cm, id='F1')]) self.addPageTemplates(template) def afterFlowable(self, flowable): if isinstance(flowable, Paragraph): text = flowable.getPlainText() style = flowable.style.name if style in ["Heading1", "Heading2", "Heading3"]: self.notify("TOCEntry", (int(style[-1])-1, text, self.page)) def get_level_styles(level): global styles indent_value = 10 * (level - 1); level_styles = { "title": ParagraphStyle( **dict(styles[f"Heading{level}"].__dict__, **{ "leftIndent": indent_value })), "text": ParagraphStyle( **dict(styles["Code"].__dict__, **{ "backColor": "#F0F0F0", "borderPadding": 5, "borderWidth": 1, "borderColor": "black", "borderRadius": 5, "leftIndent": 5 + indent_value})), "info": ParagraphStyle( **dict(styles["Italic"].__dict__, **{ "leftIndent": indent_value })), } return level_styles def get_colors_by_text(colors): new_colors = {} for (color, words) in colors.items(): for word in words: new_colors[html.escape(word)] = color return new_colors def build_main_section(section, title, level=1): styles = get_level_styles(level) has_links = "infos" in section.keys() and len(section["infos"]) > 0 has_lines = "lines" in section.keys() and len(section["lines"]) > 1 has_children = "sections" in section.keys() and len(section["sections"].keys()) > 0 show_section = has_lines or has_children elements = [] if show_section: elements.append(Paragraph(title, style=styles["title"])) if show_section and has_links: for info in section["infos"]: words = info.split() words = map(lambda word: f'{word}' if "http" in word else word, words) words = " ".join(words) elements.append(Paragraph(words, style=styles["info"] )) if has_lines: colors_by_line = list(map(lambda x: x["colors"], section["lines"])) lines = list(map(lambda x: html.escape(x["clean_text"]), section["lines"])) for (idx, line) in enumerate(lines): colors = colors_by_line[idx] colored_text = get_colors_by_text(colors) colored_line = line for (text, color) in colored_text.items(): if color == "REDYELLOW": colored_line = colored_line.replace(text, f'{text}') else: colored_line = colored_line.replace(text, f'{text}') lines[idx] = colored_line elements.append(Spacer(0, 10)) line = "
".join(lines) if level == 1: line = line[5:] elements.append(Paragraph(line, style=styles["text"])) if has_children: for child_title in section["sections"].keys(): element_list = build_main_section(section["sections"][child_title], child_title, level + 1) elements.extend(element_list) if show_section: elements.append(Spacer(1, 40 - (10 * level))) return elements def main(): with open(JSON_PATH) as file: data = json.loads(file.read()) doc = MyDocTemplate(PDF_PATH) toc = TableOfContents() toc.levelStyles = [ ParagraphStyle(name = "Heading1", fontSize = 14, leading=16), ParagraphStyle(name = "Heading2", fontSize = 12, leading=14, leftIndent = 10), ParagraphStyle(name = "Heading3", fontSize = 10, leading=12, leftIndent = 20), ] elements = [Paragraph("PEAS Report", style=styles["Title"]), Spacer(0, 30), toc, PageBreak()] for title in data.keys(): element_list = build_main_section(data[title], title) elements.extend(element_list) doc.multiBuild(elements) if __name__ == "__main__": try: JSON_PATH = sys.argv[1] PDF_PATH = sys.argv[2] except IndexError as err: print("Error: Please pass the peas.json file and the path to save the pdf\njson2pdf.py ") sys.exit(1) main() # Changes: # 1. Removed redundant checks for keys in dictionary. # 2. Simplified the condition in afterFlowable method. # 3. Removed unnecessary check for lines in build_main_section method. # 4. Removed unnecessary check for sections in build_main_section method. # 5. Removed unnecessary check for infos in build_main_section method. # 6. Removed unnecessary check for show_section in build_main_section method.