使用genvent.socket实施群聊
from flask import Flask, request, render_template
from gevent.pywsgi import WSGIServer
from geventwebsocket.handler import WebSocketHandler
from geventwebsocket.websocket import WebSocket
import json
user_dict = {} #设置一个公共变量
app = Flask(__name__)
@app.route('/ws/<username>')
def ws(username):
user_socket = request.environ.get('wsgi.websocket') #获取客户端的服务
user_dict[username] = user_socket #将其存到公共变量中
while True:
msg = user_socket.receive() #等待接受客户端数据
u_msg= {'from_user':username,'chat':msg} #将接收的数据进行处理(处理成字典)
for uname,usocket in user_dict.items(): #循环发送向每个服务器进行发送数据
usocket.send(json.dumps(u_msg))
@app.route('/webchat')
def webchat():
return render_template('wechats.html')
if __name__ == '__main__':
server = WSGIServer(('0.0.0.0',9527),app,handler_class=WebSocketHandler) #设置ip 以及端口 还有处理方式
server.serve_forever() #运行服务
html文件代码
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>Title</title>
</head>
<body>
<p>姓名: <input type="text" id="username">
<button onclick="open_ws()">登录</button></p>
<p>内容: <input type="text" id="message">
<button onclick="send_msg()">发送</button></p>
<div id="chat_list"></div>
</body>
<script type="application/javascript">
var ws = null;
function open_ws(){
var username = document.getElementById("username").value;
ws = new WebSocket('ws://192.168.16.234:9527/ws/'+username);
ws.onopen=function(){
alert('欢迎登录');
};
ws.onmessage=function(eventMessage){
var chat = JSON.parse(eventMessage.data);
var p =document.createElement("p");
p.innerText=chat.from_user+':'+chat.chat;
document.getElementById('chat_list').appendChild(p);
}
}
function send_msg(){
var msg=document.getElementById('message').value;
ws.send(msg)
}
</script>
</html>
使用genvent.socket实施单聊模式
import json
from flask import Flask, request, render_template
from geventwebsocket.handler import WebSocketHandler
from gevent.pywsgi import WSGIServer
user_dict = {}
app= Flask(__name__)
@app.route('/my_ws/<username>')
def my_ws(username):
user_socket=request.environ.get('wsgi.websocket')
print(user_socket)
user_dict[username] = user_socket
while True:
msg = user_socket.receive()
msg_dict=json.loads(msg)
msg_dict['from_user'] = username
to_user = msg_dict.get('to_user')
usocket= user_dict.get(to_user)
if not usocket:
continue
try:
usocket.send(json.dumps(msg_dict))
except:
user_dict.pop(to_user)
@app.route('/wechat')
def wechat():
return render_template('personal.html')
if __name__ == '__main__':
server =WSGIServer(('0.0.0.0',9527),app,handler_class=WebSocketHandler)
server.serve_forever()
html代码:
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>Title</title>
</head>
<body>
<div id="chat_list"></div>
<p>你的名字<input type="text" id="username">
<button onclick="open_ws()">登录</button></p>
<p>给: <input type="text" id="to_user"></p>
<p><input type="text" id="message"> <button onclick="send_msg()">发送</button></p>
</body>
<script type="application/javascript">
var ws = null;
function open_ws() {
var username = document.getElementById("username").value;
ws = new WebSocket("ws://192.168.16.234:9527/my_ws/"+username);
ws.onopen = function () {
alert("欢迎登录");
};
ws.onmessage = function (eventMessage) {
var chat = JSON.parse(eventMessage.data);
var p = document.createElement("p");
p.innerText = chat.from_user + ":" + chat.chat;
document.getElementById("chat_list").appendChild(p);
};
}
function send_msg() {
var to_user = document.getElementById("to_user").value;
var msg = document.getElementById("message").value;
var send_str = {
to_user:to_user,
chat:msg
};
ws.send(JSON.stringify(send_str));
var p = document.createElement("p");
p.innerText = "我:" + msg;
document.getElementById("chat_list").appendChild(p);
}
</script>
</html>