73 lines
2.1 KiB
Python
73 lines
2.1 KiB
Python
#!/usr/bin/env python3.11
|
|
import json
|
|
import csv
|
|
from app.util.estimateGrade import getGrade
|
|
|
|
# Get all requested courses from data
|
|
def getCourses(
|
|
data_dir: str,
|
|
log: int = False,
|
|
log_dir: str = './output/raw/courses.json',
|
|
) -> dict[dict[str: any]]:
|
|
|
|
courses = {}
|
|
|
|
with open(data_dir, newline='') as csvfile:
|
|
reader = csv.DictReader(csvfile)
|
|
for row in reader:
|
|
exists = False
|
|
for course in courses:
|
|
exists = True if courses[course]["CrsNo"] == row["CrsNo"] else False
|
|
if exists: break
|
|
if not exists:
|
|
grade = getGrade(row["CrsNo"], row["Description"])
|
|
courses[row["CrsNo"]] = {
|
|
"CrsNo": row["CrsNo"],
|
|
"Description": row["Description"],
|
|
"Grade": grade,
|
|
"Requests": 0,
|
|
"Sem1": 0, # number of classes running in sem1
|
|
"Sem2": 0, # number of classes running in sem2
|
|
"Total": 0, # total number of classes running
|
|
"Seats": 0, # total number of seats
|
|
"Occupied": 0, # number of occupied seats
|
|
}
|
|
|
|
if log:
|
|
with open(log_dir, "w") as outfile:
|
|
json.dump(courses, outfile, indent=2)
|
|
|
|
return courses
|
|
|
|
# Writes all course data to csv file
|
|
def writeCoursesToCSV(courses: dict, output_dir: str='./output/raw/csv/courses.csv', classCap: int=30) -> None:
|
|
with open(output_dir, 'w') as file:
|
|
writer = csv.writer(file)
|
|
# Write header
|
|
data = (
|
|
"CrsNo",
|
|
"Description",
|
|
"Grade",
|
|
"Requests",
|
|
"First sem. # of Classes",
|
|
"Second sem. # of Classes",
|
|
"Total # of Classes",
|
|
"Total # of Seats",
|
|
"# of Seats Occupied"
|
|
)
|
|
writer.writerow(data)
|
|
|
|
for course in courses:
|
|
courseData = (
|
|
courses[course]["CrsNo"],
|
|
courses[course]["Description"],
|
|
courses[course]["Grade"],
|
|
courses[course]["Requests"],
|
|
courses[course]["Sem1"],
|
|
courses[course]["Sem2"],
|
|
courses[course]["Total"],
|
|
courses[course]["Seats"],
|
|
courses[course]["Occupied"]
|
|
)
|
|
writer.writerow(courseData)
|