web服务器面向对象-长连接

from gevent import monkey
monkey.patch_all()  # 打补丁


import socket
import re
import gevent
import sys




class WebServer(object):


    def __init__(self, port):
        """初始化服务器套接字"""
        # 初始化服务器套接字
        self.server_soc = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.server_soc.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)  # 设置套接字复用地址
        self.server_soc.bind(("", port))
        self.server_soc.listen(128)


    def startup(self):
        """开启服务器"""
        while True:
            # 获取客户端连接
            print('等待客户端连接～～')
            client_soc, client_addr = self.server_soc.accept()


            # 为一个客户端提供服务
            gevent.spawn(self.handle_client, client_soc)


        # 关闭服务器套接字
        server_soc.close()


    def handle_client(self, client_soc):
        """处理一个客户端的请求"""
        # 获取请求头
        recv_data = client_soc.recv(1024*4)
        print(recv_data)


        if not recv_data:
            print('客户端关闭连接')
            client_soc.close()
            return


        # 获取用户请求的资源路径
        path = WebServer.parse_http(recv_data.decode('utf-8'))


        try:
            # 读取文件内容
            with open('html' + path,'rb') as file:
                file_content = file.read()


            # 返回响应数据
            response_head = 'HTTP/1.1 200 OK\r\n'
            response_head += 'Server: ItcastServer 1.0\r\n'
            response_head += 'Connection:keep-alive\r\n'
            response_head += 'Content-Length:%d\r\n' % len(file_content)
            response_head += '\r\n'


            response_body = file_content


        except:
            # 读取 404 文件内容
            with open('html/404.html','rb') as file:
                file_content = file.read()


            # 处理响应数据
            response_head = 'HTTP/1.1 404 NOT FOUND\r\n'
            response_head += 'Server: ItcastServer 1.0\r\n'
            response_head += '\r\n'


            response_body = file_content


        client_soc.send(response_head.encode())
        client_soc.send(response_body)


        # 关闭客户端套接字
        # client_soc.close()


    @staticmethod
    def parse_http(request):
        """从浏览器请求里解析出资源路径"""
        # 获取到请求行
        head_list = re.split(r'\r\n', request)
        # print(head_list)
        request_line = head_list[0]


        # 获取到请求的资源路径
        datas = re.split(r' ', request_line)
        # print(datas)


        # 对路径的安全检查
        path = datas[1]
        if path == '/':
            path = '/index.html'


        return path




def main():
    """开发一个能够响应浏览器请求的 web 服务器"""
    if len(sys.argv) < 2:
        print('必须在终端参数里指定端口号')
        return


    # 获取启动参数
    port = int(sys.argv[1])
    # print(port)


    # 创建服务器
    server = WebServer(port)
    # 开启服务器
    server.startup()




if __name__ == '__main__':

    main()

#添加内容不大，注意下面两点

#keep-alive\r\n'
 #Content-Length:%d\r\n'