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 a4268eadeebefde7128dc9945e1f50ab0d98645b..985ada0e9d32ee92d894fc57c0998855b1556911 100755 (executable)
--- a/coldcallbot-manual.py
+++ b/coldcallbot-manual.py
@@ -1,15 +1,94 @@
 #!/usr/bin/env python3
 
 from coldcall import ColdCall
-import re
+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)
 
-student_list = cc.preferred_names
+def get_missing(d=current_time):
+    date_string = f'{d.month}/{d.day}/{d.year}'
+    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[config["unique_name_rowname"]])
+
+full_names = {}
+registered_students = []
+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[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)  # useful for debug
+
+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)  # useful for debug
+
+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 = "[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}")
+
+    cc.record_coldcall(selected_student)
+    print_index += 1 ## increase the index
+
+# if we're in suffle mode
+shuffle = args.shuffle_roster
 
-# print out 100 students
+print_index = 1
 
-for i in range(100):
-    print(f"{i + 1}. {cc.coldcall(student_list)} [ ] [ ]\n")
+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)