Create email_service.py

SuperAgent/app/services/email_service.py

@@ -0,0 +1,76 @@
+import os
+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 logging
+
+logger = logging.getLogger(__name__)
+
+class EmailService:
+    def __init__(self):
+        self.smtp_host = os.getenv("SMTP_HOST")
+        self.smtp_port = os.getenv("SMTP_PORT")
+        self.smtp_user = os.getenv("SMTP_USER")
+        self.smtp_pass = os.getenv("SMTP_PASSWORD")
+        self.smtp_from = os.getenv("SMTP_FROM", "[EMAIL_ADDRESS]")
+
+    def send_report(self, recipient: str, subject: str, body_html: str, attachment_path: str = None) -> bool:
+        if not recipient:
+            logger.warning("Email recipient is empty. Skipping email delivery.")
+            return False
+
+        if not self.smtp_host or not self.smtp_user or not self.smtp_pass:
+            logger.warning("SMTP credentials are not configured in environment. Printing email to console (MOCK).")
+            print(f"\n================ MOCK EMAIL SENT ================")
+            print(f"To: {recipient}")
+            print(f"From: {self.smtp_from}")
+            print(f"Subject: {subject}")
+            print(f"Body: {body_html[:300]}...")
+            if attachment_path:
+                print(f"Attachment: {os.path.basename(attachment_path)}")
+            print(f"=================================================\n")
+            return True
+
+        try:
+            # Create message container
+            msg = MIMEMultipart()
+            msg['From'] = self.smtp_from
+            msg['To'] = recipient
+            msg['Subject'] = subject
+
+            # Record the MIME types of both parts - text/html
+            msg.attach(MIMEText(body_html, 'html'))
+
+            # Attach PDF if provided
+            if attachment_path and os.path.exists(attachment_path):
+                filename = os.path.basename(attachment_path)
+                with open(attachment_path, "rb") as attachment:
+                    part = MIMEBase("application", "octet-stream")
+                    part.set_payload(attachment.read())
+                    encoders.encode_base64(part)
+                    part.add_header(
+                        "Content-Disposition",
+                        f"attachment; filename={filename}",
+                    )
+                    msg.attach(part)
+
+            # Connect to SMTP server
+            port = int(self.smtp_port) if self.smtp_port else 587
+            server = smtplib.SMTP(self.smtp_host, port)
+            server.starttls()
+            server.login(self.smtp_user, self.smtp_pass)
+            
+            # Send email
+            server.sendmail(self.smtp_from, recipient, msg.as_string())
+            server.quit()
+            
+            logger.info(f"Email sent successfully to {recipient}")
+            return True
+
+        except Exception as e:
+            logger.error(f"Failed to send email to {recipient}: {str(e)}")
+            return False
+
+email_service = EmailService()