#!/usr/bin/env python3

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)

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_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)