Floke/Brancheneinstufung2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[32788f42] fix(list-generator): fix syntax error in f-string and fix truncation in pdf_generator.py

Browse Source
This commit is contained in:
Floke
2026-03-18 19:51:55 +00:00
parent fb17445807
commit 0acc2a4c0a
1 changed files with 5 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
list-generator/backend/app/services/pdf_generator.py
View File

@@ -5,28 +5,20 @@ from weasyprint import HTML
import datetime import datetime
def generate_school_pdf(institution: str, date_info: str, list_type: str, students_csv_path: str, families_csv_path: str = None, output_dir: str = "/tmp") -> str: def generate_school_pdf(institution: str, date_info: str, list_type: str, students_csv_path: str, families_csv_path: str = None, output_dir: str = "/tmp") -> str:
# Read CSV data
df = pd.read_csv(students_csv_path, sep=";") df = pd.read_csv(students_csv_path, sep=";")
# Clean column names
df.columns = df.columns.str.strip() df.columns = df.columns.str.strip()
if "Klasse" not in df.columns:
# Group by class df["Klasse"] = "Alle"
grouped = df.groupby("Klasse") grouped = df.groupby("Klasse")
class_data = [] class_data = []
for class_name, group in grouped: for class_name, group in grouped:
students = group.to_dict("records") students = group.to_dict("records")
class_data.append({"name": class_name, "students": students}) class_data.append({"name": class_name, "students": students})
# Prepare summary data
class_counts = [{"name": c, "count": len(g)} for c, g in grouped] class_counts = [{"name": c, "count": len(g)} for c, g in grouped]
total_students = sum(c["count"] for c in class_counts) total_students = sum(c["count"] for c in class_counts)
# Setup Jinja2 environment
template_dir = os.path.join(os.path.dirname(__file__), "..", "templates") template_dir = os.path.join(os.path.dirname(__file__), "..", "templates")
env = Environment(loader=FileSystemLoader(template_dir)) env = Environment(loader=FileSystemLoader(template_dir))
template = env.get_template("school_list.html") template = env.get_template("school_list.html")
# Render HTML
current_time = datetime.datetime.now().strftime("%d.%m.%Y %H:%M Uhr") current_time = datetime.datetime.now().strftime("%d.%m.%Y %H:%M Uhr")
html_out = template.render( html_out = template.render(
institution=institution, institution=institution,
@@ -36,10 +28,9 @@ def generate_school_pdf(institution: str, date_info: str, list_type: str, studen
class_data=class_data, class_data=class_data,
current_time=current_time current_time=current_time
) )
clean_inst = institution.replace(" ", "_").replace("/", "-")
# Generate PDF time_str = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M")
output_filename = f"Listen_{institution.replace( , _)}_{list_type}_{datetime.datetime.now().strftime(%Y-%m-%d_%H-%M)}.pdf" output_filename = f"Listen_{clean_inst}_{list_type}_{time_str}.pdf"
output_path = os.path.join(output_dir, output_filename) output_path = os.path.join(output_dir, output_filename)
HTML(string=html_out).write_pdf(output_path) HTML(string=html_out).write_pdf(output_path)
return output_path return output_path