81972f1328
本次提交完成了用户认证模块的注册扩展： 1. 新增阿里云短信和异步邮件发送工具类 2. 新增短信/邮箱相关配置类与环境变量支持 3. 添加短信验证码、待注册数据的Redis缓存工具方法 4. 扩展用户登录逻辑，支持用户名/邮箱/手机号多方式登录 5. 实现手机短信注册和邮箱验证注册完整流程 6. 更新权限初始化数据与系统配置项 7. 补充相关Pydantic请求响应模型 8. 新增依赖包并完善requirements.txt
72 lines
2.6 KiB
Python
72 lines
2.6 KiB
Python
from fastapi import HTTPException
|
||
from config import email_settings
|
||
from email.mime.text import MIMEText
|
||
from email.mime.multipart import MIMEMultipart
|
||
|
||
|
||
def _build_verification_email(token: str) -> str:
|
||
"""构建验证邮件HTML内容"""
|
||
verify_url = f"{email_settings.base_url.rstrip('/')}/verify-email?token={token}"
|
||
return f"""<!DOCTYPE html>
|
||
<html>
|
||
<head><meta charset="utf-8"></head>
|
||
<body style="font-family: Arial, sans-serif; padding: 20px;">
|
||
<h2>📧 邮箱验证</h2>
|
||
<p>感谢注册！请点击下方按钮完成邮箱验证：</p>
|
||
<p>
|
||
<a href="{verify_url}" style="display:inline-block;padding:12px 24px;background:#409eff;
|
||
color:#fff;text-decoration:none;border-radius:4px;font-size:16px;">
|
||
点击验证邮箱
|
||
</a>
|
||
</p>
|
||
<p>或者复制以下链接到浏览器打开：</p>
|
||
<p style="color:#909399;font-size:13px;">{verify_url}</p>
|
||
<p style="color:#909399;font-size:13px;">链接有效期为30分钟，请尽快完成验证。</p>
|
||
</body>
|
||
</html>"""
|
||
|
||
|
||
async def send_verification_email(to_email: str, token: str) -> bool:
|
||
"""发送邮箱验证链接"""
|
||
try:
|
||
import aiosmtplib
|
||
|
||
msg = MIMEMultipart("alternative")
|
||
msg["Subject"] = "邮箱验证 - 图书管理系统"
|
||
msg["From"] = email_settings.from_addr
|
||
msg["To"] = to_email
|
||
msg.attach(MIMEText(_build_verification_email(token), "html", "utf-8"))
|
||
|
||
await aiosmtplib.send(
|
||
msg,
|
||
hostname=email_settings.smtp_host,
|
||
port=email_settings.smtp_port,
|
||
username=email_settings.smtp_user,
|
||
password=email_settings.smtp_password,
|
||
use_tls=email_settings.use_tls,
|
||
)
|
||
return True
|
||
except ImportError:
|
||
# aiosmtplib 未安装时降级使用标准库 smtplib
|
||
import smtplib
|
||
|
||
msg = MIMEMultipart("alternative")
|
||
msg["Subject"] = "邮箱验证 - 图书管理系统"
|
||
msg["From"] = email_settings.from_addr
|
||
msg["To"] = to_email
|
||
msg.attach(MIMEText(_build_verification_email(token), "html", "utf-8"))
|
||
|
||
if email_settings.use_tls:
|
||
server = smtplib.SMTP_SSL(email_settings.smtp_host, email_settings.smtp_port)
|
||
else:
|
||
server = smtplib.SMTP(email_settings.smtp_host, email_settings.smtp_port)
|
||
server.login(email_settings.smtp_user, email_settings.smtp_password)
|
||
server.sendmail(email_settings.from_addr, to_email, msg.as_string())
|
||
server.quit()
|
||
return True
|
||
except Exception as e:
|
||
raise HTTPException(
|
||
status_code=500,
|
||
detail="邮件发送失败，请稍后重试",
|
||
)
|