server # @File: 群聊 from flask import Flask, render_template, request from geventwebsocket.handler import WebSocketHandler from gevent.pywsgi import WSGIServer from geventwebsocket.websocket import WebSocket app = Flask(__name__) user_socket_list = [] @app.route('/ws') def ws(): # 一个用户的长链接wsgi.websocket user_socket = request.environ.get("wsgi.websocket") # type:WebSocket if user_socket: # 将每个链接存入列表中 user_socket_list.append(user_socket) print(len(user_socket_list), user_socket_list) while 1: msg = user_socket.receive() print(msg) for usocket in user_socket_list: if user_socket == usocket: continue try: usocket.send(msg) except: continue @app.route('/') def index(): return render_template("many_person.html") if __name__ == '__main__': http_server = WSGIServer(("0.0.0.0", 7000), app, handler_class=WebSocketHandler) http_server.serve_forever()
many_person.html
<!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>Title</title> </head> <body> <p>发送内容：<input type="text" id="message"><button onclick="send_msg()">发送</button></p> <div id="msg_list" style=" 500px"> </div> </body> <script type="application/javascript"> var ws = new WebSocket("ws://192.168.12.41:7000/ws"); ws.onmessage = function (ws_status) { console.log(ws_status.data); var ptag = document.createElement("p"); ptag.innerText = ws_status.data; document.getElementById("msg_list").appendChild(ptag) }; function send_msg() { var msg = document.getElementById("message").value; var ptag = document.createElement("p"); ptag.style.cssText = "text-align: right"; ptag.innerText = msg; document.getElementById("msg_list").appendChild(ptag); ws.send(msg); } </script> </html>