mysql验证码过期

from flask import Flask, request, jsonify
import mysql.connector
import time

app = Flask(__name__)

# 连接数据库
db = mysql.connector.connect(
    host="localhost",
    user="root",
    password="password",
    database="test"
)
cursor = db.cursor()

# 生成验证码并存储到数据库
@app.route('/generate_captcha', methods=['POST'])
def generate_captcha():
    captcha = "1234"  # 生成验证码的逻辑
    expiration_time = int(time.time()) + 60  # 设置过期时间为60秒
    cursor.execute("INSERT INTO captchas (user_id, captcha, expiration_time) VALUES (%s, %s, %s)", (1, captcha, expiration_time))
    db.commit()
    return jsonify({"captcha": captcha})

# 验证验证码
@app.route('/verify_captcha', methods=['POST'])
def verify_captcha():
    user_input = request.json.get('captcha')
    cursor.execute("SELECT captcha, expiration_time FROM captchas WHERE user_id = %s", (1,))
    result = cursor.fetchone()
    if result:
        stored_captcha, expiration_time = result
        if int(time.time()) > expiration_time:
            return jsonify({"error": "验证码已过期"}), 400
        if user_input == stored_captcha:
            return jsonify({"message": "验证成功"}), 200
        else:
            return jsonify({"error": "验证码错误"}), 400
    else:
        return jsonify({"error": "验证码不存在"}), 400

if __name__ == '__main__':
    app.run(debug=True)