113 lines
3.6 KiB
Python
113 lines
3.6 KiB
Python
"""邮件发送模块"""
|
|
import os
|
|
import smtplib
|
|
from email.mime.multipart import MIMEMultipart
|
|
from email.mime.text import MIMEText
|
|
from email.mime.image import MIMEImage
|
|
from email.header import Header
|
|
from typing import Optional, List
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
|
|
class EmailSender:
|
|
"""邮件发送"""
|
|
|
|
def __init__(self):
|
|
self.smtp_host = os.getenv("SMTP_HOST", "")
|
|
self.smtp_port = int(os.getenv("SMTP_PORT", "587"))
|
|
self.smtp_user = os.getenv("SMTP_USER", "")
|
|
self.smtp_password = os.getenv("SMTP_PASSWORD", "")
|
|
self.sender_name = os.getenv("SENDER_NAME", "修图客服")
|
|
|
|
def send(
|
|
self,
|
|
to_email: str,
|
|
subject: str,
|
|
body: str,
|
|
images: Optional[List[str]] = None
|
|
) -> dict:
|
|
"""
|
|
发送邮件
|
|
|
|
Args:
|
|
to_email: 收件人邮箱
|
|
subject: 邮件主题
|
|
body: 邮件正文
|
|
images: 图片路径列表
|
|
|
|
Returns:
|
|
{"success": bool, "message": str}
|
|
"""
|
|
if not self.smtp_host or not self.smtp_user:
|
|
return {"success": False, "message": "未配置邮件SMTP"}
|
|
|
|
try:
|
|
# 创建邮件
|
|
msg = MIMEMultipart('related')
|
|
msg['From'] = f"{Header(self.sender_name, 'utf-8').encode()} <{self.smtp_user}>"
|
|
msg['To'] = to_email
|
|
msg['Subject'] = subject
|
|
|
|
# 添加正文
|
|
msg.attach(MIMEText(body, 'html', 'utf-8'))
|
|
|
|
# 添加图片
|
|
if images:
|
|
for idx, img_path in enumerate(images):
|
|
if os.path.exists(img_path):
|
|
with open(img_path, 'rb') as f:
|
|
img = MIMEImage(f.read())
|
|
img.add_header('Content-ID', f'<image{idx}>')
|
|
msg.attach(img)
|
|
|
|
# 发送邮件（失败时重试 1 次）
|
|
import time
|
|
last_err = None
|
|
for attempt in range(2):
|
|
try:
|
|
server = smtplib.SMTP(self.smtp_host, self.smtp_port)
|
|
server.starttls()
|
|
server.login(self.smtp_user, self.smtp_password)
|
|
server.sendmail(self.smtp_user, to_email, msg.as_string())
|
|
server.quit()
|
|
return {"success": True, "message": "发送成功"}
|
|
except Exception as e:
|
|
last_err = e
|
|
if attempt == 0:
|
|
time.sleep(2)
|
|
return {"success": False, "message": f"发送失败: {str(last_err)}"}
|
|
except Exception as e:
|
|
return {"success": False, "message": f"发送失败: {str(e)}"}
|
|
|
|
def send_completed_work(
|
|
self,
|
|
to_email: str,
|
|
customer_name: str,
|
|
image_description: str,
|
|
result_images: List[str]
|
|
) -> dict:
|
|
"""发送完成的作品"""
|
|
subject = f"您的修图作品已完成 - {image_description}"
|
|
|
|
body = f"""
|
|
<html>
|
|
<body>
|
|
<h2>您好 {customer_name}，您的修图作品已完成！</h2>
|
|
<p>感谢您选择我们的服务。以下是您处理后的图片：</p>
|
|
<p><b>处理内容：</b> {image_description}</p>
|
|
<br>
|
|
<p>如有任何问题，请随时联系我们。</p>
|
|
<br>
|
|
<p>祝您生活愉快！</p>
|
|
</body>
|
|
</html>
|
|
"""
|
|
|
|
return self.send(to_email, subject, body, result_images)
|
|
|
|
|
|
# 全局实例
|
|
email_sender = EmailSender()
|