feat: implement email and argument handeling

This commit is contained in:
Stefan Sterz 2018-09-08 21:02:29 +00:00
parent 6066a150ea
commit 0d1c1878b1
3 changed files with 63 additions and 4 deletions

7
config.json Normal file
View File

@ -0,0 +1,7 @@
{
"server":"your.email.server",
"port":587,
"user":"you@yourmail.com",
"pass":"yourPassword",
"sender":"santa@yourmail.com"
}

3
participants.json Normal file
View File

@ -0,0 +1,3 @@
[
{"name":"Alice","mail":"alice@mail.com","gender":"e"},{"name":"Bob","mail":"bob@mail.com","gender":"er"},{"name":"Chris","mail":"chris@mail.com","gender":"er"}
]

View File

@ -1,8 +1,33 @@
'''
This script creates randoms pairs from a list of participants in a secret santa
game and sends them an email to notify them who the person is that they should
buy presents for.
'''
# -*- encoding: utf-8 -*-
from argparse import ArgumentParser
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
import json
import random
import smtplib
parser = ArgumentParser()
parser.add_argument('-p', '--participants',
dest='pFileLocation',
default='participants.json',
help='read list of participants from FILE',
metavar='PARTICIPANTFILE')
parser.add_argument('-c', '--config',
dest='cFileLocation',
default='config.json',
help='read the configuration from FILE',
metavar='CONFIGFILE')
args = parser.parse_args()
# load and prepare a list containing all participants
with open('participants.json', 'r') as pFile:
with open(args.pFileLocation, 'r') as pFile:
participants = json.load(pFile)
# create a copy of the list to make choosing a partner easier
@ -20,7 +45,7 @@ for i in range(len(participants)):
partner = random.choice(participants)
current['partner'] = partner['partner']
partner['partner'] = current['name'] + ' ' + current['mail']
partner['partner'] = current['name'] + ' (' + current['mail']+')'
participants.append(current)
break
@ -30,8 +55,32 @@ for i in range(len(participants)):
while partner[0] == i:
partner = random.choice(copy)
participants[i]['partner'] = partner[1]['name']+' '+partner[1]['mail']
participants[i]['partner'] = partner[1]['name']+' ('+partner[1]['mail']+')'
copy.remove(partner)
print(participants)
# load mail server configuration and open smtp server connection
with open(args.cFileLocation, 'r') as cFile:
config = json.load(cFile)
server = smtplib.SMTP(config['server'], config['port'])
server.ehlo()
server.starttls()
server.ehlo()
server.login(config['user'], config['pass'])
# create emails for all participants and send them
# change the text for the mail here if you want to
for r in participants:
msg = MIMEMultipart('alternative')
msg['From'] = config['sender']
msg['To'] = r['mail']
msg['Subject'] = 'Wichteln 🎄'
body = 'Lieb'+r['gender']+' '+r['name']+'!'
body += '\n\nDu bist heuer Wichtel für '+r['partner']
body += '. \n Das Geschenk sollte nicht mehr als 50€ kosten.'
body += '\nFrohe Weihnachten!\n\nDein Christkind 👼'
msg.attach(MIMEText(body, 'plain', 'utf-8'))
server.sendmail(config['sender'], r['mail'], msg.as_string())