知识资产管理系统

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'] = 'your-secret-key'

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('/upload', methods=['POST'])

@jwt_required()

def upload_file():

file = request.files['file']

# Save the file to disk

file.save(file.filename)

return jsonify({"msg": "File uploaded successfully"}), 200

]]>