python--websocket数据解析

# websocket实现原理
'''
1.服务端开启socket，监听ip和端口
2.客户端发送连接请求（带上ip和端口）
3.服务端允许连接
4.客户端生成一个随机字符串，和magic string组合进行一个sha1加密，加密。并将随机字符串发送给服务端
5.然后服务端也要用相同的方式进行加密。
6.然后服务端将加密之后的密串返回给客户端
7.客户端将服务端返回的密串和自己加密之后的密串进行比对，如果一样，说明遵循同样的协议。如果不一样，就没法玩了·····
'''

import socket
import base64
import hashlib
from pprint import pprint


def get_headers(data):
    """
    将请求头格式化成字典
    :param data:
    :return:
    """
    header_dict = {}
    data = str(data, encoding='utf-8')

    header, body = data.split('\r\n\r\n', 1)
    header_list = header.split('\r\n')
    for i in range(0, len(header_list)):
        if i == 0:
            if len(header_list[i].split(' ')) == 3:
                header_dict['method'], header_dict['url'], header_dict['protocol'] = header_list[i].split(' ')
        else:
            k, v = header_list[i].split(':', 1)
            header_dict[k] = v.strip()
    return header_dict


server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.bind(('localhost', 8080))
server.listen(5)

# 等待用户连接
conn, addr = server.accept()

# 握手消息
content = conn.recv(1024)
'''
>>> print(content)

b'GET / HTTP/1.1\r\nHost: localhost:8080\r\nConnection: Upgrade\r\nPragma: no-cache\r\nCache-Control: no-cache\r\nUpgrade: websocket\