max_course_size = int(input("Enter maximum class size: "))
if max_course_size <= 0:
raise ValueError
exit_condition = True
except ValueError:
print("Max course size must be a positive integer")
continue
my_horse_of_course = Course(course_code=course_code,
max_class_size=max_course_size)
possible_options = (0, 1, 2, 3)
while True:
try:
bandit = int(
input(
"Enter 1 for add student, 2 for drop student, 3 for course info, 0 for exit: "
))
if bandit not in possible_options:
raise ValueError
if bandit == 1:
my_horse_of_course.add_student()
elif bandit == 2:
my_horse_of_course.drop_student()
elif bandit == 3:
my_horse_of_course.course_info()
elif bandit == 0:
my_horse_of_course.exit_program()
except ValueError:
print("Possible choices are:", possible_options)
continue
