refactor: format file with ruff

This commit is contained in:
Tim Grelka
2025-05-14 23:20:49 +02:00
parent 0b81e8a7cd
commit b135c18c47
+31 -33
View File
@@ -8,59 +8,56 @@ from renderer.fifty_ohm_html_renderer import FiftyOhmHtmlRenderer
def parse_katalog(): def parse_katalog():
with open('data/fragenkatalog3b.json') as fragenkatalog_file: with open("data/fragenkatalog3b.json") as fragenkatalog_file:
fragenkatalog = json.load(fragenkatalog_file) fragenkatalog = json.load(fragenkatalog_file)
questions = {} questions = {}
for exampart in fragenkatalog["sections"] : for exampart in fragenkatalog["sections"]:
for chapter in exampart["sections"] : for chapter in exampart["sections"]:
if "questions" in chapter : if "questions" in chapter:
for question in chapter["questions"] : for question in chapter["questions"]:
questions[question["number"]] = question questions[question["number"]] = question
if "sections" in chapter : if "sections" in chapter:
for section in chapter["sections"] : for section in chapter["sections"]:
for question in section["questions"] : for question in section["questions"]:
questions[question["number"]] = question questions[question["number"]] = question
return questions return questions
env = Environment(loader=FileSystemLoader('templates/html'))
question_template = env.get_template('question.html') env = Environment(loader=FileSystemLoader("templates/html"))
question_template = env.get_template("question.html")
questions = parse_katalog() questions = parse_katalog()
def question_stub(input): def question_stub(input):
""" Combines the original question dataset from BNetzA with our internal metadata """ """Combines the original question dataset from BNetzA with our internal metadata"""
with open('data/metadata.json') as metadata_file : with open("data/metadata.json") as metadata_file:
metadata = json.load(metadata_file) metadata = json.load(metadata_file)
number = metadata[f"{input}"]["number"] # Fragennummer z.B. AB123 number = metadata[f"{input}"]["number"] # Fragennummer z.B. AB123
question = questions[number] question = questions[number]
metadata = metadata[f"{input}"] metadata = metadata[f"{input}"]
if "answer_a" in question : if "answer_a" in question:
answers = [ answers = [question["answer_a"], question["answer_b"], question["answer_c"], question["answer_d"]]
question["answer_a"], else:
question["answer_b"],
question["answer_c"],
question["answer_d"]
]
else :
answers = [] answers = []
if metadata["picture_a"] != "" : if metadata["picture_a"] != "":
answer_pictures = [ answer_pictures = [
metadata["picture_a"], metadata["picture_a"],
metadata["picture_b"], metadata["picture_b"],
metadata["picture_c"], metadata["picture_c"],
metadata["picture_d"] metadata["picture_d"],
] ]
else : else:
answer_pictures = [] answer_pictures = []
if "picture_question" in question : if "picture_question" in question:
picture_question = metadata["picture_question"] picture_question = metadata["picture_question"]
else : else:
picture_question = "" picture_question = ""
return question_template.render( return question_template.render(
@@ -69,9 +66,10 @@ def question_stub(input):
layout=metadata["layout"], layout=metadata["layout"],
picture_question=picture_question, picture_question=picture_question,
answers=answers, answers=answers,
answer_pictures=answer_pictures answer_pictures=answer_pictures,
) )
def test_html(capsys): def test_html(capsys):
with capsys.disabled(): with capsys.disabled():
with open("test/acceptanceTest.md") as file: with open("test/acceptanceTest.md") as file:
@@ -79,5 +77,5 @@ def test_html(capsys):
with FiftyOhmHtmlRenderer(question_stub) as renderer: with FiftyOhmHtmlRenderer(question_stub) as renderer:
output = renderer.render(Document(content)) output = renderer.render(Document(content))
with open("test/acceptanceTest.html", "w") as output_file: with open("test/acceptanceTest.html", "w") as output_file:
pretty = BeautifulSoup(output, 'html.parser').prettify() pretty = BeautifulSoup(output, "html.parser").prettify()
output_file.write(pretty) output_file.write(pretty)