santa sol

gistfile1.txt

# draw things out of a hat

from pprint import pprint
import re
import random

NAME_REGEXP = re.compile("(?P<name>\w+) (?P<last>\w+) <(?P<mail>.*?)>")


NAMES = [
    "luke skywalker <[email protected]>",
    "leia skywalker <[email protected]>",
    "toula portokalos <[email protected]>",
    "gus portokalos <[email protected]>",
    "bruce wayne <[email protected]>",
    "virgil brigman <[email protected]>",
    "lindsley brigman <[email protected]>"
]


class Person(object):
    def __init__(self, name, last, mail):
        self.name = name
        self.last = last
        self.mail = mail

    def __repr__(self):
        return str(self)

    def __str__(self):
        return "%s %s" % (self.name, self.last)

    def is_parent(self, other):
        return self.last == other.last


def gen_people(names):
    people = []
    for name in names:
        match = NAME_REGEXP.match(name)
        person = Person(**match.groupdict())
        people.append(person)

    random.shuffle(people)
    return people


def santa_select(names):
    giver_receiver = {}
    people = gen_people(names)
    while True:
        not_givers = set(people) - set(giver_receiver.keys())
        if not not_givers:
            break

        next = not_givers.pop()
        is_possible = lambda x: (x != next) and \
            not (x.is_parent(next)) and \
            x not in giver_receiver.values()

        possible_receivers = filter(is_possible, people)
        possible_receivers.sort(key=lambda x: x not in giver_receiver)
        recv = possible_receivers.pop()
        giver_receiver[next] = recv
        print(giver_receiver)

    return giver_receiver


def test_regexp():
    match = NAME_REGEXP.match("luke skywalker <[email protected]>")
    assert match.group('name') == 'luke'


if __name__ == '__main__':
    test_regexp()
    result = santa_select(NAMES)
    pprint(result)
    print(len(result))