]> code.communitydata.science - coldcallbot-discord.git/blobdiff - coldcall.py
print preferred pronouns in the call list
[coldcallbot-discord.git] / coldcall.py
index 8cae6ca71e3fcc8791683ecce11e9b6497c8a5d4..3fb79d663c7c694b751710e48ad0f64a8bbaaf13 100644 (file)
@@ -8,19 +8,30 @@ from csv import DictReader
 
 import os.path
 import re
 
 import os.path
 import re
-import discord
+import json
 
 class ColdCall():
 
 class ColdCall():
-    def __init__ (self):
+    def __init__ (self, record_attendance=True):
+        with open("configuration.json") as config_file:
+            config = json.loads(config_file.read())
+
         self.today = str(datetime.date(datetime.now()))
         self.today = str(datetime.date(datetime.now()))
+
         # how much less likely should it be that a student is called upon?
         # how much less likely should it be that a student is called upon?
-        self.weight = 2 
+        self.weight = 2
+        self.record_attendance = record_attendance
 
         # filenames
 
         # 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_studentinfo = config["student_info_file"]
+        self.__fn_daily_calllist = config["daily_calllist_file"].format(date=self.today)
+        self.__fn_daily_attendance = config["daily_attendance"].format(date=self.today)
 
 
+        self.unique_row = config["unique_name_rowname"]
+        if "preferred_name_rowname" in config:
+            self.preferred_row = config["preferred_name_rowname"]
+        else:
+            self.preferred_row = None
+        
     def __load_prev_questions(self):
         previous_questions = defaultdict(int)
 
     def __load_prev_questions(self):
         previous_questions = defaultdict(int)
 
@@ -28,29 +39,30 @@ class ColdCall():
             if re.match("call_list-\d{4}-\d{2}-\d{2}.tsv", fn):
                 with open(f"./data/{fn}", 'r') as f:
                     for row in DictReader(f, delimiter="\t"):
             if re.match("call_list-\d{4}-\d{2}-\d{2}.tsv", fn):
                 with open(f"./data/{fn}", 'r') as f:
                     for row in DictReader(f, delimiter="\t"):
-                        previous_questions[row["discord_name"]] += 1
+                        if not row["answered"] == "FALSE":
+                            previous_questions[row[self.unique_row]] += 1
 
         return previous_questions
 
         return previous_questions
-    
-    def __get_preferred_name(self, selected_student):
-        # translate the discord name into the preferred students name,
-        # if possible, otherwise return the discord name
+
+    def get_preferred_names(self):
+        # translate the unique name into the preferred students name,
+        # if possible, otherwise return the unique name
 
         preferred_names = {}
         with open(self.__fn_studentinfo, 'r') as f:
 
         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[self.unique_row]] = row[self.preferred_row]
 
 
+        return(preferred_names)
+        
+    def __get_preferred_name(self, selected_student):
+        preferred_names = self.get_preferred_names()
         if selected_student in preferred_names:
             return preferred_names[selected_student]
         else:
             return None
 
         if selected_student in preferred_names:
             return preferred_names[selected_student]
         else:
             return None
 
-    def __select_student_from_list (self, students_present):
+    def select_student_from_list(self, students_present):
         prev_questions = self.__load_prev_questions()
         
         # created a weighted list by starting out with everybody 1
         prev_questions = self.__load_prev_questions()
         
         # created a weighted list by starting out with everybody 1
@@ -62,38 +74,43 @@ class ColdCall():
                 weights[s] = weights[s] / self.weight
 
         # choose one student from the weighted list
                 weights[s] = weights[s] / self.weight
 
         # choose one student from the weighted list
-        print(weights)
+        # print(weights) # DEBUG LINE
         return choices(list(weights.keys()), weights=list(weights.values()), k=1)[0]
 
         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 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
 
         # 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)
 
             print("\t".join([str(datetime.now()),
                              ",".join(students_present)]),
                   file=f)
 
-    def __record_coldcall(self, selected_student):
+    def record_coldcall(self, selected_student):
         # if it's the first one of the day, write it out
         if not os.path.exists(self.__fn_daily_calllist):
             with open(self.__fn_daily_calllist, "w") as f:
         # if it's the first one of the day, write it out
         if not os.path.exists(self.__fn_daily_calllist):
             with open(self.__fn_daily_calllist, "w") as f:
-                print("\t".join(["discord_name", "timestamp", "answered", "assessment"]), file=f)
+                print("\t".join([self.unique_row, self.preferred_row, "answered", "assessment", "timestamp"]), file=f)
+
+        preferred_name = self.__get_preferred_name(selected_student)
+        if preferred_name == None:
+            preferred_name = ""
 
         # open for appending the student
         with open(self.__fn_daily_calllist, "a") as f:
 
         # open for appending the student
         with open(self.__fn_daily_calllist, "a") as f:
-            print("\t".join([selected_student, str(datetime.now()),
-                             "MISSING", "MISSING"]), file=f)
+            print("\t".join([selected_student, preferred_name,
+                             "MISSING", "MISSING", str(datetime.now())]), file=f)
 
     def coldcall(self, students_present):
 
     def coldcall(self, students_present):
-        selected_student = self.__select_student_from_list(students_present)
+        selected_student = self.select_student_from_list(students_present)
 
         # record the called-upon student in the right place
 
         # record the called-upon student in the right place
-        self.__record_attendence(students_present)
-        self.__record_coldcall(selected_student)
+        if self.record_attendance:
+            self.record_attendance(students_present)
+        self.record_coldcall(selected_student)
 
         preferred_name = self.__get_preferred_name(selected_student)
         if preferred_name:
 
         preferred_name = self.__get_preferred_name(selected_student)
         if preferred_name:
@@ -102,13 +119,3 @@ class ColdCall():
             coldcall_message = f"@{selected_student}, you're up!"
         return coldcall_message
 
             coldcall_message = f"@{selected_student}, you're up!"
         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"]
-# cc.coldcall(test_student_list)
-
-# test_student_list = ["Jordan", "Kristen Larrick", "Mako"]
-# cc.coldcall(test_student_list)
-
-# test_student_list = ["Jordan", "Kristen Larrick"]
-# cc.coldcall(test_student_list)

Community Data Science Collective || Want to submit a patch?