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}!")