import subject
import user_IO
import serializer
import encryption
import exceptions
import os

subjects = []
try:
    while True:
        try:
            user_input = user_IO.user_input()
            input_or_output = user_input[0]
            if input_or_output == "input":
                _edit = user_input[1]
                _subject = user_input[2]
                if _edit == "grade":
                    _category = user_input[3]
                    _grade = user_input[4]
                    _grade_name = user_input[5]
                    subject_edit = subject.get_subject(subjects, _subject)
                    if subject_edit:
                        subject_edit.add_grade(_category, subject.Grade(_grade, _grade_name))
                elif _edit == "category":
                    _category = user_input[3]
                    _rating = user_input[4]
                    subject_edit = subject.get_subject(subjects, _subject)
                    if subject_edit:
                        subject_edit.add_grade_category(_category, _rating)
                elif _edit == "subject":
                    _main = user_input[3]
                    _oral = user_input[4]
                    subjects.append(subject.Subject(_subject, _main, _oral))
                else:
                    raise exceptions.WrongCaseException
            elif input_or_output == "output":
                _subject = user_input[1]
                subject_get = subject.get_subject(subjects, _subject)
                user_IO.user_output(subject_get.name, subject_get.return_grade_categories(),
                                [subject_get.return_average_of_category(i) for i in subject_get.return_grade_categories()],
                                subject_get.return_average_of_subject())
            elif input_or_output == "exit":
                break
            else:
                raise exceptions.WrongCaseException

        except (exceptions.NotANumberException, exceptions.CategoryNotExistingException,
                exceptions.SubjectNotExistingException, exceptions.WrongCaseException) as e:
            user_IO.exception_raised(e)

    

except Exception as e:
    print('ERROR' + str(e))
    raise