修复验证码bug
This commit is contained in:
parent
89cb0af79a
commit
9c92be13d9
23
app.py
23
app.py
@ -198,29 +198,6 @@ def get_db_connection():
|
|||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
||||||
def generate_captcha():
|
|
||||||
# 生成6位随机验证码
|
|
||||||
captcha_code = ''.join(random.choices(string.ascii_uppercase + string.digits, k=6))
|
|
||||||
conn = get_db_connection()
|
|
||||||
if conn:
|
|
||||||
try:
|
|
||||||
cursor = conn.cursor()
|
|
||||||
# 清除旧的验证码
|
|
||||||
cursor.execute("DELETE FROM captcha WHERE created_at < NOW() - INTERVAL 10 MINUTE")
|
|
||||||
# 插入新验证码
|
|
||||||
cursor.execute("INSERT INTO captcha (code) VALUES (%s)", (captcha_code,))
|
|
||||||
conn.commit()
|
|
||||||
return captcha_code
|
|
||||||
except Error as e:
|
|
||||||
print(f"Database error: {e}")
|
|
||||||
return None
|
|
||||||
finally:
|
|
||||||
if conn.is_connected():
|
|
||||||
cursor.close()
|
|
||||||
conn.close()
|
|
||||||
return None
|
|
||||||
|
|
||||||
|
|
||||||
def verify_captcha(user_input):
|
def verify_captcha(user_input):
|
||||||
"""验证用户输入的验证码是否正确(只验证最新的4位验证码)"""
|
"""验证用户输入的验证码是否正确(只验证最新的4位验证码)"""
|
||||||
conn = get_db_connection()
|
conn = get_db_connection()
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user