60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
#!/usr/bin/env python3.11
|
|
import json
|
|
import csv
|
|
from app.util.estimateGrade import getEstimatedGrade
|
|
from app.util.globals import flex
|
|
|
|
# sort data into usable dictionary
|
|
def getStudents(
|
|
data_dir: str,
|
|
log: bool=False,
|
|
totalBlocks: int=10,
|
|
log_dir: str='./output/raw/students.json'
|
|
) -> list[ dict[ str: any ] ]:
|
|
|
|
students: list[ dict[ str: any ] ] = []
|
|
|
|
with open(data_dir, newline='') as csvfile:
|
|
reader = csv.DictReader(csvfile)
|
|
for row in reader:
|
|
exists = False
|
|
for student in students:
|
|
exists = True if student["Pupil #"] == row["Pupil #"] else False
|
|
if exists: break
|
|
alternate = True if row["Alternate?"] == 'TRUE' else False
|
|
if exists:
|
|
if len(students[student["studentIndex"]]["requests"]) >= totalBlocks and not alternate and row["CrsNo"] not in flex: alternate = True
|
|
students[student["studentIndex"]]["requests"].append({
|
|
"CrsNo": row["CrsNo"],
|
|
"Description": row["Description"],
|
|
"alt": alternate
|
|
})
|
|
if row["CrsNo"] not in flex and not alternate and students[student["studentIndex"]]["expectedClasses"] < 10:
|
|
students[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(students)
|
|
}
|
|
for i in range(1, totalBlocks+1): newStudent["schedule"][f'block{i}'] = []
|
|
students.append(newStudent)
|
|
|
|
# Estimate student grades
|
|
for student in students:
|
|
student["gradelevel"] = getEstimatedGrade(student)
|
|
|
|
if log:
|
|
with open(log_dir, "w") as outfile:
|
|
json.dump(students, outfile, indent=2)
|
|
|
|
return students
|