Update pdf_utils.py

pdf_utils.py

@@ -1,32 +1,33 @@
-from reportlab.lib.pagesizes import A4, landscape
-from reportlab.platypus import SimpleDocTemplate, Table, TableStyle, Paragraph, Spacer
+from reportlab.lib.pagesizes import letter, landscape
+from reportlab.pdfgen import canvas
 from reportlab.lib import colors
-from reportlab.lib.styles import getSampleStyleSheet
 
-def export_pdf(df, filename="timetable.pdf"):
-    pdf = SimpleDocTemplate(filename, pagesize=landscape(A4))
-    elements = []
-    styles = getSampleStyleSheet()
-
-    current_class = None
-    for idx, row in df.iterrows():
-        if row['Class/Program'] != current_class:
-            current_class = row['Class/Program']
-            elements.append(Spacer(1,12))
-            elements.append(Paragraph(f"<b>Class/Program: {current_class}</b>", styles['Heading2']))
-            class_df = df[df['Class/Program']==current_class]
-            data = [class_df.columns.tolist()] + class_df.values.tolist()
-            table = Table(data)
-            style = TableStyle([
-                ('BACKGROUND', (0,0), (-1,0), colors.lightblue),
-                ('TEXTCOLOR', (0,0), (-1,0), colors.white),
-                ('ALIGN', (0,0), (-1,-1), 'CENTER'),
-                ('VALIGN', (0,0), (-1,-1), 'MIDDLE'),
-                ('GRID', (0,0), (-1,-1), 1, colors.black),
-                ('FONTSIZE', (0,0), (-1,-1), 9)
-            ])
-            table.setStyle(style)
-            elements.append(table)
-
-    pdf.build(elements)
+def export_timetable_pdf(timetable_dict, filename="timetable.pdf"):
+    c = canvas.Canvas(filename, pagesize=landscape(letter))
+    width, height = landscape(letter)
+    y_start = height - 50
+    for cls, df in timetable_dict.items():
+        c.setFont("Helvetica-Bold", 16)
+        c.drawString(50, y_start, cls)
+        y_start -= 30
+        c.setFont("Helvetica", 12)
+        row_height = 50
+        col_width = (width - 100)/len(df.columns)
+        # Draw header
+        for j, day in enumerate(df.columns):
+            c.setFillColor(colors.lightgrey)
+            c.rect(50 + j*col_width, y_start, col_width, row_height, fill=True)
+            c.setFillColor(colors.black)
+            c.drawString(50 + j*col_width + 5, y_start + row_height/2 - 6, day)
+        # Draw rows
+        for i, slot in enumerate(df.index):
+            for j, day in enumerate(df.columns):
+                x = 50 + j*col_width
+                y = y_start - (i+1)*row_height
+                c.rect(x, y, col_width, row_height)
+                text = df.loc[slot, day] if df.loc[slot, day] is not None else ""
+                c.drawString(x + 5, y + row_height/2 - 6, f"{slot}\n{text}")
+        y_start = y - 70  # space before next class
+        c.showPage()
+    c.save()
     return filename