群聊
from flask import Flask,request,render_templatefrom geventwebsocket.handler import WebSocketHandlerfrom gevent.pywsgi import WSGIServerfrom geventwebsocket.websocket import WebSocketapp = Flask(__name__)user_socket_list = []@app.route("/conn_ws")def ws_app(): ???# print(request.environ) ???user_socket = request.environ.get("wsgi.websocket") ?# type:WebSocket ???user_socket_list.append(user_socket) ???print(len(user_socket_list),user_socket_list) ???while True: ???????msg = user_socket.receive() ???????print(msg) ???????for usocket in ?user_socket_list: ???????????usocket.send(msg) ???????# user_socket.send(msg) ???# 1 已经开启的websocket连接 ???# 3 开启了websocket连接 然后 断开了 ???# return "666"@app.route("/")def index(): ???return render_template("my_ws.html")if __name__ == ‘__main__‘: ???# app.run() ???http_serv = WSGIServer(("0.0.0.0",9527),app,handler_class=WebSocketHandler) ???http_serv.serve_forever()<!DOCTYPE html><html lang="en"><head> ???<meta charset="UTF-8"> ???<title>Title</title></head><body><input type="text" id="send_str"><button id="send_btn" onclick="send()">发送消息</button><p><div id="chat_list"></div></p></body><script type="application/javascript"> ???var ws = new WebSocket("ws://192.168.11.182:9527/conn_ws"); ???ws.onmessage = function (messageEvent) { ???????console.log(messageEvent.data); ???????var ptag = document.createElement("p"); ???????ptag.innerText = messageEvent.data; ???????document.getElementById("chat_list").appendChild(ptag); ???}; ???// document.getElementById("send_btn").addEventListener("onclick",) ???function send() { ???????var send_str = document.getElementById("send_str").value; ???????ws.send(send_str); ???}</script></html>单聊
import jsonfrom flask import Flask,request,render_templatefrom geventwebsocket.handler import WebSocketHandlerfrom gevent.pywsgi import WSGIServerfrom geventwebsocket.websocket import WebSocketapp = Flask(__name__)user_socket_dict = {}@app.route("/conn_ws/<user_nick>")def ws_app(user_nick): ???# print(request.environ) ???user_socket = request.environ.get("wsgi.websocket") ?# type:WebSocket ???user_socket_dict[user_nick] = user_socket ???print(len(user_socket_dict),list(user_socket_dict.keys())) ???while True: ???????msg = user_socket.receive() ???????# {from_user:"alexDSB",to_user:"YWB",message:"666"} ???????msg_dict = json.loads(msg) ???????to_user = msg_dict.get("to_user") ???????to_user_socket = user_socket_dict.get(to_user) ???????to_user_socket.send(msg) ???# 1 已经开启的websocket连接 ???# 3 开启了websocket连接 然后 断开了@app.route("/")def index(): ???return render_template("my_ws.html")if __name__ == ‘__main__‘: ???# app.run() ???http_serv = WSGIServer(("0.0.0.0",9527),app,handler_class=WebSocketHandler) ???http_serv.serve_forever()<!DOCTYPE html><html lang="en"><head> ???<meta charset="UTF-8"> ???<title>Title</title></head><body><p><input type="text" id="nick"><button onclick="login()">登陆聊天室</button></p>发送给：<input type="text" id="to_user">消息：<input type="text" id="send_str"><button id="send_btn" onclick="send()">发送消息</button><p><div id="chat_list"></div></p></body><script type="application/javascript"> ???var ws = null; ???// ws.onopen = function(){ ???// ????ws.send("123"); ???// } ???function login() { ???????var nick = document.getElementById("nick").value; ???????ws = new WebSocket("ws://192.168.11.182:9527/conn_ws/"+nick); ???????ws.onmessage = function (messageEvent) { ???????????console.log(messageEvent.data); ???????????var ptag = document.createElement("p"); ???????????var message = JSON.parse(messageEvent.data); ???????????ptag.innerText =message.from_user + " : " + message.message ; ???????????document.getElementById("chat_list").appendChild(ptag); ???????}; ???} ???// document.getElementById("send_btn").addEventListener("onclick",) ???function send() { ???????var message = document.getElementById("send_str").value; ???????var send_str = { ???????????from_user:document.getElementById("nick").value, ???????????to_user:document.getElementById("to_user").value, ???????????message:message ???????}; ???????var json_send_str = JSON.stringify(send_str); ???????ws.send(json_send_str); ???}</script></html>基于flask 写的web_socket 单聊和群聊
原文地址：https://www.cnblogs.com/awfj/p/10602761.html