55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
|
"""
|
|||
|
|
邮件发送模块。
|
|||
|
|
|
|||
|
|
通过 SMTP 发送验证码邮件，配置项从 .env 读取：
|
|||
|
|
- SMTP_HOST / SMTP_PORT / SMTP_USER / SMTP_AUTH_CODE / EMAIL_FROM
|
|||
|
|
163 邮箱：smtp.163.com，端口 465（SSL），密码处填写授权码。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import smtplib
|
|||
|
|
from email.header import Header
|
|||
|
|
from email.mime.text import MIMEText
|
|||
|
|
from email.utils import formataddr
|
|||
|
|
|
|||
|
|
from dotenv import load_dotenv
|
|||
|
|
|
|||
|
|
from .database import PROJECT_ROOT
|
|||
|
|
|
|||
|
|
# 加载项目根目录下的 .env（重复调用无副作用）
|
|||
|
|
load_dotenv(PROJECT_ROOT / ".env")
|
|||
|
|
|
|||
|
|
SMTP_HOST = os.getenv("SMTP_HOST", "smtp.163.com")
|
|||
|
|
SMTP_PORT = int(os.getenv("SMTP_PORT") or "465")
|
|||
|
|
SMTP_USER = os.getenv("SMTP_USER", "")
|
|||
|
|
SMTP_AUTH_CODE = os.getenv("SMTP_AUTH_CODE", "")
|
|||
|
|
EMAIL_FROM = os.getenv("EMAIL_FROM", SMTP_USER)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def send_verification_code(to_email: str, code: str) -> None:
|
|||
|
|
"""向指定邮箱发送验证码邮件；发送失败时抛出异常由调用方处理。"""
|
|||
|
|
if not SMTP_USER or not SMTP_AUTH_CODE:
|
|||
|
|
raise RuntimeError("SMTP 未配置：请在 .env 中设置 SMTP_USER / SMTP_AUTH_CODE")
|
|||
|
|
|
|||
|
|
subject = "博客 - 邮箱验证码"
|
|||
|
|
body = (
|
|||
|
|
f"您好，\n\n"
|
|||
|
|
f"您的验证码是：{code}\n"
|
|||
|
|
f"验证码 10 分钟内有效，请勿泄露给他人。\n\n"
|
|||
|
|
f"如非本人操作，请忽略本邮件。"
|
|||
|
|
)
|
|||
|
|
message = MIMEText(body, "plain", "utf-8")
|
|||
|
|
message["Subject"] = Header(subject, "utf-8")
|
|||
|
|
message["From"] = formataddr(("MyBlog", EMAIL_FROM))
|
|||
|
|
message["To"] = to_email
|
|||
|
|
|
|||
|
|
if SMTP_PORT == 465:
|
|||
|
|
server = smtplib.SMTP_SSL(SMTP_HOST, SMTP_PORT, timeout=15)
|
|||
|
|
else:
|
|||
|
|
server = smtplib.SMTP(SMTP_HOST, SMTP_PORT, timeout=15)
|
|||
|
|
server.starttls()
|
|||
|
|
try:
|
|||
|
|
server.login(SMTP_USER, SMTP_AUTH_CODE)
|
|||
|
|
server.sendmail(EMAIL_FROM, [to_email], message.as_string())
|
|||
|
|
finally:
|
|||
|
|
server.quit()
|