def get_course():
print("Available Courses:")
[print(course.code) for course in course_list]
user_input = get_string("Select course")
matching_courses = [
course for course in course_list if course.code == user_input
]
if not matching_courses:
get_course()
course = matching_courses[0]
return course
def get_student_course(student):
print("Enrolled courses:")
student_courses = [
course for course, grade in student.course_grade.items()
]
if not student_courses:
print("No enrolled courses!")
else:
for course in student_courses:
print(course.code)
# [print(course) for course in student_courses]
user_input = get_string("Select course")
matching_courses = [
course for course in student_courses if course.code == user_input
]
if not matching_courses:
get_student_course(student)
return matching_courses[0]
def edit_name():
new_name = get_string("Enter your new name")
student.name = new_name
def edit_code():
new_code = get_string("Enter new code")
course.code = new_code
def add_course():
code = get_string("Enter Code")
unit = get_int("Enter unit")
course_list.append(Course(unit=unit, code=code))
print(course_list[0].code) # TEST
def register():
name = get_string("Enter name")
id = get_int("Enter ID Number")
student_list.append(Student(name=name, id=id))
print(f"Welcome {name}!")