80 lines
2.6 KiB
Python
80 lines
2.6 KiB
Python
#!/usr/bin/env python 3.11
|
|
from docx.shared import Inches
|
|
from app.util.globals import flex
|
|
import xlsxwriter
|
|
import docx
|
|
|
|
def putMasterTimetable(
|
|
table: dict,
|
|
output_dir: str = './output/final'
|
|
) -> None:
|
|
|
|
workbook = xlsxwriter.Workbook(f'{output_dir}/master_timetable.xlsx')
|
|
worksheet = workbook.add_worksheet()
|
|
|
|
sem1Header = workbook.add_format({'bold': True, 'align': 'center', 'bg_color': '#91c5fa', 'border': 1})
|
|
sem2Header = workbook.add_format({'bold': True, 'align': 'center', 'bg_color': '#a2fa91', 'border': 1})
|
|
sem1Data = workbook.add_format({'bg_color': '#cee1f5', 'border': 1})
|
|
sem2Data = workbook.add_format({'bg_color': '#d8f7d2', 'border': 1})
|
|
worksheet.merge_range(0, 0, 0, len(table) * 2 - 1, 'Semester 1', sem1Header)
|
|
worksheet.merge_range(0, len(table) * 2, 0, (len(table) * 4) - 1, 'Semester 2', sem2Header)
|
|
|
|
tracker = 0
|
|
for i in range(0, len(table) * 4, 4):
|
|
tracker += 1
|
|
block, index = tracker, f'block{tracker}'
|
|
header = sem1Header if tracker <= len(table) / 2 else sem2Header
|
|
block = block if block <= len(table) / 2 else block - (len(table) // 2)
|
|
worksheet.merge_range(1, i, 1, i + 3, f'Block {block}', header)
|
|
|
|
row = 2
|
|
for course in table[index]:
|
|
style = sem1Data if tracker <= len(table) / 2 else sem2Data
|
|
worksheet.merge_range(row, i, row + 1, i + 3, f'{table[index][course]["Description"]}\n{course[:-2]}', style)
|
|
row += 2
|
|
|
|
workbook.close()
|
|
|
|
def putScheduleToWord(
|
|
courses: dict,
|
|
student: dict,
|
|
output_dir: str = './output/final/student_schedules'
|
|
) -> None:
|
|
|
|
# create an instance of a word doc.
|
|
doc = docx.Document()
|
|
|
|
doc.add_heading(f'Schedule for Student {student["Pupil #"]}')
|
|
|
|
# Create a table object
|
|
table = doc.add_table(rows=1, cols=3)
|
|
table.autofit = True
|
|
table.style = 'Colorful List'
|
|
|
|
# course name + (code) | block | semester
|
|
row = table.rows[0].cells
|
|
row[0].text = 'Course'
|
|
row[1].text = 'Block'
|
|
row[2].text = 'Sem.'
|
|
|
|
table.columns[0].width = Inches(3.5)
|
|
table.columns[1].width = Inches(0.75)
|
|
table.columns[2].width = Inches(0.75)
|
|
|
|
for block in student['schedule']:
|
|
courseCode = student["schedule"][block][0]
|
|
if courseCode in flex: courseName = 'Study'
|
|
else: courseName = courses[courseCode[:-2]]["Description"]
|
|
blockNum = int(block.split('block')[1])
|
|
semester = 2 if blockNum > 5 else 1
|
|
if blockNum > (len(student["schedule"])/2): blockNum -= (len(student["schedule"])/2)
|
|
row = table.add_row().cells
|
|
row[0].text = f'{courseName}\n({courseCode})'
|
|
row[1].text = str(blockNum)
|
|
row[2].text = str(semester)
|
|
|
|
for row in table.rows:
|
|
row.height = Inches(0.75)
|
|
|
|
doc.save(f'{output_dir}/{student["Pupil #"]}_schedule.docx')
|