updated cold call script to do a pure shuffle

[coldcallbot-discord.git] / coldcallbot-manual.py

diff --git a/coldcallbot-manual.py b/coldcallbot-manual.py
index 6c128bac844e7cf7bd99e8006105543c62535783..985ada0e9d32ee92d894fc57c0998855b1556911 100755 (executable)
--- a/coldcallbot-manual.py
+++ b/coldcallbot-manual.py
@@ -3,51 +3,92 @@
 from coldcall import ColdCall
 from datetime import datetime
 from csv import DictReader
+from random import sample
+import json
+import argparse
+
+parser = argparse.ArgumentParser(description='run the coldcall bot manually to create a coldcall list')
+
+parser.add_argument('-n', '--num', dest="num_calls", default=100, const=100, type=int, nargs='?',
+                    help="how many students should be called")
+
+parser.add_argument('-s', '--shuffle', dest="shuffle_roster", action="store_true",
+                    help="select without replacement (i.e., call each person once with n equal to the group size)")
+
+args = parser.parse_args()
 
 current_time = datetime.today()
+with open("configuration.json") as config_file:
+    config = json.loads(config_file.read())
 
 ## create the coldcall object
-cc = ColdCall(record_attendance=False, preferred_name_field="Name you'd like to go by in class")
+cc = ColdCall(record_attendance=False)
 
 def get_missing(d=current_time):
     date_string = f'{d.month}/{d.day}/{d.year}'
-    with open("data/absence_poll_data.tsv", 'r') as f:
+    with open(config["optout_file"], 'r') as f:
         for row in DictReader(f, delimiter="\t"):
             if row["Date of class session you will be absent"] == date_string:
-                yield(row["Your UW student number"])
+                yield(row[config["unique_name_rowname"]])
 
 full_names = {}
 registered_students = []
-with open("data/2022_winter_COM_481_A_students.csv", 'r') as f:
+with open(config["roster_file"], 'r') as f:
     for row in DictReader(f, delimiter=","):
         student_no = row["StudentNo"].strip()
         registered_students.append(student_no)
-        full_names[student_no] = f"{row['FirstName']} {row['LastName']}"
-## print("Registered:", registered_students)
+        full_names[student_no] = f"{row[config['roster_firstname_rowname']]} {row[config['roster_lastname_rowname']]}"
+# print("Registered:", registered_students) # useful for debug
+
+# get pronouns
+with open(config["student_info_file"], 'r') as f:
+    preferred_pronouns = {}
+    for row in DictReader(f, delimiter="\t"):
+        preferred_pronouns[row[config["unique_name_rowname"]]] = row["Preferred pronouns"]
+# print(preferred_pronouns)
 
 missing_today = [x for x in get_missing(current_time)]
-## print("Missing Today: ", missing_today)
+# print("Missing Today: ", missing_today)  # useful for debug
 
-preferred_names = {}
-with open("data/student_information.tsv", 'r') as f:
-    for row in DictReader(f, delimiter="\t"):
-        preferred_names[row["Your UW student number"]] = row["Name you'd like to go by in class"]
-## print("Preferred names:", preferred_names)
+preferred_names = cc.get_preferred_names()
+# print("Preferred names:", preferred_names)  # useful for debug
 
 students_present = [s for s in registered_students if s not in missing_today]
-## print("Students present:", students_present)
+# print("Students present:", students_present)  # useful for debug
 
-for i in range(100):
-    selected_student = cc.select_student_from_list(students_present)
+def print_selected(selected_student):
+    if "print_index" in globals():
+        global print_index
+    else:
+        global print_index
+        print_index = 1
 
     try:
         preferred_name = preferred_names[selected_student]
     except KeyError:
-        preferred_name = "MISSING PREFERRED NAME"
+        preferred_name = "[unknown preferred name]"
+
+    if selected_student in preferred_pronouns:
+        pronouns = preferred_pronouns[selected_student]
+    else:
+        pronouns = "[unknown pronouns]"
+
+    print(f"{print_index}. {preferred_name} :: {pronouns} :: {full_names[selected_student]} :: {selected_student}")
 
-    print(f"{i + 1}.",
-          preferred_name, "::",
-          selected_student, "::",
-          full_names[selected_student])
     cc.record_coldcall(selected_student)
+    print_index += 1 ## increase the index
+
+# if we're in suffle mode
+shuffle = args.shuffle_roster
+
+print_index = 1
+
+if shuffle:
+    for selected_student in sample(students_present, len(students_present)):
+        print_selected(selected_student)
+else:
+    num_calls = args.num_calls
 
+    for i in range(num_calls):
+        selected_student = cc.select_student_from_list(students_present)
+        print_selected(selected_student)