python + django + dwebsocket 实现简单的聊天室

使用库dwebsocket,具体参考此处

views.py:

from dwebsocket.decorators import accept_websocket,require_websocket
from collections import defaultdict
# 保存所有接入的用户地址
allconn = defaultdict(list)
@accept_websocket
def echo(request, userid):
    allresult = {}
    # 获取用户信息
    userinfo = request.user
    allresult['userinfo'] = userinfo
    # 声明全局变量
    global allconn
    if not request.is_websocket():#判断是不是websocket连接
        try:#如果是普通的http方法
            message = request.GET['message']
            return HttpResponse(message)
        except:
            return render(request, 'myproject/chat.html', allresult)
    else:
        # 将链接(请求?)存入全局字典中
        allconn[str(userid)] = request.websocket
        # 遍历请求地址中的消息
        for message in request.websocket:
            # 将信息发至自己的聊天框
            request.websocket.send(message)
            # 将信息发至其他所有用户的聊天框
            for i in allconn:
                if i != str(userid):
                    allconn[i].send(message)

chat.html :

DOCTYPE html>
<html>
<head>
    <title>django-websockettitle>
    <script src="http://code.jquery.com/jquery-1.11.1.min.js">script>
    <script type="text/javascript">
    $(document).ready(function () {
        // 获取当前时间的函数
        function getNowFormatDate() {
            var date = new Date();
            var seperator1 = "-";
            var seperator2 = ":";
            var month = date.getMonth() + 1;
            var strDate = date.getDate();
            if (month >= 1 && month <= 9) {
                month = "0" + month;
            }
            if (strDate >= 0 && strDate <= 9) {
                strDate = "0" + strDate;
            }
            var currentdate = date.getFullYear() + seperator1 + month + seperator1 + strDate
                    + " " + date.getHours() + seperator2 + date.getMinutes()
                    + seperator2 + date.getSeconds();
            return currentdate;
        }
 
        $('#connect_websocket').click(function () {
            if (window.s) {
                window.s.close()
            };
            /*创建socket连接*/
            var socket = new WebSocket("ws://" + window.location.host + "/myproject/echo/" + $('#userid').val());
            socket.onopen = function () {
                console.log('WebSocket open');//成功连接上Websocket
            };
            socket.onmessage = function (e) {
                console.log('message: ' + e.data);//打印出服务端返回过来的数据
                $('#messagecontainer').prepend('

' + e.data + '

'); }; // Call onopen directly if socket is already open if (socket.readyState === WebSocket.OPEN) socket.onopen(); window.s = socket; }); $('#send_message').click(function () { //如果未连接到websocket if (!window.s) { alert("websocket未连接."); } else { window.s.send($('#username').val() + ' ' + getNowFormatDate() + ':
' + $('#message').val());//通过websocket发送数据 } }); $('#close_websocket').click(function () { if (window.s) { window.s.close();//关闭websocket console.log('websocket已关闭'); } }); }); script> head> <body> <br> <input type="hidden" id="userid" value="{{ userinfo.id }}"/> <input type="hidden" id="username" value="{{ userinfo.username }}"/> <input type="text" id="message" value="Hello, World!"/> <button type="button" id="connect_websocket">连接 websocketbutton> <button type="button" id="send_message">发送 messagebutton> <button type="button" id="close_websocket">关闭 websocketbutton> <h1>Received Messagesh1> <div id="messagecontainer"> div> body> html>

urls.py :

url(r'^echo/(?P[0-9]+)$', views.echo, name='echo'),

 

阅读原文

Django WebSocket Redis 在线聊天室

https://github.com/huguodong/dj_dwebsocket(demo代码)

你可能感兴趣的:(python + django + dwebsocket 实现简单的聊天室)