error handling (issue n#9)

This commit is contained in:
2019-07-31 19:36:22 +02:00
parent 44a5fbc4b7
commit ad2fad6b29
+18 -7
View File
@@ -2,10 +2,21 @@ import subject
import user_IO import user_IO
subjects = [] subjects = []
while True: try:
user_input = user_IO.user_input() while True:
if user_input[0] == "input": user_input = user_IO.user_input()
if user_input[1] == "grade": if user_input[0] == "input":
subject_edit = subject.get_subject(subjects, user_input[2]) if user_input[1] == "grade":
if subject_edit: subject_edit = subject.get_subject(subjects, user_input[2])
subject_edit.add_grade(user_input[3], subject.Grade(user_input[4], user_input[5])) 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))