统一消息系统

from flask import Flask, request, jsonify

from flask_jwt_extended import JWTManager, create_access_token, jwt_required

app = Flask(__name__)

app.config['JWT_SECRET_KEY'] = 'super-secret' # 更改此值为实际使用的密钥

jwt = JWTManager(app)

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

def login():

username = request.json.get('username', None)

password = request.json.get('password', None)

if username != 'test' or password != 'test':

return jsonify({"msg": "Bad username or password"}), 401

access_token = create_access_token(identity=username)

return jsonify(access_token=access_token)

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

@jwt_required()

def whitepapers():

# 这里可以添加获取白皮书列表的逻辑

return jsonify({"message": "欢迎访问白皮书", "user": get_jwt_identity()})

def get_jwt_identity():

from flask_jwt_extended import get_jwt_identity

return get_jwt_identity()

if __name__ == '__main__':

app.run(debug=True)

]]>