99 lines
3.1 KiB
Python
99 lines
3.1 KiB
Python
# blueprints/yzm.py
|
||
|
||
import random
|
||
import smtplib
|
||
import string
|
||
import ssl
|
||
import time
|
||
|
||
from email.mime.text import MIMEText
|
||
from flask import request, jsonify, Blueprint
|
||
from flask_cors import CORS
|
||
|
||
bp = Blueprint('yzm', __name__)
|
||
CORS(bp, supports_credentials=True)
|
||
|
||
verification_codes = {}
|
||
|
||
def generate_code():
|
||
return ''.join(random.choices(string.digits, k=6))
|
||
|
||
def send_email(receiver, code):
|
||
sender = '3492073524@qq.com'
|
||
password = 'xhemkcgrgximchcd'
|
||
|
||
msg = MIMEText(f'您的验证码是：{code}，5分钟内有效。')
|
||
msg['From'] = sender
|
||
msg['To'] = receiver
|
||
msg['Subject'] = '禾境智联后台管理系统 - 验证码'
|
||
|
||
max_retries = 3
|
||
|
||
for attempt in range(1, max_retries + 1):
|
||
try:
|
||
print(f"[尝试 {attempt}/{max_retries}] 正在连接 SMTP 服务器...")
|
||
context = ssl.create_default_context()
|
||
|
||
with smtplib.SMTP_SSL('smtp.qq.com', 465, context=context) as server:
|
||
print("✅ SMTP 连接成功")
|
||
server.login(sender, password)
|
||
print("🔑 登录成功")
|
||
server.sendmail(sender, receiver, msg.as_string())
|
||
print("📧 邮件发送成功")
|
||
return True # 成功发送
|
||
|
||
except smtplib.SMTPAuthenticationError as e:
|
||
print(f"❌ 认证失败（授权码错误）: {str(e)}")
|
||
return False
|
||
except smtplib.SMTPConnectError as e:
|
||
print(f"❌ 连接失败: {str(e)}")
|
||
except smtplib.SMTPException as e:
|
||
error_msg = str(e)
|
||
if error_msg == "(-1, b'\\x00\\x00\\x00')" or "unexpected EOF" in error_msg:
|
||
print("⚠️ 警告: 忽略非致命异常，假设邮件已发送成功")
|
||
return True # 假设邮件已成功发送
|
||
else:
|
||
print(f"❌ SMTP 异常: {error_msg}")
|
||
except Exception as e:
|
||
print(f"❌ 未知错误: {str(e)}")
|
||
|
||
if attempt < max_retries:
|
||
print("🔄 正在等待重试...")
|
||
time.sleep(2)
|
||
else:
|
||
print("💥 达到最大重试次数，邮件发送失败")
|
||
return False
|
||
|
||
@bp.route("/captcha/email", methods=["POST"])
|
||
def send_code():
|
||
data = request.json
|
||
email = data.get("email")
|
||
|
||
if not email:
|
||
return jsonify({"message": "邮箱不能为空"}), 400
|
||
|
||
code = generate_code()
|
||
verification_codes[email] = code
|
||
print(f"验证码已生成: {email} -> {code}")
|
||
|
||
if send_email(email, code):
|
||
return jsonify({"message": "验证码已发送"}), 200
|
||
else:
|
||
return jsonify({"message": "邮件发送失败"}), 500
|
||
|
||
@bp.route("/captcha/verify", methods=["POST"])
|
||
def verify_code():
|
||
data = request.json
|
||
email = data.get("email")
|
||
user_code = data.get("code")
|
||
|
||
if not email or not user_code:
|
||
return jsonify({"message": "参数错误", "valid": False}), 400
|
||
|
||
stored_code = verification_codes.get(email)
|
||
|
||
if stored_code and stored_code == user_code:
|
||
del verification_codes[email]
|
||
return jsonify({"valid": True}), 200
|
||
else:
|
||
return jsonify({"valid": False}), 400 |