张工：李工，最近我们部门需要搭建一个统一的身份认证平台，你有什么好的建议吗？
李工：首先，我们需要明确平台的核心功能，比如用户登录、权限管理等。其次，可以使用OAuth 2.0或JWT（JSON Web Token）来实现安全的认证机制。
张工：听起来不错，那具体怎么实现呢？能给我一些代码示例吗？
李工：当然可以。我们可以先从简单的JWT开始。下面是一个Python Flask应用中的JWT认证代码示例：
from flask import Flask, request, jsonify
import jwt
from datetime import datetime, timedelta
app = Flask(__name__)
SECRET_KEY = 'your_secret_key'
@app.route('/login', methods=['POST'])
def login():
username = request.json.get('username')
password = request.json.get('password')
# 这里假设用户名和密码正确
if username == 'admin' and password == 'admin':
payload = {
'exp': datetime.utcnow() + timedelta(minutes=30),
'iat': datetime.utcnow(),
'sub': username
}
token = jwt.encode(payload, SECRET_KEY, algorithm='HS256')
return jsonify({'token': token}), 200
else:
return jsonify({'error': 'Invalid credentials'}), 401
@app.route('/protected', methods=['GET'])
def protected():
token = request.headers.get('Authorization').split()[1]
try:
payload = jwt.decode(token, SECRET_KEY, algorithms=['HS256'])
return jsonify({'message': f'Hello, {payload["sub"]}!'}), 200
except jwt.ExpiredSignatureError:
return jsonify({'error': 'Token expired'}), 401
except jwt.InvalidTokenError:
return jsonify({'error': 'Invalid token'}), 401
if __name__ == '__main__':
app.run(debug=True)
]]>
张工：这个例子很清晰，它展示了如何生成和验证JWT令牌。那么，关于权限管理呢？
李工：对于权限管理，我们可以定义不同的角色并分配相应的权限。例如，使用一个角色表来存储用户的角色信息，并在每次请求时检查用户的权限。
张工：明白了，这确实是一个实用的手册。感谢你的分享！
李工：不客气，希望对你有帮助。
]]>