166 lines
5.3 KiB
Python
166 lines
5.3 KiB
Python
"""
|
||
邮件验证服务
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from email.mime.multipart import MIMEMultipart
|
||
from email.mime.text import MIMEText
|
||
import secrets
|
||
import smtplib
|
||
import string
|
||
|
||
from app.config import settings
|
||
from app.log import logger
|
||
|
||
|
||
class EmailService:
|
||
"""邮件发送服务"""
|
||
|
||
def __init__(self):
|
||
self.smtp_server = getattr(settings, "smtp_server", "localhost")
|
||
self.smtp_port = getattr(settings, "smtp_port", 587)
|
||
self.smtp_username = getattr(settings, "smtp_username", "")
|
||
self.smtp_password = getattr(settings, "smtp_password", "")
|
||
self.from_email = getattr(settings, "from_email", "noreply@example.com")
|
||
self.from_name = getattr(settings, "from_name", "osu! server")
|
||
|
||
def generate_verification_code(self) -> str:
|
||
"""生成8位验证码"""
|
||
# 只使用数字，避免混淆
|
||
return "".join(secrets.choice(string.digits) for _ in range(8))
|
||
|
||
async def send_verification_email(self, email: str, code: str, username: str) -> bool:
|
||
"""发送验证邮件"""
|
||
try:
|
||
msg = MIMEMultipart()
|
||
msg["From"] = f"{self.from_name} <{self.from_email}>"
|
||
msg["To"] = email
|
||
msg["Subject"] = "邮箱验证 - Email Verification"
|
||
|
||
# HTML 邮件内容
|
||
html_content = f"""
|
||
<!DOCTYPE html>
|
||
<html>
|
||
<head>
|
||
<meta charset="utf-8">
|
||
<style>
|
||
.container {{
|
||
max-width: 600px;
|
||
margin: 0 auto;
|
||
font-family: Arial, sans-serif;
|
||
line-height: 1.6;
|
||
}}
|
||
.header {{
|
||
background: linear-gradient(135deg, #ff66aa, #ff9966);
|
||
color: white;
|
||
padding: 20px;
|
||
text-align: center;
|
||
border-radius: 10px 10px 0 0;
|
||
}}
|
||
.content {{
|
||
background: #f9f9f9;
|
||
padding: 30px;
|
||
border: 1px solid #ddd;
|
||
}}
|
||
.code {{
|
||
background: #fff;
|
||
border: 2px solid #ff66aa;
|
||
border-radius: 8px;
|
||
padding: 15px;
|
||
text-align: center;
|
||
font-size: 24px;
|
||
font-weight: bold;
|
||
letter-spacing: 3px;
|
||
margin: 20px 0;
|
||
color: #333;
|
||
}}
|
||
.footer {{
|
||
background: #333;
|
||
color: #fff;
|
||
padding: 15px;
|
||
text-align: center;
|
||
border-radius: 0 0 10px 10px;
|
||
font-size: 12px;
|
||
}}
|
||
.warning {{
|
||
background: #fff3cd;
|
||
border: 1px solid #ffeaa7;
|
||
border-radius: 5px;
|
||
padding: 10px;
|
||
margin: 15px 0;
|
||
color: #856404;
|
||
}}
|
||
</style>
|
||
</head>
|
||
<body>
|
||
<div class="container">
|
||
<div class="header">
|
||
<h1>osu! 邮箱验证</h1>
|
||
<p>Email Verification</p>
|
||
</div>
|
||
|
||
<div class="content">
|
||
<h2>你好 {username}！</h2>
|
||
<p>感谢你注册我们的 osu! 服务器。为了完成账户验证，请输入以下验证码：</p>
|
||
|
||
<div class="code">{code}</div>
|
||
|
||
<p>这个验证码将在 <strong>10 分钟后过期</strong>。</p>
|
||
|
||
<div class="warning">
|
||
<strong>注意：</strong>
|
||
<ul>
|
||
<li>请不要与任何人分享这个验证码</li>
|
||
<li>如果你没有请求此验证码，请忽略这封邮件</li>
|
||
<li>验证码只能使用一次</li>
|
||
</ul>
|
||
</div>
|
||
|
||
<p>如果你有任何问题，请联系我们的支持团队。</p>
|
||
|
||
<hr style="border: none; border-top: 1px solid #ddd; margin: 20px 0;">
|
||
|
||
<h3>Hello {username}!</h3>
|
||
<p>Thank you for registering on our osu! server. To complete your account verification, please enter the following verification code:</p>
|
||
|
||
<p>This verification code will expire in <strong>10 minutes</strong>.</p>
|
||
|
||
<p><strong>Important:</strong> Do not share this verification code with anyone. If you did not request this code, please ignore this email.</p>
|
||
</div>
|
||
|
||
<div class="footer">
|
||
<p>© 2025 g0v0! Private Server. 此邮件由系统自动发送，请勿回复。</p>
|
||
<p>This email was sent automatically, please do not reply.</p>
|
||
</div>
|
||
</div>
|
||
</body>
|
||
</html>
|
||
""" # noqa: E501
|
||
|
||
msg.attach(MIMEText(html_content, "html", "utf-8"))
|
||
|
||
# 发送邮件
|
||
if not settings.enable_email_sending:
|
||
# 邮件发送功能禁用时只记录日志，不实际发送
|
||
logger.info(f"[Email Verification] Mock sending verification code to {email}: {code}")
|
||
return True
|
||
|
||
with smtplib.SMTP(self.smtp_server, self.smtp_port) as server:
|
||
if self.smtp_username and self.smtp_password:
|
||
server.starttls()
|
||
server.login(self.smtp_username, self.smtp_password)
|
||
|
||
server.send_message(msg)
|
||
|
||
logger.info(f"[Email Verification] Successfully sent verification code to {email}")
|
||
return True
|
||
|
||
except Exception as e:
|
||
logger.error(f"[Email Verification] Failed to send email: {e}")
|
||
return False
|
||
|
||
|
||
# 全局邮件服务实例
|
||
email_service = EmailService()
|