added code to compute /days/ present or absent

[coldcallbot-discord.git] / coldcall.py

diff --git a/coldcall.py b/coldcall.py
index 8cae6ca71e3fcc8791683ecce11e9b6497c8a5d4..4987c741978a1826c4505a74eae90808251f1494 100644 (file)
--- a/coldcall.py
+++ b/coldcall.py
@@ -19,7 +19,7 @@ class ColdCall():
         # filenames
         self.__fn_studentinfo = "data/student_information.tsv"
         self.__fn_daily_calllist = f"data/call_list-{self.today}.tsv"
-        self.__fn_daily_attendence = f"data/attendence-{self.today}.tsv"
+        self.__fn_daily_attendance = f"data/attendance-{self.today}.tsv"
 
     def __load_prev_questions(self):
         previous_questions = defaultdict(int)
@@ -38,12 +38,8 @@ class ColdCall():
 
         preferred_names = {}
         with open(self.__fn_studentinfo, 'r') as f:
-            for line in f.readlines():
-                x = line.strip().split("\t")
-                if x[0] == "Timestamp":
-                    continue
-
-                preferred_names[x[4]] = x[2]
+            for row in DictReader(f, delimiter="\t"):
+                preferred_names[row["Your username on the class Discord server"]] = row["Name you'd like to go by in class"]
 
         if selected_student in preferred_names:
             return preferred_names[selected_student]
@@ -65,14 +61,14 @@ class ColdCall():
         print(weights)
         return choices(list(weights.keys()), weights=list(weights.values()), k=1)[0]
 
-    def __record_attendence(self, students_present):
+    def __record_attendance(self, students_present):
         # if it's the first one of the day, write it out
-        if not os.path.exists(self.__fn_daily_attendence):
-            with open(self.__fn_daily_attendence, "w") as f:
-                print("\t".join(["timestamp", "attendence_list"]), file=f)
+        if not os.path.exists(self.__fn_daily_attendance):
+            with open(self.__fn_daily_attendance, "w") as f:
+                print("\t".join(["timestamp", "attendance_list"]), file=f)
 
         # open for appending the student
-        with open(self.__fn_daily_attendence, "a") as f:
+        with open(self.__fn_daily_attendance, "a") as f:
             print("\t".join([str(datetime.now()),
                              ",".join(students_present)]),
                   file=f)
@@ -92,7 +88,7 @@ class ColdCall():
         selected_student = self.__select_student_from_list(students_present)
 
         # record the called-upon student in the right place
-        self.__record_attendence(students_present)
+        self.__record_attendance(students_present)
         self.__record_coldcall(selected_student)
 
         preferred_name = self.__get_preferred_name(selected_student)
@@ -103,12 +99,12 @@ class ColdCall():
         return coldcall_message
 
 # cc = ColdCall()
+ 
+# test_student_list = ["jordan", "Kristen Larrick", "Madison Heisterman", "Maria.Au20", "Laura (Alia) Levi", "Leona Aklipi", "Linya Feng", "anne", "kirst", "emmaaitelli", "ashleylee", "allie_partridge", "Tiana_Cole", "Hamin", "Ella Qu", "Angel Su", "Shizuka", "Ben Baird", "Kim Do", "Isaacm24", "Sam Bell", "Courtneylg", "TasnimHasan"]
+# print(cc.coldcall(test_student_list))
 
-# test_student_list = ["Jordan", "Kristen Larrick", "Madison Heisterman", "Maria.Au20", "Laura (Alia) Levi", "Leona Aklipi", "Linya Feng", "anne", "kirst", "emmaaitelli", "ashleylee", "allie_partridge", "Tiana_Cole", "Hamin", "Ella Qu", "Angel Su", "Shizuka", "Ben Baird", "Kim Do", "Isaacm24", "Sam Bell", "Courtneylg", "TasnimHasan"]
-# cc.coldcall(test_student_list)
-
-# test_student_list = ["Jordan", "Kristen Larrick", "Mako"]
-# cc.coldcall(test_student_list)
+# test_student_list = ["jordan", "Kristen Larrick", "Mako"]
+# print(cc.coldcall(test_student_list))
 
-# test_student_list = ["Jordan", "Kristen Larrick"]
-# cc.coldcall(test_student_list)
+# test_student_list = ["jordan", "Kristen Larrick"]
+# print(cc.coldcall(test_student_list))