消息推送系统

from flask import Flask, request, jsonify

import jwt

import datetime

app = Flask(__name__)

SECRET_KEY = 'your-secret-key'

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

def login():

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

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

# 这里应该校验用户名和密码

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

token = jwt.encode({

'user': username,

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

}, SECRET_KEY, algorithm='HS256')

return jsonify({'token': token})

else:

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

if __name__ == '__main__':

app.run(debug=True)