utilEmail.py

#!/usr/bin/env python3 

import os
import sys 
import time 
import configparser 
import smtplib
import subprocess


class configClass:
    """
    a class for config, currently gives a default account, please don't change the password 
    """ 
    def __init__(self):
        self.receiver       =   None 
        self.sender         =   "tasksubmitter0817@gmail.com"
        self.sender_pass    =   "IamTaskSubmitter"
        self.sender_name    =   "taskSubmitter"

        self.telegram_api_address = "https://integram.org/caLeNAvl6fq"

        # currently only support gmail 
        self.email_service  = "gmail"

        self.load_config()


    def load_config(self):
        """
        load config from ~/.taskSubmitter_config or /.taskSubmitter_config, 
        user level config is preferred
        """ 

        # gmail may not allow this by default, see https://support.google.com/accounts/answer/6010255?hl=en

        if os.path.exists(os.path.expanduser("~/.taskSubmitter_config")):
            self.config_loc = os.path.expanduser("~/.taskSubmitter_config")
        elif os.path.exists("/.taskSubmitter_config"):
            self.config_loc = "/.taskSubmitter_config"
        else:
            err_msg = '''
            you didn't provide config file, please copy and edit the following lines, 
            then save it to $HOME/.taskSubmitter_config or /.taskSubmitter_config, 
            remember to change the permission of the file to 600, so others won't see your password 
            optionally you can use an email dedicatd to sending email 
            
            [info]
            receiver=PUT_THE_EMAIL_ADDRESS_YOU_WANT_TO_RECEIVE_EMAIL_NOTIFICATION 
            sender=PUT_YOUR_EMAIL_ADDRESS_HERE(it needs to support smtp, currently only support gmail)
            sender_pass=YOUR PASSWORD 
            sender_name=OPTIONAL
            '''

            raise RuntimeError(err_msg) 


        self.cparser = configparser.ConfigParser() 
        self.cparser.read(self.config_loc)
        #assert "receiver" in self.cparser["info"], "Please provide reciever email address in config" 
        if "receiver" in self.cparser["info"] and len(self.cparser["info"]["receiver"]): 
            self.receiver = self.cparser["info"]["receiver"]
        if "sender" in self.cparser["info"] and len(self.cparser["info"]["sender"]): 
            self.sender = self.cparser["info"]["sender"]
            self.sender_pass = self.cparser["info"]["sender_pass"]
        if "sender_name" in self.cparser["info"] and len(self.cparser["info"]["sender_name"]): 
            self.sender_name = self.cparser["info"]["sender_name"]
        if "receiver" in self.cparser["info"] and len(self.cparser["info"]["receiver"]): 
            self.receiver = self.cparser["info"]["receiver"]

        if "email_service" in self.cparser["info"] and len(self.cparser["info"]["email_service"]): 
            self.email_service = self.cparser["info"]["email_service"]
        if "telegram_api_address" in self.cparser["info"] and len(self.cparser["info"]["telegram_api_address"]): 
            self.telegram_api_address = self.cparser["info"]["telegram_api_address"]



    def __str__(self):
        return "sender_addr: {}\npass: {}\nname: {}\nreceiver: {}".format(self.sender, 
            self.sender_pass, self.sender_name, self.receiver)

    def __repr__(self):
        return self.__str__()

class emailConst:
    """ some constants and templates used in the module 
    """

    smtp_server_gmail = "smtp.gmail.com"
    smtp_port_gmail = 587 

    message_template = "From: From {sender_name} <{sender}>\r\n"\
        "To: To {receiver} <{receiver}>\r\n"\
        "MIME-Version: 1.0\r\n"\
        "Content-type: text/html\r\n"\
        "Subject: {topic}\r\n\r\n{message}"



class emailClient: 
    def __init__(self, smtp_server, 
                    login_username, login_password, sender_name=None,
                    smtp_port=25, use_SSL=False, use_TLS=False):
        """
        initialize a class for sending emails 
        """

        self.smtp_server = smtp_server 
        self.smtp_port = smtp_port 
        self.login_username = login_username
        self.login_password = login_password 
        if sender_name is None: 
            self.sender_name = self.login_username 
        else:
            self.sender_name = sender_name 

        self.use_SSL = use_SSL  
        self.use_TLS = use_TLS 
        
        if use_SSL: 
            self.email_server = smtplib.SMTP_SSL(smtp_server, smtp_port)
        else: 
            self.email_server = smtplib.SMTP(smtp_server, smtp_port) 

        if use_TLS: 
            self.email_server.starttls() 


        try: 
            # server.ehlo()
            # server.starttls()   
            self.email_server.login(login_username, login_password)
        except Exception as e: 
            print("failed to initialize email client: {}".format(e), file=sys.stderr) 
            exit(1)



    def send_email(self, message, receiver=None, topic="No topic"):

        if self.config.email_service == "gmail":
            if receiver is None:
                assert self.receiver is not None, "please provide receiver email" 
            receiver = self.receiver
            try: 
                self.email_server.sendmail(self.login_username, receiver, 
                                emailConst.message_template.format(sender=self.login_username
                                                                , sender_name=self.sender_name
                                                                , receiver=receiver
                                                                , topic=topic
                                                                , message=message))         

            except Exception as e: 
                print("ERROR: failed to send email {}".format(e), file=sys.stderr)
                exit(1)
        if self.config.email_service == "telegram":
            try: 
                send_command = 'curl -d \'{"text":"' + '*&#9886;' + topic + ':&#9887;*<br/>'  + message + '"}\' -H "Content-Type: application/json" -X POST ' + self.telegram_api_address 
                subprocess.call(send_command, shell=True)
                print(send_command)
            except Exception as e: 
                print("ERROR: failed to send telegram {}".format(e), file=sys.stderr)
                exit(1)
    def close(self):
        self.email_server.quit() 

    def __exit__(self):
        self.close() 


class gmailClient(emailClient):
    def __init__(self, login_username, login_password, sender_name=None):
        super(gmailClient, self).__init__(emailConst.smtp_server_gmail, 
                                            login_username, 
                                            login_password, 
                                            sender_name, 
                                            smtp_port=emailConst.smtp_port_gmail, 
                                            use_SSL=False, use_TLS=True)


class defaultEmailClient(emailClient):
    def __init__(self):
        super(defaultEmailClient, self).__init__(emailConst.smtp_server_gmail, 
                                                "tasksubmitter0817@gmail.com", 
                                                "IamTaskSubmitter", 
                                                sender_name="email_sender", 
                                                smtp_port=emailConst.smtp_port_gmail, 
                                                use_SSL=False, use_TLS=True)


class configEmailClient(emailClient):
    def __init__(self):
        self.config = configClass()
        print(self.config)
        #assert self.config.email_service == "gmail", "only gmail is supported for now" 

        if self.config.email_service == "gmail":
            super(configEmailClient, self).__init__(emailConst.smtp_server_gmail, 
                                                    self.config.sender, 
                                                    self.config.sender_pass, 
                                                    sender_name=self.config.sender_name,
                                                    smtp_port=emailConst.smtp_port_gmail, 
                                                    use_SSL=False, use_TLS=True)
            self.receiver = self.config.receiver

        if self.config.email_service == "telegram":
            self.telegram_api_address = self.config.telegram_api_address
        



if __name__ == "__main__": 
    client = defaultEmailClient()
    client.send_email("peter.waynechina@gmail.com", sys.argv[1]) 
    client.close()