协议选择的是新的Hybi-10，参考文章如下：

http://www.cnblogs.com/zhuweisky/p/3930780.html

http://blog.mycolorway.com/2011/11/22/a-minimal-python-websocket-server/

http://blog.csdn.net/icechenbing/article/details/7407588

实现一个简单的聊天室程序，代码如下：

#-*- coding:utf8 -*-

import threading

import hashlib

import socket

import base64

global clients

clients = {}

#通知客户端

def notify(message):

    for connection in clients.values():

        connection.send('%c%c%s' % (0x81, len(message), message))

#客户端处理线程

class websocket_thread(threading.Thread):

    def __init__(self, connection, username):

        super(websocket_thread, self).__init__()

        self.connection = connection

        self.username = username

    def run(self):

        print 'new websocket client joined!'

        data = self.connection.recv(1024)

        headers = self.parse_headers(data)

        token = self.generate_token(headers['Sec-WebSocket-Key'])

        self.connection.send('\

HTTP/1.1 101 WebSocket Protocol Hybi-10\r\n\

Upgrade: WebSocket\r\n\

Connection: Upgrade\r\n\

Sec-WebSocket-Accept: %s\r\n\r\n' % token)

        while True:

            try:

                data = self.connection.recv(1024)

            except socket.error, e:

                print "unexpected error: ", e

                clients.pop(self.username)

                break

            data = self.parse_data(data)

            if len(data) == 0:

                continue

            message = self.username + ": " + data

            notify(message)

    def parse_data(self, msg):

        v = ord(msg[1]) & 0x7f

        if v == 0x7e:

            p = 4

        elif v == 0x7f:

            p = 10

        else:

            p = 2

        mask = msg[p:p+4]

        data = msg[p+4:]

        return ''.join([chr(ord(v) ^ ord(mask[k%4])) for k, v in enumerate(data)])

    def parse_headers(self, msg):

        headers = {}

        header, data = msg.split('\r\n\r\n', 1)

        for line in header.split('\r\n')[1:]:

            key, value = line.split(': ', 1)

            headers[key] = value

        headers['data'] = data

        return headers

    def generate_token(self, msg):

        key = msg + '258EAFA5-E914-47DA-95CA-C5AB0DC85B11'

        ser_key = hashlib.sha1(key).digest()

        return base64.b64encode(ser_key)

#服务端

class websocket_server(threading.Thread):

    def __init__(self, port):

        super(websocket_server, self).__init__()

        self.port = port

    def run(self):

        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

        sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

        sock.bind(('127.0.0.1', self.port))

        sock.listen(5)

        print 'websocket server started!'

        while True:

            connection, address = sock.accept()

            try:

                username = "ID" + str(address[1])

                thread = websocket_thread(connection, username)

                thread.start()

                clients[username] = connection

            except socket.timeout:

                print 'websocket connection timeout!'

if __name__ == '__main__':

    server = websocket_server(9000)

    server.start()

测试页面：

<!--

@http://www.cnblogs.com/zhuweisky/p/3930780.html

-->

<!DOCTYPE html>

</html>

    <head>

        <meta charset="utf-8">

    </head>

    <body>

        <h3>WebSocketTest</h3>

        <div id="login">

            <div>

                <input id="serverIP" type="text" placeholder="服务器IP" value="127.0.0.1" autofocus="autofocus" />

                <input id="serverPort" type="text" placeholder="服务器端口" value="9000" />

                <input id="btnConnect" type="button" value="连接" onclick="connect()" />

            </div>

            <div>

                <input id="sendText" type="text" placeholder="发送文本" value="I'm WebSocket Client!" />

                <input id="btnSend" type="button" value="发送" onclick="send()" />

            </div>

            <div>

                <div>

                    来自服务端的消息

                </div>

                <textarea id="txtContent" cols="50" rows="10" readonly="readonly"></textarea>

            </div>

        </div>

    </body>

    <script>

        var socket;

        function connect() {

            var host = "ws://" + $("serverIP").value + ":" + $("serverPort").value + "/"

            socket = new WebSocket(host);

            try {

                socket.onopen = function (msg) {

                    $("btnConnect").disabled = true;

                    alert("连接成功！");

                };

                socket.onmessage = function (msg) {

                    if (typeof msg.data == "string") {

                        displayContent(msg.data);

                    }

                    else {

                        alert("非文本消息");

                    }

                };

                socket.onclose = function (msg) { alert("socket closed!") };

            }

            catch (ex) {

                log(ex);

            }

        }

        function send() {

            var msg = $("sendText").value

            socket.send(msg);

        }

        window.onbeforeunload = function () {

            try {

                socket.close();

                socket = null;

            }

            catch (ex) {

            }

        };

        function $(id) { return document.getElementById(id); }

        Date.prototype.Format = function (fmt) { //author: meizz

            var o = {

                "M+": this.getMonth() + 1, //月份

                "d+": this.getDate(), //日

                "h+": this.getHours(), //小时

                "m+": this.getMinutes(), //分

                "s+": this.getSeconds(), //秒

                "q+": Math.floor((this.getMonth() + 3) / 3), //季度

                "S": this.getMilliseconds() //毫秒

            };

            if (/(y+)/.test(fmt)) fmt = fmt.replace(RegExp.$1, (this.getFullYear() + "").substr(4 - RegExp.$1.length));

            for (var k in o)

                if (new RegExp("(" + k + ")").test(fmt)) fmt = fmt.replace(RegExp.$1, (RegExp.$1.length == 1) ? (o[k]) : (("00" + o[k]).substr(("" + o[k]).length)));

            return fmt;

        }

        function displayContent(msg) {

            $("txtContent").value += "\r\n" +new Date().Format("yyyy/MM/dd hh:mm:ss")+ ":  " + msg;

        }

        function onkey(event) { if (event.keyCode == 13) { send(); } }

    </script>

</html>

运行效果：

python简单实现websocket的更多相关文章

  1. 一个用python简单的封装了aria2的jsonrpc中adduri的脚本

    aria2是一个十分牛逼的下载神器,有时候项目需要一个很牛逼的下载中间件的话,aria2是一个不错的选择.其中支持jsonrpc和websocket的特性尤其诱人.但是python用起来还是有点不爽, ...

  2. Python简单爬虫入门三

    我们继续研究BeautifulSoup分类打印输出 Python简单爬虫入门一 Python简单爬虫入门二 前两部主要讲述我们如何用BeautifulSoup怎去抓取网页信息以及获取相应的图片标题等信 ...

  3. Python简单爬虫入门二

    接着上一次爬虫我们继续研究BeautifulSoup Python简单爬虫入门一 上一次我们爬虫我们已经成功的爬下了网页的源代码,那么这一次我们将继续来写怎么抓去具体想要的元素 首先回顾以下我们Bea ...

  4. 亲身试用python简单小爬虫

    前几天基友分享了一个贴吧网页,有很多漂亮的图片,想到前段时间学习的python简单爬虫,刚好可以实践一下. 以下是网上很容易搜到的一种方法: #coding=utf-8 import urllib i ...

  5. GJM : Python简单爬虫入门（二） [转载]

    感谢您的阅读.喜欢的.有用的就请大哥大嫂们高抬贵手"推荐一下"吧!你的精神支持是博主强大的写作动力以及转载收藏动力.欢迎转载! 版权声明:本文原创发表于 [请点击连接前往] ,未经 ...

  6. Selenium + PhantomJS + python 简单实现爬虫的功能

    Selenium 一.简介 selenium是一个用于Web应用自动化程序测试的工具,测试直接运行在浏览器中,就像真正的用户在操作一样 selenium2支持通过驱动真实浏览器(FirfoxDrive ...

  7. 【美妙的Python之中的一个】Python简单介绍及环境搭建

    美妙的Python之Python简单介绍及安装         简而言之: Python 是能你无限惊喜的语言,与众不同.             1.Python:                  ...

  8. 用 Go 编写一个简单的 WebSocket 推送服务

    用 Go 编写一个简单的 WebSocket 推送服务 本文中代码可以在 github.com/alfred-zhong/wserver 获取. 背景 最近拿到需求要在网页上展示报警信息.以往报警信息 ...

  9. python 简单图像识别--验证码

    python  简单图像识别--验证码 记录下,准备工作安装过程很是麻烦. 首先库:pytesseract,image,tesseract,PIL windows安装PIL,直接exe进行安装更方便( ...

随机推荐

  1. Servlet路径映射

    Serlvlet路径映射 当一个请求发送到servlet容器的时候,容器先会将请求的url减去当前应用上下文的路径作为servlet的映射url. 容器默认请求的是一个Servlet,所以不会直接根据 ...

  2. 第 10 章 MySQL Server 性能优化

    前言: 本章主要通过针对MySQL Server(mysqld)相关实现机制的分析,得到一些相应的优化建议.主要涉及MySQL的安装以及相关参数设置的优化,但不包括mysqld之外的比如存储引擎相关的 ...

  3. (HTTPS)-tomcat 实现 https 登录，去掉端口号

    最近项目组要给日本客户做个产品,升级服务器交由我来升级.为了测试用,想要在自己电脑上搭个服务器. 服务器需要由https登录,并且不显示端口号. 费了些劲儿,看了n多帖子,好不容易弄好了.趁在没忘记之 ...

  4. 【Python的迭代器，生成器】

    一.可迭代对象和迭代器 1.迭代的概念 上一次输出的结果为下一次输入的初始值,重复的过程称为迭代,每次重复即一次迭代,并且每次迭代的结果是下一次迭代的初始值 注:循环不是迭代 while True: ...

  5. vue使用Axios做ajax请求

    vue2.0之后,就不再对vue-resource更新,而是推荐使用axios 1. 安装 axios $ npm install axios 或 $ bower install axios 2. 在 ...

  6. Centos 环境一键部署脚本（shell脚本）

    谨以此文纪念吊炸天的Centos环境一键部署方案的新鲜出炉 辛苦大半年,产品准备上线了,BOSS亲自体验安装部署,看着超过200+页的安装文档直接崩溃了(需要部署23个基础服务),经历了超过3个小时的 ...

  7. JSP/Servlet Web 学习笔记 DaySix —— EL表达式

    1)EL从scope中得到参数时可以自动转换类型,因此对于类型转换的限制更加宽松. 2)使用EL表达式,可以简化变量和对象的访问. 3)EL表达式必须以 ${XXXX} 来表示. 4)EL提供点(.) ...

  8. c++ 库函数返回的字符串指针是否需要手动释放

    #include <stdio.h> char * tmpnam(char *s); tmpnam函数返回一个不与任何已存在文件同名的有效文件名,如果字符串s不为空,文件名也会写入它.对t ...

  9. javascript 生成 uuid

    全局唯一标识符(GUID,Globally Unique Identifier)也称作 UUID(Universally Unique IDentifier) . GUID是一种由算法生成的二进制长度 ...

  10. Windows下彻底卸载删除SQL Serever2012

    在安装了SQL Server2012之后,当由于某些原因我们需要卸载它时,我们应该怎么操作呢?相信这个问题困扰着不少人,博主经过亲身实践之后,给大家提供这样一种方法. 第一步.在控制面板里面找到程序— ...