mailbot.py

import smtplib
import json
import sys
from email.mime.text import MIMEText as text

if len(sys.argv) != 2:
    print("Incorrect number of arguments.")
    sys.exit(0)

with open(sys.argv[1]) as data_file:
    data = json.load(data_file)

# populate data fields
from_address = data["name"]
username = data["username"]
password = data["password"]
raw_message = data["message"]
raw_subject = data["subject"]
variable_to = data["to"]
variable_dictionary = {}

# configure server
# try to see if there's a way to send from NYU servers
server = smtplib.SMTP('smtp.gmail.com:587')
if "@nyu.edu" in username:
    server = smtplib.SMTP('smtp.nyu.com:465')

# populate the dictionary of variables and replacers
list_length = -1
for rows in data["variables"]:
    var = rows["var"]
    list_vals = []
    for vals in rows["vals"]:
        list_vals.append(vals["val"])
    if (list_length == -1):
        list_length = len(list_vals)
    elif list_length != len(list_vals):
        print("Error: every list of variables must be of the same length.")
        sys.exit(0)
    variable_dictionary[var] = list_vals

server.ehlo()
server.starttls()
server.login(username, password)

# replace variables in the message
for i in range(0, list_length):
    message = raw_message
    subject = raw_subject
    to_address = variable_dictionary[variable_to][i]
    sys.stdout.write("Sending email to " + to_address + "... ")
    for key, values in variable_dictionary.iteritems():
        message = message.replace(key, values[i])
        subject = subject.replace(key, values[i])

    m = text(message)
    m['Subject'] = subject
    m['From'] = from_address
    m['To'] = to_address

    # send the message out
    server.sendmail(from_address, to_address, m.as_string())
    sys.stdout.write("Done!\n")

server.quit()