Secret Santa Source Code

calculate_secret_santas.py

import csv
import random

input_spreadsheet_name = 'secret_santas_input.csv'
secret_spreadsheet_name = 'secret_santa_assigments.csv'


# recursive match making
def make_matches(remaining_people, next_santa=None, original_santa=None, processed_people={}):
    if len(remaining_people) is 0:
        return processed_people
    if next_santa is None:
        next_santa = random.choice(remaining_people)
    remaining_people.remove(next_santa)
    if original_santa is None:
        original_santa = next_santa
    if len(remaining_people) is 0:
        processed_people[next_santa] = original_santa
        return processed_people
    # randomly select a recipient from remaining pool
    recipient = random.choice(remaining_people)
    processed_people[next_santa] = recipient
    # make the receipient the next santa
    return make_matches(remaining_people, recipient, original_santa, processed_people)


with open(input_spreadsheet_name) as csv_file:
    csv_reader = csv.reader(csv_file, delimiter=',')
    line_count = 0
    unique_rows = 0
    duplicates = []
    people = {}
    for row in csv_reader:
        if line_count == 0:
            print(f'processing columns {", ".join(row)}')
        elif people.get(row[1], None) is not None:
            duplicates.append(row[1])
        else:
            unique_rows += 1
            people[row[1]] = {'email': row[1], 'address': row[2], 'name': row[3] or "Anonymous", 'gender': (row[4] or "none given"), 'interests': (
                row[5] or "none given"), 'allergies': (row[6] or "none given")}
        line_count += 1
    # assign one person to another randomly
    emails = [k for k in people.keys()]
    processed_emails = []
    matches = make_matches(emails)
    print(f'{line_count} lines in the spreadsheet. {unique_rows} unique emails (duplicates: {duplicates}). matches: {len(matches.keys())}')
    written_line_count = 0
    with open(secret_spreadsheet_name, 'w', newline='') as csvfile:
        fieldnames = ['santa_email', 'recipient_email', 'recipient_name', 'recipient_gender',
                      'recipient_address', 'recipient_interests', 'recipient_allergies']
        writer = csv.DictWriter(csvfile, fieldnames=fieldnames)

        writer.writeheader()
        for match in matches.keys():
            recipient = people[matches[match]]
            writer.writerow({'santa_email': match,
                             'recipient_email': recipient['email'],
                             'recipient_name': recipient['name'],
                             'recipient_gender': recipient['gender'],
                             'recipient_address': recipient['address'],
                             'recipient_interests': recipient['interests'],
                             'recipient_allergies': recipient['allergies']})
            written_line_count += 1
    print(f'{written_line_count} lines written to secret spreadsheet {secret_spreadsheet_name}')

secret_santas_assignments.csv

secret_santas_input.csv