class Student: def __init__(self, name: str, courses=None): if courses is None: courses = {} self.name = name self.courses = courses # {course_name: [notes]} def enroll(self, course_name: str, notes, add_course_notes: str = ""): if course_name in self.courses.keys(): [self.courses[course_name].append(x) for x in notes] return "Course already added. Notes have been updated." if add_course_notes == "Y" or add_course_notes == "": self.courses[course_name] = notes return "Course and course notes have been added." self.courses[course_name] = [] return "Course has been added." def add_notes(self, course_name, notes): if course_name in self.courses.keys(): self.courses[course_name].append(notes) return "Notes have been updated" raise Exception("Cannot add notes. Course not found.") def leave_course(self, course_name): if course_name in self.courses.keys(): self.courses.pop(course_name) return "Course has been removed" raise Exception("Cannot remove course. Course not found.")