锦中融合门户系统

from flask import Flask, request, jsonify

from flask_jwt import JWT, jwt_required

app = Flask(__name__)

users = {

"admin": "password"

}

def authenticate(username, password):

if username in users and users[username] == password:

return {"username": username}

def identity(payload):

return {"username": payload['identity']}

jwt = JWT(app, authenticate, identity)

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

def login():

data = request.get_json()

username = data.get('username')

password = data.get('password')

if username in users and users[username] == password:

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

else:

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

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

@jwt_required()

def protected():

return jsonify({"message": "You are authenticated"}), 200

if __name__ == '__main__':

app.run(debug=True)