统一消息系统

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():

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

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

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

token = generate_token(1)

return jsonify({'token': token})

else:

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

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

def send_message():

token = request.headers.get('Authorization')

try:

payload = jwt.decode(token, SECRET_KEY, algorithms=['HS256'])

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

return jsonify({'status': 'success', 'message': message})

except jwt.ExpiredSignatureError:

return jsonify({'error': 'Token expired'}), 401

except jwt.InvalidTokenError:

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

if __name__ == '__main__':

app.run(ssl_context='adhoc') # 启用HTTPS