数据风险识别新购优惠

from flask import Flask, request, jsonify
import hashlib

app = Flask(__name__)

# 模拟用户数据库
users_db = {}

@app.route('/register', methods=['POST'])
def register():
    data = request.json
    username = data.get('username')
    password = data.get('password')
    
    if username in users_db:
        return jsonify({"error": "Username already exists"}), 400
    
    # 使用SHA-256进行密码哈希
    hashed_password = hashlib.sha256(password.encode()).hexdigest()
    users_db[username] = hashed_password
    
    return jsonify({"message": "User registered successfully"}), 201

@app.route('/login', methods=['POST'])
def login():
    data = request.json
    username = data.get('username')
    password = data.get('password')
    
    if username not in users_db:
        return jsonify({"error": "Invalid username or password"}), 401
    
    hashed_password = hashlib.sha256(password.encode()).hexdigest()
    
    if users_db[username] != hashed_password:
        return jsonify({"error": "Invalid username or password"}), 401
    
    return jsonify({"message": "Login successful"}), 200

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