import smtplib
from email.mime.application import MIMEApplication
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
from string import Template
from config import gmail_login, gmail_password
import os
import csv
import sys
import xlrd

root = "/home/share/serenitas/Fund Raising"

book = xlrd.open_workbook(os.path.join(root, sys.argv[1]))
sheet = book.sheet_by_name("Distribution List")
emails, names, include, firm = sheet.col(4), sheet.col(5), sheet.col(6), sheet.col(2)
mailing_list = [(emails[i].value, names[i].value, firm[i].value) for i, y in enumerate(include[3:], 3) if y.value.lower()=="x"]

#SMTP server
server = smtplib.SMTP('smtp.gmail.com:587')
server.starttls()
server.login(gmail_login, gmail_password)

with open("templates/template-2014-06-13.txt") as fh:
    s_text = Template(fh.read())

with open("templates/template-2014-06-13.html") as fh:
    s_html = Template(fh.read())

attachment_name = 'Serenitas Capital (SCGMF Returns).pdf'
with open(os.path.join(root, "Pitchbook", attachment_name), 'rb') as fh:
    pdf_attach1 = MIMEApplication(fh.read(), 'pdf')
pdf_attach1.add_header('Content-Disposition', 'attachment', filename = attachment_name)

# attachment_name = 'Serenitas Credit Gamma Master Fund - Investor Letter 1Q2014.pdf'
# with open(os.path.join(root, "InvestorLetters", attachment_name), 'rb') as fh:
#     pdf_attach2 = MIMEApplication(fh.read(), 'pdf')
# pdf_attach2.add_header('Content-Disposition', 'attachment', filename = attachment_name)

for email, name, firm in mailing_list:
    msg = MIMEMultipart()
    msg['Subject'] = 'Serenitas Credit Gamma Master Fund performance update'
    msg['From'] = "David Weeks <{0}>".format(gmail_login)
    msg['To'] = email
    msg_alternative = MIMEMultipart('alternative')
    if name:
        name = " " + name
    # msg_alternative.attach(MIMEText(s_text.substitute(name = name, firm_name=firm.rstrip()), 'plain'))
    # msg_alternative.attach(MIMEText(s_html.substitute(name = name, firm_name=firm.rstrip()), 'html'))
    msg_alternative.attach(MIMEText(s_text.substitute(name = name), 'plain'))
    msg_alternative.attach(MIMEText(s_html.substitute(name = name), 'html'))
    msg.attach(msg_alternative)
    # msg.attach(pdf_attach2)
    msg.attach(pdf_attach1)
    server.send_message(msg)
server.quit()