#!/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') -> 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)