shuffler.py

import random
from datetime import date, datetime

from utils import extract_name_from_raw

EXCEPTIONAL_NAMES = ['@weichenlin_']


def extract_request_content(request):
  request = request[request.find('1.'):]
  request = request.split('- Venmo')[0]
  return request.strip()


def get_random_seed(date):
  return int(date.strftime('%Y%m%d'))


def remove_exceptional_names(all_names):
  names = []
  for name in all_names:
    if name[0] in EXCEPTIONAL_NAMES:
      continue
    names.append(name)
  return names


def get_shuffle_response(request,
                         date,
                         enable_extraction=False,
                         enable_pair_on_duty=True,
                         enable_index_to_name=False):
  if enable_extraction:
    request = extract_request_content(request)
  all_names = extract_name_from_raw(request)
  names = remove_exceptional_names(all_names)
  slots = ['1:30', '2:00', '2:30', '3:00', '3:30', '4:00', '4:30', '5:00']

  random.seed(get_random_seed(date))
  selected_num_multipier = 2 if enable_pair_on_duty else 1
  selected = random.sample(names, len(slots) * selected_num_multipier)

  response = 'On-duty ' + str(date) + ' (' + str(
      len(all_names)) + ' attended)\n--\n'
  for i in range(len(slots)):
    end_slot = slots[i + 1] if i + 1 < len(slots) else 'end'
    if enable_pair_on_duty:
      on_duty_name = get_printed_name(selected[i*2], enable_index_to_name) + \
          ' + ' + get_printed_name(selected[i*2 + 1], enable_index_to_name)
    else:
      on_duty_name = get_printed_name(selected[i], enable_index_to_name)
    response += slots[i] + '-' + end_slot + ': ' + on_duty_name + '\n'
  return response + get_instructions()


def get_printed_name(name, enable_index_to_name):
  return name[0] + ' (' + name[1] + ')' if enable_index_to_name else name[0]


def get_instructions():
  return '''--
Instruction

1. If there are more than 4 "unassigned" people, sign every 4 people to a court until a court has 8 of our people and there are no more than 4 "unassigned" people.
2. If a signed court ends in 25 mins, unassigned the first 4 people and reassign them back to the queue.
3. Update all changes within the Instagram group chat.
4. Remind the next on-duty (and pass the pager to them).

1. 在群裡，如果"unassigned"達4人，把每4人sign到未滿8人的場（優先選已經有sign 4人的場，再選附近的快結束的場）。
2. 在電腦上，任何有我們sign的球場，如果剩下時間在25分鐘內，把在場上的4人unassign後再sign回queue去。
3. 做任何變更之後，更新群裡面的訊息。
4. 結束時提醒下一個On-duty，把pager給他。
--
Note: On-Duty需要調整的人，自行跟其他團員協調即可。'''