103 lines
3.5 KiB
Python
Executable File
103 lines
3.5 KiB
Python
Executable File
#!/usr/bin/env python3.11
|
|
import eel
|
|
import os
|
|
import json
|
|
from app.generator import generateScheduleV3
|
|
from app.util.globals import Error
|
|
from app.util.courses import getCourses, writeCoursesToCSV
|
|
from app.util.students import getStudents, writeStudentsToCSV
|
|
from app.util.convertRawData import putScheduleToWord, putMasterTimetable
|
|
|
|
eel.init('template')
|
|
|
|
@eel.expose
|
|
def start(
|
|
raw_file_data: str,
|
|
min_req: int,
|
|
class_cap: int,
|
|
block_class_limit: int,
|
|
total_blocks: int,
|
|
) -> dict:
|
|
|
|
raw_data_dir = './output/temp/course_selection_data.csv'
|
|
raw_json_dir = './output/raw/json'
|
|
|
|
# Ensure output paths exists
|
|
eel.post_data('Ensuring Directories Exists...')
|
|
|
|
if not os.path.exists('output'): os.makedirs('output')
|
|
if not os.path.exists('output/temp'): os.makedirs('output/temp')
|
|
if not os.path.exists('output/final'): os.makedirs('output/final')
|
|
if not os.path.exists('output/final/student_schedules'): os.makedirs('output/final/student_schedules')
|
|
if not os.path.exists('output/raw'): os.makedirs('output/raw')
|
|
if not os.path.exists('output/raw/json'): os.makedirs('output/raw/json')
|
|
if not os.path.exists('output/raw/csv'): os.makedirs('output/raw/csv')
|
|
|
|
# save raw file data to local file
|
|
eel.post_data('Saving raw data to local file...')
|
|
with open(raw_data_dir, 'w') as raw_file:
|
|
raw_file.write(raw_file_data)
|
|
|
|
# Ensure params are of correct type
|
|
min_req = int(min_req)
|
|
class_cap = int(class_cap)
|
|
block_class_limit = int(block_class_limit)
|
|
total_blocks = int(total_blocks)
|
|
|
|
# call pre-algorithm functions read raw data into a processable format
|
|
eel.post_data('Collecting student information...')
|
|
students = getStudents(
|
|
raw_data_dir,
|
|
log=True,
|
|
totalBlocks=total_blocks,
|
|
log_dir='./output/raw/json/students.json'
|
|
)
|
|
eel.post_data('Collecting course information...')
|
|
courses = getCourses(
|
|
raw_data_dir,
|
|
log=True,
|
|
log_dir=f'{raw_json_dir}/courses.json'
|
|
)
|
|
|
|
# call algorithm to sort data
|
|
eel.post_data('Generating timetables...')
|
|
master_timetable, err = generateScheduleV3(
|
|
students,
|
|
courses,
|
|
minReq=min_req,
|
|
classCap=class_cap,
|
|
blockClassLimit=block_class_limit,
|
|
totalBlocks=total_blocks,
|
|
studentsDir=f'{raw_json_dir}/students.json',
|
|
conflictsDir=f'{raw_json_dir}/conflicts.json',
|
|
coursesDir=f'{raw_json_dir}/courses.json'
|
|
)
|
|
|
|
if err is not None:
|
|
eel.post_data(f'An error has occured while generating the timetable: {err.Title}')
|
|
return err.__dict__
|
|
|
|
# Get updated students
|
|
eel.post_data('Gathering latest data...')
|
|
with open(f'{raw_json_dir}/students.json', 'r') as studentFile: students = json.load(studentFile)
|
|
|
|
# call post-algorithm functions to present sorted data
|
|
eel.post_data('Writing data to .csv files...')
|
|
# get updated raw data
|
|
with open(f'{raw_json_dir}/students.json', 'r') as sFile: students = json.load(sFile)
|
|
with open(f'{raw_json_dir}/courses.json', 'r') as cFile: courses = json.load(cFile)
|
|
writeStudentsToCSV(students, output_dir='./output/raw/csv/students.csv')
|
|
writeCoursesToCSV(courses, output_dir='./output/raw/csv/courses.csv')
|
|
|
|
eel.post_data('Writing timetables to .docx files...')
|
|
for student in students:
|
|
putScheduleToWord(courses, student, './output/final/student_schedules')
|
|
|
|
return err.__dict__ if err is not None else None
|
|
|
|
def main():
|
|
eel.start('index.html', size=(800, 1000))
|
|
|
|
if __name__ == '__main__':
|
|
main()
|