X-Git-Url: https://code.communitydata.science/coldcallbot-discord.git/blobdiff_plain/56fb61e8b00a0cde7cde1138c3f49b89b302e773..f7270293f2febd56d1570fef67085223d5d91d33:/coldcallbot-manual.py

diff --git a/coldcallbot-manual.py b/coldcallbot-manual.py
index 75a88dc..985ada0 100755
--- a/coldcallbot-manual.py
+++ b/coldcallbot-manual.py
@@ -3,7 +3,19 @@
 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:
@@ -28,6 +40,13 @@ with open(config["roster_file"], 'r') as f:
         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
 
@@ -37,17 +56,39 @@ preferred_names = cc.get_preferred_names()
 students_present = [s for s in registered_students if s not in missing_today]
 # 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)