[wip] added get_student

reidoko · Feb 3, 2022 · c0d8831 · c0d8831
1 parent 81ccd9a
commit c0d8831
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/core/course.py b/core/course.py
@@ -47,3 +47,13 @@ def get_roster(self) -> List[GradescopeStudent]:
                     )
 
         return self.roster
+
+    def get_student(self, sid: Optional[str] = None, email: Optional[str] = None) -> Optional[GradescopeStudent]:
+        assert sid or email
+        roster = self.get_roster()
+        for student in roster:
+            if sid != None and student.sid == sid:
+                return student
+            if email != None and student.email == email:
+                return student
+        return None