diff --git a/project/app/models.py b/project/app/models.py index e754f3c..4237a9e 100644 --- a/project/app/models.py +++ b/project/app/models.py @@ -47,3 +47,7 @@ class Roster: return True return False """Remove student(s).""" + + def reset_roster(self): + """Reset students to their original list (like at the start of the day).""" + self.students = self.original_students[:] diff --git a/project/app/views.py b/project/app/views.py index d617783..c457f66 100644 --- a/project/app/views.py +++ b/project/app/views.py @@ -9,7 +9,38 @@ print(f"{roster.count_students()} students remaining on Roster.") student = roster.pick_random_stuent() if student: - print(f"[{student}, you were selected!!") + print(f"[{student}, you were selected!!]") else: print("Name not/no longer on current roster.") + +roster = Roster() + +roster.add_student("Bob") +roster.add_student("Linda") +roster.add_student("Tina") +roster.add_student("Gene") +roster.add_student("Louis") +roster.add_student("Gayle, Checkin No Cold Call") +roster.add_student("Mort") +roster.add_student("Teddy") + +removed = roster.remove_student() +if removed: + print(f"[{student} was removed from Roster.]") +else: + print(f"[{student} not on roster.]") + +roster.load_from_list(["Ava", "Eli", "Noah", "Luna"]) + +print("Full class:", roster.students) + +roster.remove_student("Bob") +roster.remove_student("Linda") +print("Remaining today:", roster.students) + +"""Reset at end of day""" +roster.reset_roster() +print("Next day reset:", roster.students) + +