Document_QnA / src /tools /pretty_print.py
Quent1Fvr's picture
v2.
e2e8616
raw
history blame
1.26 kB
from src.model.paragraph import Paragraph
from src.model.container import Container
#function that pretty prints the paragraphs
def pretty_printer_paragraphs(paragraphs):
for p in paragraphs:
if (p.font_style == "title1"):
print(f"Titre 1 {p.text}")
elif (p.font_style == "title2"):
print(f"---> Titre 2 {p.text}")
elif (p.font_style == "title3"):
print(f"-------> Titre 3 {p.text}")
elif (p.font_style == "title4"):
print(f"-----------> Titre 4 {p.text}")
elif (p.font_style == "content"):
print(f"---------------> {p.text}")
elif (p.font_style == "code"):
print(f"----------code------------> {p.text}")
elif (p.font_style == "table"):
print(f"----------table------------> {p.text}")
def pretty_print_container_structure(container):
if container.title:
print(f"{'-'*container.level} {container.title.text}")
for p in container.paragraphs:
print(f"{'-'*container.level} {p.text}")
for c in container.children:
pretty_print_container_structure(c)
def print_all_block_indexes(container):
for b in container.blocks:
print(f'{b.index} : {b.title if b.title else ""}')