EmailNotification.py

import smtplib
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
from string import Template
from email.mime.base import MIMEBase
from email import encoders
import datetime
import os.path as op
import os
import sys


'''
Author: @blackram
Contributer: R@jesh
A simple python script to send email notifications for multiple receipeints based on the
SMPT lib available in python. 

***requirements***
need to install email package for this script
'''


# function for reading names and emails from text file
def get_contacts(file_name):
    names = []
    emails = []
    with open(file_name, mode='r') as contacts_file:
        for a_contact in contacts_file:
            names.append(a_contact.split()[0])
            emails.append(a_contact.split()[1])
    return names, emails


# function for reading a message template
def get_message_template(file_name):
    with open(file_name, mode='r') as template_file:
        template_file_content = template_file.read()
    return Template(template_file_content)


# function to get my account details to send mail
def get_my_account_info(file_name):
    with open(file_name, mode='r') as my_details:
        for my_det in my_details:
            my_address = my_det.split()[0]
            my_password = my_det.split()[1]
    return my_address, my_password


# main function
def main():

    my_address, my_password = get_my_account_info('C:/Users/Rajesh.Gundupalli/Desktop/PythonScripts/resources/from_mail.txt')  # get my account details
    #print(my_address)
    #print(my_password)
    names, emails = get_contacts('C:/Users/Rajesh.Gundupalli/Desktop/PythonScripts/resources/test_contacts.txt')  # read contacts
    # print(emails)
    # return
    message_template = get_message_template('C:/Users/Rajesh.Gundupalli/Desktop/PythonScripts/resources/email_template.txt')  # read template

    # using gmail smtp
    server = smtplib.SMTP(host='smtp.gmail.com', port='587')
    server.starttls()

    # Log into server
    server.login(my_address, my_password)
    failed_rcpts = {}
    for name, email in zip(names, emails):
        msg = MIMEMultipart()
        message = message_template.substitute(PERSON_NAME=name.title())
        print(message)

        msg['From'] = my_address
        msg['To'] = email
        msg['Subject'] = "Email Notification Test "+str(datetime.date.today())
        msg.attach(MIMEText(message, 'plain'))

        part = MIMEBase('application', "octet-stream")

        # adding attachement to the mail message
        #Todo - Take Command Line Argument to send files - may be path to files
        path = 'C:/Users/Rajesh.Gundupalli/Desktop/PythonScripts/resources/EMailAttachments/' 			#ToDo
        files = os.listdir(path)    #f.name for f in os.scandir(path) if f.is_file() #os.listdir(path)
        '''for fname in os.scandir(path):
			if(fname.is_file()):
				print fname'''

        #Attaching all the files in the directory
        for name in files:
            print(name)
            filePath = path + name
            attachment = open(filePath, "rb")
            part = MIMEBase('application', 'octet-stream')
            # To change the payload into encoded form 
            part.set_payload((attachment).read())
            # encode into base64 
            encoders.encode_base64(part) 
            part.add_header('Content-Disposition', "attachment; filename= %s" % name)
            msg.attach(part)
            attachment.close()
        #send the mail
        failed_rcpts = server.sendmail(my_address, email, msg.as_string().encode())
        del msg
        # failed receipt will be printed here
        print(failed_rcpts)

    server.quit()  # stopping SMTP server


if __name__ == '__main__':
    main()