65 lines
2.2 KiB
Python
Executable File
65 lines
2.2 KiB
Python
Executable File
#!/usr/bin/env python3.11
|
|
import json
|
|
import csv
|
|
from util.estimateGrade import getEstimatedGrade
|
|
|
|
flex= ("XAT--12A-S", "XAT--12B-S")
|
|
|
|
mockStudents: list[dict] = []
|
|
|
|
# sort real sample data into usable dictionary
|
|
def getSampleStudents(data_dir: str, log: bool = False, totalBlocks: int = 10) -> list[dict]:
|
|
with open(data_dir, newline='') as csvfile:
|
|
reader = csv.DictReader(csvfile)
|
|
for row in reader:
|
|
exists = False
|
|
for student in mockStudents:
|
|
exists = True if student["Pupil #"] == row["Pupil #"] else False
|
|
if exists: break
|
|
alternate = True if row["Alternate?"] == 'TRUE' else False
|
|
if exists:
|
|
if len(mockStudents[student["studentIndex"]]["requests"]) >= totalBlocks and not alternate and row["CrsNo"] not in flex: alternate = True
|
|
mockStudents[student["studentIndex"]]["requests"].append({
|
|
"CrsNo": row["CrsNo"],
|
|
"Description": row["Description"],
|
|
"alt": alternate
|
|
})
|
|
if row["CrsNo"] not in flex and not alternate and mockStudents[student["studentIndex"]]["expectedClasses"] < 10:
|
|
mockStudents[student["studentIndex"]]["expectedClasses"] += 1
|
|
else:
|
|
newStudent = {
|
|
"Pupil #": row["Pupil #"],
|
|
"requests": [{
|
|
"CrsNo": row["CrsNo"],
|
|
"Description": row["Description"],
|
|
"alt": alternate
|
|
}],
|
|
"schedule": {},
|
|
"expectedClasses": 1,
|
|
"classes": 0,
|
|
"remainingAlts": [],
|
|
"studentIndex": len(mockStudents)
|
|
}
|
|
for i in range(1, totalBlocks+1): newStudent["schedule"][f'block{i}'] = []
|
|
mockStudents.append(newStudent)
|
|
|
|
# Estimate student grades
|
|
for student in mockStudents:
|
|
student["gradelevel"] = getEstimatedGrade(student)
|
|
|
|
if log:
|
|
with open("./output/students.json", "w") as outfile:
|
|
json.dump(mockStudents, outfile, indent=2)
|
|
|
|
return mockStudents
|
|
|
|
def main():
|
|
studentRequests: list[dict] = getSampleStudents("../sample_data/course_selection_data.csv")
|
|
|
|
with open("../output/students.json", "w") as outfile:
|
|
json.dump(studentRequests, outfile, indent=2)
|
|
|
|
print("done")
|
|
|
|
if __name__ == '__main__':
|
|
main() |