账号安全服务新年促销

from flask import Flask, request, jsonify
import pyotp

app = Flask(__name__)

@app.route('/generate_otp', methods=['POST'])
def generate_otp():
    data = request.json
    user_id = data.get('user_id')
    secret_key = pyotp.random_base32()
    totp = pyotp.TOTP(secret_key)
    otp = totp.now()
    # 这里应该将secret_key和otp存储到数据库中，与user_id关联
    return jsonify({'user_id': user_id, 'otp': otp})

@app.route('/verify_otp', methods=['POST'])
def verify_otp():
    data = request.json
    user_id = data.get('user_id')
    otp = data.get('otp')
    # 从数据库中获取与user_id关联的secret_key
    secret_key = get_secret_key_from_db(user_id)
    totp = pyotp.TOTP(secret_key)
    if totp.verify(otp):
        return jsonify({'status': 'success'})
    else:
        return jsonify({'status': 'failure'})

def get_secret_key_from_db(user_id):
    # 模拟从数据库获取数据
    return 'BASE32ENCODEDSECRETKEY'

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