wzj/certmanager
1
0
Fork 0
Code Issues 7 Pull Requests Actions Packages Projects Releases 10 Wiki Activity

修复验证码bug

Browse Source
This commit is contained in:
wzj 2025-06-16 12:04:59 +08:00
parent 2b36de569d
commit 375816df87
1 changed files with 17 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
app.py
View File

@ -222,13 +222,23 @@ def generate_captcha():
def verify_captcha(user_input): def verify_captcha(user_input):
"""验证用户输入的验证码是否正确（只验证最新的4位验证码）"""
conn = get_db_connection() conn = get_db_connection()
if conn: if conn:
try: try:
cursor = conn.cursor() cursor = conn.cursor()
cursor.execute("SELECT code FROM captcha ORDER BY created_at DESC LIMIT 1") # 只查询最新的验证码（确保是4位的）
cursor.execute("""
SELECT code FROM captcha
WHERE LENGTH(code) = 4 -- 只查询4位验证码
ORDER BY created_at DESC
LIMIT 1
""")
result = cursor.fetchone() result = cursor.fetchone()
if result and user_input.upper() == result[0]: if result and user_input.upper() == result[0]:
# 验证成功后删除已使用的验证码
cursor.execute("DELETE FROM captcha WHERE code = %s", (result[0],))
conn.commit()
return True return True
return False return False
except Error as e: except Error as e:
@ -859,11 +869,11 @@ def register():
conn.close() conn.close()
# 生成新验证码 # 生成新验证码
captcha_code = generate_captcha() captcha_url = url_for('captcha') # 使用图片验证码
return render_template('register.html', return render_template('register.html',
captcha_code=captcha_code, captcha_url=captcha_url, # 前端改为显示图片验证码
registration_open=current_app.config['REGISTRATION_OPEN'], registration_open=current_app.config['REGISTRATION_OPEN'],
email_required=current_app.config['EMAIL_VERIFICATION_REQUIRED']) email_required=current_app.config['EMAIL_VERIFICATION_REQUIRED'])
@app.route('/verify-email/<token>') @app.route('/verify-email/<token>')
@ -1081,8 +1091,8 @@ def login():
cursor.close() cursor.close()
conn.close() conn.close()
captcha_code = generate_captcha() captcha_url = url_for('captcha')
return render_template('login.html', captcha_code=captcha_code) return render_template('login.html', captcha_url=captcha_url)
@app.route('/logout') @app.route('/logout')