统一身份认证系统的信息处理与实现

from flask import Flask, request, jsonify

app = Flask(__name__)

users = {}

@app.route('/register', methods=['POST'])

def register():

data = request.get_json()

username = data['username']

password = data['password']

if username in users:

return jsonify({'message': 'User already exists'}), 409

users[username] = password

return jsonify({'message': 'User registered successfully'}), 201

if __name__ == '__main__':

app.run(debug=True)

]]>

@app.route('/login', methods=['POST'])

def login():

data = request.get_json()

username = data['username']

password = data['password']

if username not in users or users[username] != password:

return jsonify({'message': 'Invalid credentials'}), 401

return jsonify({'message': 'Login successful'}), 200

]]>

roles = {'admin': ['read', 'write'], 'user': ['read']}

@app.route('/check_permission//', methods=['GET'])

def check_permission(role, permission):

if role not in roles:

return jsonify({'message': 'Role not found'}), 404

if permission in roles[role]:

return jsonify({'message': 'Permission granted'}), 200

return jsonify({'message': 'Permission denied'}), 403

]]>