names = {}

file = open(nameCsv,"rb")

reader = csv.reader(file) # read a line from the csv

try:

for row in reader: # row is a list of elements

names[row[0]] = row[1:] # create key/value pairs (name/emails)

except csv.Error, e:

sys.exit('file %s, line %d: %s' % (file,reader.line_num,e))

file.close()

# write the file

filename += '.pdf'

wf = file(filename,"wb")

pdfWriter.write(wf)

wf.close()

# send email to recipients

msg = MIMEMultipart()

msg['From'] = 'fyockm@gmail.com'

msg['To'] = ', '.join(names[prevName])

msg['Subject'] = 'PDF from Vendor'

# message text

msg.attach(MIMEText("See attachment!"))

# read the recent file and add as email attachment

fp = open(filename,'rb')

attach = MIMEApplication(fp.read(),'pdf')

attach.add_header('Content-Disposition','attachment',filename=filename)

msg.attach(attach)

fp.close()

# Send the email

pdfReader = PdfFileReader(file(pdfFile,"rb"))

# read the names and emails from csv file

names = get_names(nameFile)

# create an instance in SMTP server

smtp = smtplib.SMTP('localhost')

# loop through the pages of the pdf

# when a name is found, write pages to a new pdf until next name is found

# then write the file and email as attachment

i = 0

prevName = ""

while i<pdfReader.getNumPages():

page = pdfReader.getPage(i)

pageStr = page.extractText() # extract the pdf text

for name in names.keys():

if pageStr.lower().find(name.lower())!=-1:

if 'pdfWriter' in locals(): # send the current pdf

send_email(smtp,pdfWriter,prevName,names)

pdfWriter = PdfFileWriter() # create new pdfWriter file and add current page

prevName = name # save off previous name

break

if 'pdfWriter' in locals():

pdfWriter.addPage(page)

i+=1

# send the last file

if 'pdfWriter' in locals():

send_email(smtp,pdfWriter,prevName,names)

# quit the smtp server

if len(sys.argv) != 3:

print 'usage: ./kjlc_pdf_split.py pdfFile nameCsv'

sys.exit(1)

pdfFile = sys.argv[1]

nameCsv = sys.argv[2]