Upload mailer_quant.py

mailer_quant.py

@@ -0,0 +1,63 @@
+import email, smtplib, ssl
+from datetime import datetime
+from email import encoders
+from email.mime.base import MIMEBase
+from email.mime.multipart import MIMEMultipart
+from email.mime.text import MIMEText
+import os 
+
+class Mailer():
+
+    def __init__(self, subject, body, html, receiver):
+        self.subject = subject
+        self.body = body
+        self.html = html
+        self.sender_email = os.environ.get('EMAIL_SENDER')
+        self.receiver_email = receiver
+        self.password = os.environ.get('EMAIL_PASSWORD')
+        self.message = None
+        self.create_message()
+        
+        
+    def create_message(self, filepath=None, filename=None):
+        # Create a multipart message and set headers
+        message = MIMEMultipart()
+        message["From"] = self.sender_email
+        message["To"] = self.receiver_email
+        message["Subject"] = self.subject
+        message["Bcc"] = self.receiver_email  # Recommended for mass emails
+        # Add body to email
+        if self.html:
+            message.attach(MIMEText(self.html, "html"))
+        elif self.body:
+            message.attach(MIMEText(self.body, "plain"))
+        #filename = "test.pdf"  # In same directory as script
+        # Open PDF file in binary mode
+        if filepath and filename:
+            with open(filepath, "rb") as attachment:
+                # Add file as application/octet-stream
+                # Email client can usually download this automatically as attachment
+                part = MIMEBase("application", "octet-stream")
+                part.set_payload(attachment.read())
+    
+            # Encode file in ASCII characters to send by email    
+            encoders.encode_base64(part)
+    
+            # Add header as key/value pair to attachment part
+            part.add_header(
+                "Content-Disposition",
+                f"attachment; filename= {filename}",
+            )
+    
+            # Add attachment to message and convert message to string
+            message.attach(part)
+        self.text = message.as_string()
+        return
+    
+    def send_message(self, receiver):    
+        # Log in to server using secure context and send email
+        context = ssl.create_default_context()
+        with smtplib.SMTP_SSL("smtp.gmail.com", 465, context=context) as server:
+            server.login(self.sender_email, self.password)
+            server.sendmail(self.sender_email, receiver, self.text)    
+