短信申请qq

from flask import Flask, request, jsonify
import random
import string

app = Flask(__name__)

# 模拟数据库
users = {}

@app.route('/send_code', methods=['POST'])
def send_code():
    phone = request.json.get('phone')
    if phone in users:
        return jsonify({"error": "Phone number already registered"}), 400
    
    code = ''.join(random.choices(string.digits, k=6))
    users[phone] = {'code': code, 'registered': False}
    
    # 这里应该调用短信服务发送验证码，简化为打印
    print(f"Sending code {code} to {phone}")
    
    return jsonify({"message": "Verification code sent successfully"}), 200

@app.route('/register', methods=['POST'])
def register():
    phone = request.json.get('phone')
    user_code = request.json.get('code')
    new_password = request.json.get('password')
    
    if phone not in users or users[phone]['code'] != user_code:
        return jsonify({"error": "Invalid verification code"}), 400
    
    users[phone]['registered'] = True
    users[phone]['password'] = new_password
    
    return jsonify({"message": "Registration successful"}), 200

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