python怎么写web socket接口脚本_python简单实现websocket

#-*- coding:utf8 -*-

importthreadingimporthashlibimportsocketimportbase64globalclients

clients={}#通知客户端

defnotify(message):for connection inclients.values():

connection.send('%c%c%s' % (0x81, len(message), message))#客户端处理线程

classwebsocket_thread(threading.Thread):def __init__(self, connection, username):

super(websocket_thread, self).__init__()

self.connection=connection

self.username=usernamedefrun(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)whileTrue:try:

data= self.connection.recv(1024)exceptsocket.error, e:print "unexpected error:", e

clients.pop(self.username)breakdata=self.parse_data(data)if len(data) ==0:continuemessage= self.username + ":" +data

notify(message)defparse_data(self, msg):

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

if v == 0x7e:

p= 4

elif v == 0x7f:

p= 10

else:

p= 2mask= msg[p:p+4]

data= msg[p+4:]return ''.join([chr(ord(v) ^ ord(mask[k%4])) for k, v inenumerate(data)])defparse_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'] =datareturnheadersdefgenerate_token(self, msg):

key= msg + '258EAFA5-E914-47DA-95CA-C5AB0DC85B11'ser_key=hashlib.sha1(key).digest()returnbase64.b64encode(ser_key)#服务端

classwebsocket_server(threading.Thread):def __init__(self, port):

super(websocket_server, self).__init__()

self.port=portdefrun(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!'

whileTrue:

connection, address=sock.accept()try:

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

thread=websocket_thread(connection, username)

thread.start()

clients[username]=connectionexceptsocket.timeout:print 'websocket connection timeout!'

if __name__ == '__main__':

server= websocket_server(9000)

server.start()