From ad2fad6b292993030d62bb90a09e350cb7f99b31 Mon Sep 17 00:00:00 2001 From: Lars Bogner Date: Wed, 31 Jul 2019 19:36:22 +0200 Subject: [PATCH] error handling (issue n#9) --- main.py | 25 ++++++++++++++++++------- 1 file changed, 18 insertions(+), 7 deletions(-) diff --git a/main.py b/main.py index b340ca6..209796f 100644 --- a/main.py +++ b/main.py @@ -2,10 +2,21 @@ import subject import user_IO subjects = [] -while True: - user_input = user_IO.user_input() - if user_input[0] == "input": - if user_input[1] == "grade": - subject_edit = subject.get_subject(subjects, user_input[2]) - if subject_edit: - subject_edit.add_grade(user_input[3], subject.Grade(user_input[4], user_input[5])) +try: + while True: + user_input = user_IO.user_input() + if user_input[0] == "input": + if user_input[1] == "grade": + subject_edit = subject.get_subject(subjects, user_input[2]) + if subject_edit: + subject_edit.add_grade(user_input[3], subject.Grade(user_input[4], user_input[5])) + elif user_input[1] == "category": + subject_edit = subject.get_subject(subjects, user_input[2]) + if subject_edit: + subject_edit.add_grade_category(user_input[3], user_input[4]) + elif user_input[1] == "subject": + subjects.append(subject.Subject(user_input[2], _, _)) + raise SystemError + +except Exception as e: + print('ERROR' + str(e)) \ No newline at end of file