消息推送系统

from flask import Flask, request, jsonify

from werkzeug.security import check_password_hash

app = Flask(__name__)

users = {

"alice": {"password": "alicepass"}

}

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

def login():

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

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

if username in users and check_password_hash(users[username]['password'], password):

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

else:

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

if __name__ == '__main__':

app.run(debug=True)

]]>

from flask_socketio import SocketIO, emit

socketio = SocketIO(app)

@socketio.on('connect')

def on_connect():

print("Client connected")

@socketio.on('disconnect')

def on_disconnect():

print("Client disconnected")

@socketio.on('send_message')

def handle_send_message(json):

print("Received message: " + str(json))

emit('receive_message', json, broadcast=True)

if __name__ == '__main__':

socketio.run(app)

]]>