A simple script to email secret Santa participants.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

191 lines
5.4 KiB

import yaml
import re
import random
import smtplib
import datetime
import pytz
import time
import socket
import sys
import getopt
import os
help_message = '''
To use, fill out config.yml with your own participants. You can also specify
DONT-PAIR so that people don't get assigned their significant other.
You'll also need to specify your mail server settings. An example is provided
for routing mail through gmail.
For more information, see README.
'''
REQRD = (
'SMTP_SERVER',
'SMTP_PORT',
'USERNAME',
'PASSWORD',
'TIMEZONE',
'PARTICIPANTS',
'DONT-PAIR',
'FROM',
'SUBJECT',
'MESSAGE',
)
HEADER = """Date: {date}
Content-Type: text/plain; charset="utf-8"
Message-Id: {message_id}
From: {frm}
To: {to}
Subject: {subject}
"""
CONFIG_PATH = os.path.join(os.path.dirname(__file__), 'config.yml')
class Person:
def __init__(self, name, email, invalid_matches, wishlist):
self.name = name
self.email = email
self.invalid_matches = invalid_matches
self.wishlist = wishlist
def __str__(self):
return "{} <{}>".format(self.name, self.email)
class Pair:
def __init__(self, giver, receiver):
self.giver = giver
self.receiver = receiver
def __str__(self):
return "{} ---> {}".format(self.giver.name, self.receiver.name)
def parse_yaml(yaml_path=CONFIG_PATH):
return yaml.load(open(yaml_path), yaml.Loader)
def choose_receiver(giver, receivers):
choice = random.choice(receivers)
if choice.name in giver.invalid_matches or giver.name == choice.name:
if len(receivers) == 1:
raise Exception('Only one receiver left, try again')
return choose_receiver(giver, receivers)
else:
return choice
def create_pairs(g, r):
givers = g[:]
receivers = r[:]
pairs = []
for giver in givers:
try:
receiver = choose_receiver(giver, receivers)
receivers.remove(receiver)
pairs.append(Pair(giver, receiver))
except:
return create_pairs(g, r)
return pairs
class Usage(Exception):
def __init__(self, msg):
self.msg = msg
def main(argv=None):
if argv is None:
argv = sys.argv
try:
try:
opts, args = getopt.getopt(argv[1:], "shc", ["send", "help"])
except getopt.error as msg:
raise Usage(msg)
# option processing
send = False
for option, value in opts:
if option in ("-s", "--send"):
send = True
if option in ("-h", "--help"):
raise Usage(help_message)
config = parse_yaml()
for key in REQRD:
if key not in config.keys():
raise Exception(
'Required parameter {} not in yaml config file!'.format(key,))
participants = config['PARTICIPANTS']
dont_pair = config['DONT-PAIR']
if len(participants) < 2:
raise Exception('Not enough participants specified.')
givers = []
for person in participants:
name, email, wishlist = person.split(" ", maxsplit=2)
name = name.strip()
invalid_matches = []
for pair in dont_pair:
names = [n.strip() for n in pair.split(',')]
if name in names:
# is part of this pair
for member in names:
if name != member:
invalid_matches.append(member)
person = Person(name, email, invalid_matches, wishlist)
givers.append(person)
receivers = givers[:]
pairs = create_pairs(givers, receivers)
if not send:
print("""
Test pairings:
{}
To send out emails with new pairings,
call with the --send argument:
$ python secret_santa.py --send
""".format("\n".join([str(p) for p in pairs])))
if send:
server = smtplib.SMTP(config['SMTP_SERVER'], config['SMTP_PORT'])
server.starttls()
server.login(config['USERNAME'], config['PASSWORD'])
for pair in pairs:
zone = pytz.timezone(config['TIMEZONE'])
now = zone.localize(datetime.datetime.now())
date = now.strftime('%a, %d %b %Y %T %Z') # Sun, 21 Dec 2008 06:25:23 +0000
message_id = '<{}@{}>'.format(str(time.time())+str(random.random()), socket.gethostname())
frm = config['FROM']
to = pair.giver.email
subject = config['SUBJECT'].format(santa=pair.giver.name, santee=pair.receiver.name)
body = (HEADER+config['MESSAGE']).format(
date=date,
message_id=message_id,
frm=frm,
to=to,
subject=subject,
santa=pair.giver.name,
santee=pair.receiver.name,
gifts_suggestions=pair.receiver.wishlist,
)
if send:
result = server.sendmail(frm, [to], body.encode('utf-8'))
print("Emailed {} <{}>" .format(pair.giver.name, to))
if send:
server.quit()
except Usage as err:
print >> sys.stderr, sys.argv[0].split("/")[-1] + ": " + str(err.msg)
print >> sys.stderr, "\t for help use --help"
return 2
if __name__ == "__main__":
# sys.exit(main())
main()