v2.0

.gitignore

@@ -101,6 +101,7 @@ wittymail_ve/
 
 # VS Code
 .vscode/
+.idea/
 
 # WittyMail log file
-wittymail_log.txt
+wittymail_log.txt

backend/EmailProviderTest.py

@@ -0,0 +1,28 @@
+from mailer.Email import Email
+from mailer.EmailProvider import EmailProvider
+import sys
+
+'''
+Basic Test1
+-----------
+1. Get username and password from CLI
+2. Login using SMTP
+3. Send a basic email
+4. Logout
+'''
+def basic_test1(username, password):
+    e = Email([username], [username], 'Subject', 'Body')
+    eP = EmailProvider()
+    selected_eP = eP.choose_email_provider("SMTP")
+    seP = selected_eP(username, 'smtp.gmail.com', 587, username, password)
+    seP.login()
+    seP.send_email(e)
+    seP.logout()
+
+if __name__ == '__main__':
+    if len(sys.argv) != 3:
+        print("EmailProviderTest.py <username> <password>")
+
+    username = sys.argv[1]
+    password = sys.argv[2]
+    basic_test1(username, password)

backend/mailer/Email.py

@@ -0,0 +1,41 @@
+#!/usr/bin/env python
+# coding=utf-8
+
+import smtplib
+from email.mime.multipart import MIMEMultipart
+from email.mime.text import MIMEText 
+from email.mime.base import MIMEBase 
+from email import encoders 
+import os
+import util.logger as logger
+
+log = logger.get_logger(__name__)
+
+class Email():
+  subject_template = None
+  body_template = None
+  frm = None
+  common_attachment_dir = None
+
+  def __init__(self, tos, ccs, attachments, subject, body, status = None):
+    assert tos is not None, log.error('Incorrect value provided to=%s cc=%s' % (tos, ccs))
+    self.tos = tos
+    self.ccs = ccs
+    self.attachments = attachments
+    self.subject = subject
+    self.body = body
+    self.status = status
+
+  def get_email_from(self):
+    return Email.frm
+
+  def set_email_frm(self, frm):
+    Email.frm = frm
+
+  def set_email_common_attachment_dir(self, attachment_dir):
+    Email.common_attachment_dir = attachment_dir
+
+  def __del__(self):
+    self.to = None
+    Email.subject_template = None
+    Email.body_template = None

backend/mailer/EmailProvider.py

@@ -0,0 +1,19 @@
+import importlib
+import pdb
+#from mailer.SMTPEmailProvider import SMTPEmailProvider
+
+class EmailProvider:
+    def __init__(self):
+        pass
+
+    def choose_email_provider(self, name):
+        from mailer.SMTPEmailProvider import SMTPEmailProvider
+        return SMTPEmailProvider
+
+    def login(self):
+        raise NotImplementedError()
+    def send_email(self):
+        raise NotImplementedError()
+    def logout(self):
+        raise NotImplementedError()
+

backend/mailer/SMTPEmailProvider.py

@@ -0,0 +1,107 @@
+#!/usr/bin/env python
+# coding=utf-8
+
+import smtplib
+from email.mime.multipart import MIMEMultipart
+from email.mime.text import MIMEText 
+from email.mime.base import MIMEBase 
+from email import encoders 
+import os
+import util.logger as logger
+from mailer.EmailProvider import EmailProvider
+from os import listdir 
+from os.path import isfile
+import pdb
+
+log = logger.get_logger(__name__)
+
+class SMTPEmailProvider(EmailProvider):
+    def __init__(self, server, port, username, password):
+        EmailProvider.__init__(self)
+        assert server is not None and port is not None and username is not None and password is not None,\
+            log.error('server, port number, username or password is None')
+
+        self.server = server
+        self.port = port
+        self.username = username
+        self.password = password
+        self.s = None
+
+    def logout(self):
+        self.s.quit() 
+        log.debug('emailapi teardown successful')
+
+    # TODO Can you check if SMTPlib has different types of Exception (*Error) classes and write an except block for each?
+    def login(self):
+        # Connect to the SMTP server and login
+        try:
+            self.s = smtplib.SMTP(self.server, self.port)
+            self.s.ehlo()
+            self.s.starttls() 
+            self.s.login(self.username, self.password) 
+
+        except:
+            log.exception('Incorrect server/user information provided')
+            return [-1, 'Incorrect server/user information provided']
+
+        log.debug('Login successful')
+        return [0, 'Login successful']
+
+    def send_email(self, e):
+        try:
+            # Extra <p> tags show up as multiple line breaks in the resulting email, so replace them
+            # with single line breaks
+            body = e.body.replace('</p><p>', '<br />')
+
+            log.debug("Cleaned body: %s", body)
+
+            msg = MIMEMultipart() 
+            msg['From'] = e.frm
+            msg['To'] = ", ".join(e.tos)
+
+            ccs_to_send = []
+            if e.ccs is not None and e.ccs[0] != 'None':
+                msg['Cc'] = ", ".join(e.ccs)
+                for cc in e.ccs:
+                    ccs_to_send.append(cc.strip())
+            else:
+                log.debug('No cc mailer provided')
+
+            msg['Subject'] = e.subject
+            msg.attach(MIMEText(e.body, 'html'))  
+
+            common_attachments = listdir(e.common_attachment_dir)
+            common_attachments = [os.path.join(e.common_attachment_dir, ca) for ca in common_attachments]
+
+            attachments_dir = os.path.dirname(e.common_attachment_dir)
+            attachments = [os.path.join(attachments_dir, a) for a in e.attachments]
+
+            attachments = attachments + common_attachments
+
+            if (attachments is not None):
+                for attachment in attachments:
+                    if not os.path.isfile(attachment):
+                        return [-1, 'Attachment not found'] 
+                    try:
+                        a = open(attachment, "rb")
+                    except:
+                        log.exception("Failed to open attachment: %s", attachment)
+                        raise
+                    p = MIMEBase('application', 'octet-stream') 
+                    p.set_payload((a).read()) 
+                    encoders.encode_base64(p) 
+                    p.add_header('Content-Disposition', "attachment; filename= %s" % os.path.basename(attachment)) 
+                    msg.attach(p)
+
+            msg_str = msg.as_string()
+            log.debug("ccs_to_send: %s", ccs_to_send)
+            if ccs_to_send:
+                self.s.sendmail(msg['From'], e.tos + ccs_to_send, msg_str) 
+            else:
+                self.s.sendmail(msg['From'], e.tos, msg_str) 
+                log.info('Email sent by %s to %s' % (msg['From'], msg['To']))
+            return [0, 'Email sent successfully']
+        except:
+            self.login()
+            log.exception("Failed to send email")
+            return [-1, 'Error']