feat: update email send using sendgrid

.gitignore

@@ -1,4 +1,5 @@
 .env
 PLAN.md
 PLAN_UPDATE.md
-__pycache__/
+__pycache__/
+test.py

modules/email_sender.py

@@ -1,5 +1,4 @@
 import os
-import smtplib
 import tempfile
 from email.mime.multipart import MIMEMultipart
 from email.mime.text import MIMEText
@@ -9,6 +8,12 @@ from markdown_it import MarkdownIt
 from datetime import datetime
 import weasyprint
 
+import base64
+from sendgrid import SendGridAPIClient
+from sendgrid.helpers.mail import (
+    Mail, Attachment, FileContent, FileName,
+    FileType, Disposition)
+
 # Tải biến môi trường
 load_dotenv()
 
@@ -184,43 +189,54 @@ def _generate_pdf_from_markdown(markdown_string):
             os.unlink(temp_html_path)
         raise e
 
-def send_report_via_email(report_markdown, recipient_email):
-    """Send market report via email"""
+def send_report_via_email(report_markdown: str, recipient_email: str):
+    """
+    Tạo báo cáo PDF từ markdown và gửi qua email bằng API của SendGrid.
+    """
+    sendgrid_api_key = os.getenv("SENDGRID_API_KEY")
+    sender_email = os.getenv("SENDER_EMAIL")
+
+    if not sendgrid_api_key or not sender_email:
+        raise ValueError("Thiếu thông tin SENDGRID_API_KEY hoặc SENDER_EMAIL trong biến môi trường.")
+
+    # Tạo đối tượng Mail của SendGrid
+    message = Mail(
+        from_email=sender_email,
+        to_emails=recipient_email,
+        subject=f"Bản tin Thị trường Tài chính ngày {datetime.now().strftime('%d-%m-%Y')}",
+        html_content="<p>Xin chào,</p><p>Đây là bản tin thị trường được tạo tự động bởi AI Financial Dashboard.</p><p>Vui lòng xem chi tiết trong file PDF đính kèm.</p><p>Trân trọng,</p>"
+    )
+
     try:
-        # Generate PDF from markdown
+        # Tạo file PDF trong bộ nhớ
         pdf_data = _generate_pdf_from_markdown(report_markdown)
         
-        # Create message
-        message = MIMEMultipart()
-        message["From"] = SENDER_EMAIL
-        message["To"] = recipient_email
-        message["Subject"] = f"AI Financial Dashboard - Daily Market Report {datetime.now().strftime('%d/%m/%Y')}"
-        
-        # Add content with UTF-8 encoding
-        body = """
-        Dear User,
-        
-        Attached is today's financial market report, automatically generated by AI Financial Dashboard.
-        
-        Best regards,
-        AI Financial Dashboard Team
-        """
-        message.attach(MIMEText(body, "plain", "utf-8"))
-        
-        # Attach PDF file
-        attachment = MIMEApplication(pdf_data, _subtype="pdf")
-        attachment.add_header(
-            "Content-Disposition", "attachment", 
-            filename=f"Market_Report_{datetime.now().strftime('%Y%m%d')}.pdf"
+        # Mã hóa base64 cho file đính kèm
+        encoded_file = base64.b64encode(pdf_data).decode()
+
+        # Tạo đối tượng Attachment
+        attached_file = Attachment(
+            FileContent(encoded_file),
+            FileName(f'Ban_tin_Thi_truong_{datetime.now().strftime("%Y%m%d")}.pdf'),
+            FileType('application/pdf'),
+            Disposition('attachment')
         )
-        message.attach(attachment)
-        
-        # Connect to SMTP server and send email
-        with smtplib.SMTP_SSL("smtp.gmail.com", 465) as server:
-            server.login(SENDER_EMAIL, SENDER_APP_PASSWORD)
-            server.send_message(message)
+        message.attachment = attached_file
+
+        # Gửi email qua API
+        sg = SendGridAPIClient(sendgrid_api_key)
+        response = sg.send(message)
         
-        return True, "Email sent successfully!"
-    
+        # Kiểm tra response từ SendGrid
+        if 200 <= response.status_code < 300:
+            print(f"Email đã được gửi thành công đến {recipient_email} qua SendGrid.")
+            return True, "Email đã được gửi thành công!"
+        else:
+            print(f"Lỗi khi gửi email qua SendGrid: {response.body}")
+            return False, f"Không thể gửi email. Mã lỗi: {response.status_code}"
+
     except Exception as e:
-        return False, f"Error sending email: {str(e)}"
+        import traceback
+        traceback.print_exc()
+        print(f"Lỗi khi gửi email: {e}")
+        return False, f"Đã xảy ra lỗi: {e}"

requirements.txt

@@ -19,6 +19,7 @@ decorator==5.2.1
 defusedxml==0.7.1
 distro==1.9.0
 dnspython==2.7.0
+ecdsa==0.19.1
 email_validator==2.2.0
 executing==2.2.0
 fonttools==4.59.0
@@ -85,12 +86,14 @@ pyphen==0.17.2
 PySocks==1.7.1
 python-dateutil==2.9.0.post0
 python-dotenv==1.1.1
+python-http-client==3.3.7
 pytz==2025.2
 referencing==0.36.2
 requests==2.32.4
 rpds-py==0.26.0
 rsa==4.9.1
 selenium==4.34.2
+sendgrid==6.12.4
 setuptools==78.1.1
 six==1.17.0
 smmap==5.0.2
@@ -122,6 +125,7 @@ weasyprint==66.0
 webencodings==0.5.1
 websocket-client==1.8.0
 websockets==15.0.1
+Werkzeug==3.1.3
 wheel==0.45.1
 widgetsnbextension==4.0.14
 wsproto==1.2.0

test.py

@@ -1,235 +0,0 @@
-import os
-import smtplib
-import tempfile
-from email.mime.multipart import MIMEMultipart
-from email.mime.text import MIMEText
-from email.mime.application import MIMEApplication
-from dotenv import load_dotenv
-from markdown_it import MarkdownIt
-from datetime import datetime
-import weasyprint
-
-# Tải biến môi trường
-load_dotenv()
-
-# Thông tin email từ biến môi trường
-SENDER_EMAIL = os.getenv("SENDER_EMAIL")
-SENDER_APP_PASSWORD = os.getenv("SENDER_APP_PASSWORD")
-
-def _markdown_to_html(markdown_string):
-    """Chuyển đổi Markdown thành HTML"""
-    md = MarkdownIt()
-    html_content = md.render(markdown_string)
-    
-    # Định dạng ngày hiện tại
-    current_date = datetime.now().strftime("%d/%m/%Y")
-    
-    # Tạo HTML hoàn chỉnh với CSS để định dạng đẹp
-    full_html = f"""
-    <!DOCTYPE html>
-    <html>
-    <head>
-        <meta charset="UTF-8">
-        <meta name="viewport" content="width=device-width, initial-scale=1.0">
-        <title>Bản tin Thị trường Ngày {current_date}</title>
-        <style>
-            @page {{
-                size: A4;
-                margin: 2cm;
-            }}
-            body {{
-                font-family: Arial, Helvetica, sans-serif;
-                line-height: 1.5;
-                color: #333;
-                max-width: 800px;
-                margin: 0 auto;
-            }}
-            .report-header {{
-                text-align: center;
-                margin-bottom: 30px;
-            }}
-            .report-date {{
-                font-style: italic;
-                color: #666;
-                margin-bottom: 10px;
-            }}
-            .report-title {{
-                font-size: 24pt;
-                margin-bottom: 5px;
-                color: #2c3e50;
-            }}
-            .report-subtitle {{
-                font-size: 14pt;
-                color: #7f8c8d;
-                margin-top: 0;
-            }}
-            .report-body {{
-                text-align: justify;
-            }}
-            h1, h2, h3, h4, h5, h6 {{
-                color: #2c3e50;
-                margin-top: 20px;
-            }}
-            h1 {{ font-size: 20pt; }}
-            h2 {{ font-size: 18pt; }}
-            h3 {{ font-size: 16pt; }}
-            h4 {{ font-size: 14pt; }}
-            h5 {{ font-size: 12pt; }}
-            h6 {{ font-size: 10pt; }}
-            
-            p {{
-                margin-bottom: 10px;
-            }}
-            
-            a {{
-                color: #3498db;
-                text-decoration: none;
-            }}
-            
-            a:hover {{
-                text-decoration: underline;
-            }}
-            
-            ul, ol {{
-                margin: 10px 0 10px 20px;
-            }}
-            
-            li {{
-                margin-bottom: 5px;
-            }}
-            
-            blockquote {{
-                border-left: 4px solid #eee;
-                padding-left: 10px;
-                margin-left: 0;
-                color: #777;
-            }}
-            
-            .section {{
-                margin-bottom: 30px;
-            }}
-            
-            .footer {{
-                text-align: center;
-                margin-top: 40px;
-                padding-top: 20px;
-                font-size: 12px;
-                color: #777;
-                border-top: 1px solid #eee;
-            }}
-            
-            /* Custom styling for bullet points */
-            ul {{
-                list-style-type: disc;
-            }}
-            ul ul {{
-                list-style-type: circle;
-            }}
-            ul ul ul {{
-                list-style-type: square;
-            }}
-        </style>
-    </head>
-    <body>
-        <div class="report-header">
-            <div class="report-date">Ngày: {current_date}</div>
-            <h1 class="report-title">Bản tin Thị trường</h1>
-            <h2 class="report-subtitle">AI Financial Dashboard</h2>
-        </div>
-        
-        <div class="report-body">
-            {html_content}
-        </div>
-        
-        <div class="footer">
-            Bản tin được tạo tự động bởi AI Financial Dashboard. Thông tin chỉ mang tính chất tham khảo.
-        </div>
-    </body>
-    </html>
-    """
-    
-    return full_html
-
-def _generate_pdf_from_markdown(markdown_string):
-    """Tạo PDF từ Markdown sử dụng WeasyPrint"""
-    # Chuyển đổi markdown sang HTML
-    html_content = _markdown_to_html(markdown_string)
-    
-    # Tạo file HTML tạm thời
-    with tempfile.NamedTemporaryFile(suffix='.html', delete=False) as temp_html:
-        temp_html_path = temp_html.name
-        temp_html.write(html_content.encode('utf-8'))
-    
-    # Tạo file PDF từ HTML
-    try:
-        # Tạo tên file PDF tạm thời
-        with tempfile.NamedTemporaryFile(suffix='.pdf', delete=False) as temp_pdf:
-            temp_pdf_path = temp_pdf.name
-        
-        # Tạo PDF
-        weasyprint.HTML(filename=temp_html_path).write_pdf(temp_pdf_path)
-        
-        # Đọc nội dung PDF
-        with open(temp_pdf_path, 'rb') as f:
-            pdf_data = f.read()
-        
-        # Xóa các file tạm
-        os.unlink(temp_html_path)
-        os.unlink(temp_pdf_path)
-        
-        return pdf_data
-    except Exception as e:
-        # Xử lý lỗi và đảm bảo xóa file tạm
-        if os.path.exists(temp_html_path):
-            os.unlink(temp_html_path)
-        raise e
-
-def send_report_via_email(report_markdown, recipient_email):
-    """Gửi báo cáo thị trường qua email"""
-    try:
-        # Tạo PDF từ markdown
-        pdf_data = _generate_pdf_from_markdown(report_markdown)
-        
-        # Tạo message
-        message = MIMEMultipart()
-        message["From"] = SENDER_EMAIL
-        message["To"] = recipient_email
-        message["Subject"] = f"AI Financial Dashboard - Bản tin Thị trường {datetime.now().strftime('%d/%m/%Y')}"
-        
-        # Thêm phần nội dung với encoding UTF-8
-        body = """
-        Kính gửi Quý khách,
-        
-        Đính kèm là bản tin thị trường tài chính hôm nay, được tổng hợp tự động bởi AI Financial Dashboard.
-        
-        Trân trọng,
-        AI Financial Dashboard Team
-        """
-        message.attach(MIMEText(body, "plain", "utf-8"))
-        
-        # Đính kèm file PDF
-        attachment = MIMEApplication(pdf_data, _subtype="pdf")
-        attachment.add_header(
-            "Content-Disposition", "attachment", 
-            filename=f"Market_Report_{datetime.now().strftime('%Y%m%d')}.pdf"
-        )
-        message.attach(attachment)
-        
-        # Kết nối đến server SMTP và gửi email
-        with smtplib.SMTP_SSL("smtp.gmail.com", 465) as server:
-            server.login(SENDER_EMAIL, SENDER_APP_PASSWORD)
-            server.send_message(message)
-        
-        return True, "Email đã được gửi thành công!"
-    
-    except Exception as e:
-        return False, f"Lỗi khi gửi email: {str(e)}"
-
-if __name__ == "__main__":
-    report_markdown = """
-    # Bản tin Thị trường
-    ## Ngày 29/07/2025
-    - Cổ phiếu A tăng 10%
-    - Cổ phiếu B giảm 5%
-    """
-    send_report_via_email(report_markdown, "[email protected]")