企业身份管理特惠

from flask import Flask, request, jsonify
from functools import wraps

app = Flask(__name__)

# 模拟用户数据库
users = {
    'user1': {'password': 'password1', 'roles': ['admin']},
    'user2': {'password': 'password2', 'roles': ['user']}
}

def require_auth(f):
    @wraps(f)
    def decorated(*args, **kwargs):
        auth = request.authorization
        if not auth or not check_auth(auth.username, auth.password):
            return jsonify({"message": "Authentication required."}), 401
        return f(*args, **kwargs)
    return decorated

def check_auth(username, password):
    if username in users and users[username]['password'] == password:
        return True
    return False

@app.route('/protected')
@require_auth
def protected():
    return jsonify({"message": "You are authorized to see this."})

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