Subversion Repositories SmartDukaan

from email.mime.text import MIMEText

from shop2020.clients.HelperClient import HelperClient

from shop2020.thriftpy.utils.ttypes import Mail

from shop2020.utils.daemon import Daemon

import sched

import time

import smtplib

import sys

import optparse

class EmailSender(Daemon):

    def __init__(self, logfile='/var/log/utils/emailsender/emailsender.log', pidfile='/var/run/email-sender.pid'):

        Daemon.__init__(self, pidfile, stdout=logfile, stderr=logfile)

        print 'EmailSender initiated at', time.time()

        self.email_schedular = sched.scheduler(time.time, time.sleep)

        self.time_to_sleep = 2

        self.authenticate()

    def authenticate(self):

        try:

            print "Attempting authentication at", time.time()

            self.mailServer = smtplib.SMTP("smtp.sendgrid.net", 587)

            self.sender = "shop2020mail"

            self.password = "shop2020"

            self.mailServer.ehlo()

            self.mailServer.starttls()

            self.mailServer.ehlo()

            self.mailServer.login(self.sender, self.password)

            print "Authentication successful", time.time()

        except smtplib.SMTPConnectError as e:

            print 'Specified host did not respond correctly', e        

        except smtplib.SMTPHeloError as e:

            print 'The server did not reply properly to the HELO greeting.', e

        except smtplib.SMTPAuthenticationError as e:

            print 'The server did not accept the username/password combination.', e

        except smtplib.SMTPException as e:

            print e

        except Exception as e:

            print sys.exc_info()[0]

            print e

    def sendMail(self, mail):

        if mail.sender:

            mail.data = "This mail is sent by " + mail.sender + "\n" + mail.data

        msg = MIMEText(mail.data, 'html')

        msg['To'] = ';'.join(mail.to)

        if mail.sender:

            msg['From'] = mail.sender

        else:

            msg['From'] = "help@saholic.com"

        msg['Subject'] = mail.subject

        try:

            rs = self.mailServer.sendmail(self.password, mail.to, msg.as_string())

        except smtplib.SMTPServerDisconnected as e:

            print 'Server Disconnected at', time.time()

            self.authenticate()

            rs = self.mailServer.sendmail(self.password, mail.to, msg.as_string())

        print msg['To']

        # Should be mailServer.quit(), but that crashes...

        return rs

    def send(self):

        print "Despatch stared at ", time.time()

        helper_client = HelperClient().get_client()

        emails_to_be_dispatched = helper_client.getEmailsToBeSent()

        print len(emails_to_be_dispatched)

        for email in emails_to_be_dispatched:

            toList = []

            for to in email.emailTo.split(';') :

                toList.append(to.strip())

            mail = Mail( toList, email.subject, email.body, None, None, None)

            try:

                send_result = self.sendMail(mail)

            except Exception as e:

                print sys.exc_info()[0]

                print e

                print 'Error occurred sending emails. Will retry.'

                continue

            if len(send_result) == 0:

                print "Sent at", time.time()

                helper_client.markEmailAsSent(email.id)

            else:

                print "Failed sending email. Id:", " Time:", time.time()

        self.email_schedular.enter(self.time_to_sleep, 1, self.send, ())

    def run(self):

        self.email_schedular.enter(self.time_to_sleep, 1, self.send, ())

        self.email_schedular.run()

if __name__ == "__main__":

    parser = optparse.OptionParser()

    parser.add_option("-l", "--logfile", dest="logfile",

                      type="string",

                      help="Log all output to LOG_FILE",

                      )

    parser.add_option("-i", "--pidfile", dest="pidfile",

                      type="string",

                      help="Write the PID to pidfile")

    (options, args) = parser.parse_args()

    daemon = EmailSender(options.logfile, options.pidfile)

    if len(args) == 0:

        daemon.run()

    elif len(args) == 1:

        if 'start' == args[0]:

            daemon.start()

        elif 'stop' == args[0]:

            daemon.stop()

        elif 'restart' == args[0]:

            daemon.restart()

        else:

            print "Unknown command"

            sys.exit(2)

        sys.exit(0)

    else:

        print "usage: %s start|stop|restart" % sys.argv[0]

        sys.exit(2)