from flask import Flask, request, jsonify

import jwt

import datetime

app = Flask(__name__)

SECRET_KEY = 'your-secret-key'

def generate_token(user_id):

payload = {

'user_id': user_id,

'exp': datetime.datetime.utcnow() + datetime.timedelta(hours=1)

}

return jwt.encode(payload, SECRET_KEY, algorithm='HS256')

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

def login():

data = request.json

username = data.get('username')

password = data.get('password')

# 简单模拟用户验证

if username == 'admin' and password == '123456':

token = generate_token(1)

return jsonify({'token': token})

else:

return jsonify({'error': 'Invalid credentials'}), 401

if __name__ == '__main__':

app.run(debug=True)