vf/vf_react
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update pay ui

Browse Source
This commit is contained in:
zzlgreat
2025-12-12 07:23:46 +08:00
parent 036aef1171
commit 002c3beeac
1 changed files with 73 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

111
app.py
View File

@@ -192,17 +192,61 @@ es_client = Elasticsearch(
)
app = Flask(__name__)
# 存储验证码的临时字典（生产环境应使用Redis）
verification_codes = {}
# ============ 微信登录 Session 管理（Redis 存储，支持多进程） ============
# Redis 连接配置（支持环境变量覆盖，与 Flask-Session 共享配置）
# ============ Redis 连接配置（支持环境变量覆盖） ============
_REDIS_HOST = os.environ.get('REDIS_HOST', 'localhost')
_REDIS_PORT = int(os.environ.get('REDIS_PORT', 6379))
redis_client = redis.Redis(host=_REDIS_HOST, port=_REDIS_PORT, db=0, decode_responses=True)
print(f"📦 Redis 配置: {_REDIS_HOST}:{_REDIS_PORT}/db=0")
# ============ 验证码 Redis 存储（支持多进程/多 Worker） ============
VERIFICATION_CODE_PREFIX = "vf_code:"
VERIFICATION_CODE_EXPIRE = 300 # 验证码过期时间（5分钟）
def set_verification_code(key, code, expires_in=VERIFICATION_CODE_EXPIRE):
"""存储验证码到 Redis"""
try:
data = {
'code': code,
'expires': time.time() + expires_in
}
redis_client.setex(
f"{VERIFICATION_CODE_PREFIX}{key}",
expires_in,
json.dumps(data)
)
return True
except Exception as e:
print(f"❌ Redis 存储验证码失败: {e}")
return False
def get_verification_code(key):
"""从 Redis 获取验证码"""
try:
data = redis_client.get(f"{VERIFICATION_CODE_PREFIX}{key}")
if data:
return json.loads(data)
return None
except Exception as e:
print(f"❌ Redis 获取验证码失败: {e}")
return None
def delete_verification_code(key):
"""从 Redis 删除验证码"""
try:
redis_client.delete(f"{VERIFICATION_CODE_PREFIX}{key}")
except Exception as e:
print(f"❌ Redis 删除验证码失败: {e}")
print(f"📦 验证码存储: Redis, 过期时间: {VERIFICATION_CODE_EXPIRE}秒")
# ============ 微信登录 Session 管理（Redis 存储，支持多进程） ============
WECHAT_SESSION_EXPIRE = 300 # Session 过期时间（5分钟）
WECHAT_SESSION_PREFIX = "wechat_session:"
print(f"📦 微信登录 Redis 配置: {_REDIS_HOST}:{_REDIS_PORT}/db=0")
def set_wechat_session(state, data):
@@ -3299,11 +3343,8 @@ def send_sms_verification():
# 发送短信
if send_sms_code(phone, code, SMS_TEMPLATE_REGISTER):
# 存储验证码（5分钟有效）
verification_codes[f'phone_{phone}'] = {
'code': code,
'expires': time.time() + 300
}
# 存储验证码到 Redis（5分钟有效）
set_verification_code(f'phone_{phone}', code)
return jsonify({'message': '验证码已发送'}), 200
else:
return jsonify({'error': '验证码发送失败'}), 500
@@ -3326,11 +3367,8 @@ def send_email_verification():
# 发送邮件
if send_email_code(email, code):
# 存储验证码（5分钟有效）
verification_codes[f'email_{email}'] = {
'code': code,
'expires': time.time() + 300
}
# 存储验证码到 Redis（5分钟有效）
set_verification_code(f'email_{email}', code)
return jsonify({'message': '验证码已发送'}), 200
else:
return jsonify({'error': '验证码发送失败'}), 500
@@ -3348,8 +3386,8 @@ def register_with_phone():
if not all([phone, code, password, username]):
return jsonify({'success': False, 'error': '所有字段都是必填的'}), 400
# 验证验证码
stored_code = verification_codes.get(f'phone_{phone}')
# 验证验证码（从 Redis 获取）
stored_code = get_verification_code(f'phone_{phone}')
if not stored_code or stored_code['expires'] < time.time():
return jsonify({'success': False, 'error': '验证码已过期'}), 400
@@ -3379,8 +3417,8 @@ def register_with_phone():
db.session.commit()
# 清除验证码
del verification_codes[f'phone_{phone}']
# 清除验证码（从 Redis 删除）
delete_verification_code(f'phone_{phone}')
# 自动登录
session.permanent = True
@@ -3423,10 +3461,8 @@ def send_sms_bind_code():
code = generate_verification_code()
if send_sms_code(phone, code, SMS_TEMPLATE_REGISTER):
verification_codes[f'bind_{phone}'] = {
'code': code,
'expires': time.time() + 300
}
# 存储验证码到 Redis（5分钟有效）
set_verification_code(f'bind_{phone}', code)
return jsonify({'message': '验证码已发送'}), 200
else:
return jsonify({'error': '验证码发送失败'}), 500
@@ -3445,7 +3481,8 @@ def bind_phone():
if not phone or not code:
return jsonify({'error': '手机号和验证码不能为空'}), 400
stored = verification_codes.get(f'bind_{phone}')
# 从 Redis 获取验证码
stored = get_verification_code(f'bind_{phone}')
if not stored or stored['expires'] < time.time():
return jsonify({'error': '验证码已过期'}), 400
if stored['code'] != code:
@@ -3461,8 +3498,8 @@ def bind_phone():
user.phone = phone
user.confirm_phone()
# 清除验证码
del verification_codes[f'bind_{phone}']
# 清除验证码（从 Redis 删除）
delete_verification_code(f'bind_{phone}')
return jsonify({'message': '绑定成功', 'success': True}), 200
except Exception as e:
@@ -3517,11 +3554,8 @@ def send_email_bind_code():
code = ''.join(random.choices(string.digits, k=6))
if send_email_code(email, code):
# 存储验证码（5分钟有效）
verification_codes[f'bind_{email}'] = {
'code': code,
'expires': time.time() + 300
}
# 存储验证码到 Redis（5分钟有效）
set_verification_code(f'bind_{email}', code)
return jsonify({'message': '验证码已发送'}), 200
else:
return jsonify({'error': '验证码发送失败'}), 500
@@ -3540,7 +3574,8 @@ def bind_email():
if not email or not code:
return jsonify({'error': '邮箱和验证码不能为空'}), 400
stored = verification_codes.get(f'bind_{email}')
# 从 Redis 获取验证码
stored = get_verification_code(f'bind_{email}')
if not stored or stored['expires'] < time.time():
return jsonify({'error': '验证码已过期'}), 400
if stored['code'] != code:
@@ -3558,8 +3593,8 @@ def bind_email():
user.confirm_email()
db.session.commit()
# 清除验证码
del verification_codes[f'bind_{email}']
# 清除验证码（从 Redis 删除）
delete_verification_code(f'bind_{email}')
return jsonify({
'message': '邮箱绑定成功',
@@ -3608,8 +3643,8 @@ def register_with_email():
if not all([email, code, password, username]):
return jsonify({'success': False, 'error': '所有字段都是必填的'}), 400
# 验证验证码
stored_code = verification_codes.get(f'email_{email}')
# 验证验证码（从 Redis 获取）
stored_code = get_verification_code(f'email_{email}')
if not stored_code or stored_code['expires'] < time.time():
return jsonify({'success': False, 'error': '验证码已过期'}), 400
@@ -3638,8 +3673,8 @@ def register_with_email():
db.session.commit()
# 清除验证码
del verification_codes[f'email_{email}']
# 清除验证码（从 Redis 删除）
delete_verification_code(f'email_{email}')
# 自动登录
session.permanent = True