#!/usr/bin/python3
from prettytable import PrettyTable
from typing import Tuple
import json
import sys

# Import required utilities
from util.mockStudents import generateMockStudents, getSampleStudents
from util.generateCourses import getSampleCourses
from util.courses import mockCourses

# Import Algorithms
from scheduleGenerator.generator_v1 import generateScheduleV1
from scheduleGenerator.generator_v2 import generateScheduleV2
from scheduleGenerator.generator_v3 import generateScheduleV3

def errorOutput(students) -> Tuple[PrettyTable, dict, dict]:
  # Error Table calulation / output  
  f = open('./output/conflicts.json')
  conflicts = json.load(f)
  f.close()

  critical, acceptable = 0, 0
  for student in conflicts["All"]:
    studentHasCritical, studentHasAcceptable = False, False
    read = False
    for conflict in conflicts["All"][student]:
      if read: break
      if conflict["Type"] == "Critical":
        critical += 1
        studentHasCritical = True
        if studentHasAcceptable: read = True
      elif conflict["Type"] == "Acceptable":
        acceptable += 1
        studentHasAcceptable = True
        if studentHasCritical: read = True
    continue

  t = PrettyTable(['Type', 'Error %', 'Success %', 'Error Ratio'])
  
  errorsC = round(critical / len(students) * 100, 2)
  successC = round(100 - errorsC, 2)
  errorsA = round(acceptable / len(students) * 100, 2)
  successA = round(100 - errorsA, 2)
  
  t.add_row(['Critical', f"{errorsC} %", f"{successC} %", f"{critical}/{len(students)}"])
  t.add_row(['Acceptable', f"{errorsA} %", f"{successA} %", f"{acceptable}/{len(students)}"])
  
  return t, conflicts["Critical"], conflicts["Acceptable"]

if __name__ == '__main__':
  
  if len(sys.argv) == 1:
    print("Missing argument")
    exit()

  if sys.argv[1].upper() == 'V1':
    print("Processing...")

    mockStudents = generateMockStudents(400)
    timetable = {}
    timetable["Version"] = 1
    timetable["timetable"] = generateScheduleV1(mockStudents, mockCourses)
  
  elif sys.argv[1].upper() == 'V2':
    print("Processing...")
  
    mockStudents = generateMockStudents(400)
    timetable = {}
    timetable["Version"] = 2
    timetable["timetable"] = generateScheduleV2(mockStudents, mockCourses)
  

  elif sys.argv[1].upper() == 'V3':
  
    print("Processing...\n")
  
    sampleStudents = getSampleStudents("./sample_data/course_selection_data.csv", True)
    samplemockCourses = getSampleCourses("./sample_data/course_selection_data.csv", True)
    timetable = {}
    timetable["Version"] = 3
    timetable["timetable"] = generateScheduleV3(sampleStudents, samplemockCourses, 40, "./output/students.json", "./output/conflicts.json")

    errors, _, _ = errorOutput(sampleStudents)
    print(errors)

  elif sys.argv[1].upper() == "ERRORS":
    f = open('./output/students.json')
    studentData = json.load(f)
    f.close()
    errors, critical, acceptable = errorOutput(studentData)
    print()
    print(errors)

    print(f"\n{len(critical)} critical errors")
    c1, c2, co = 0, 0, 0
    for error in critical:
      if error["Code"] == "C-MC": c1 +=1
      elif error["Code"] == "C-CR": c2 += 1
      else: co += 1

    print(f"x{c1} C-MC Errors:   Critical - Missing too many Classes")
    print(f"x{c2} C-CR Errors:   Critical - Couldn't Resolve")
    print(f"x{co} Other/Undefined Critical Errors")

    print(f"\n{len(acceptable)} acceptable errors")
    a1, ao = 0, 0
    for error in acceptable:
      if error["Code"] == "A-MC": a1 +=1
      else: ao += 1

    print(f"x{a1} A-MC Errors: Acceptable - Missing 1-2 Classes")
    print(f"x{ao} Other/Undefined Acceptable Errors")

    exit()

  else:
    print("Invalid argument")
    exit()

  with open("./output/timetable.json", "w") as outfile:
    json.dump(timetable, outfile, indent=2)

  print("\nDone")