from flask import Flask, request, jsonify
app = Flask(__name__)
# 模拟用户数据库
users = {"admin": "password123"}
@app.route('/login', methods=['POST'])
def login():
data = request.get_json()
username = data['username']
password = data['password']
if username in users and users[username] == password:
return jsonify({"status": "success", "message": "Login successful!"})
else:
return jsonify({"status": "failure", "message": "Invalid credentials."})
if __name__ == '__main__':
app.run(debug=True)
]]>
import jwt
def generate_token(username):
token = jwt.encode({"username": username}, "secret_key", algorithm="HS256")
return token
def verify_token(token):
try:
decoded = jwt.decode(token, "secret_key", algorithms=["HS256"])
return decoded["username"]
except jwt.ExpiredSignatureError:
return None
]]>