httpserver v2.0
io多路复用 和 http训练

from socket import *
from select import *
class HTTPServer:   # 具体功能实现
    def __init__(self, host='0.0.0.0', port=8000, dir=None):
        self.host = host
        self.port = port
        self.dir = dir
        self.address = (host, port)
        self.rlist = []  # 多路复用列表
        self.wlist = []
        self.xlist = []       
        self.create_socket()  # 实例化对象时直接创建套接字
        self.bind() 
    def create_socket(self):     # 创建套接字
        self.sockfd = socket()
        self.sockfd.setsockopt(SOL_SOCKET,SO_REUSEADDR, 1)  
    def bind(self):    # 绑定地址
        self.sockfd.bind(self.address)  
    def serve_forever(self):    # 启动服务
        self.sockfd.listen(3)
        print("Listen the port %d" % self.port)      
        self.rlist.append(self.sockfd)   # IO多路复用接收客户端请求
        while True:
            rs, wx, xs = select(self.rlist,self.wlist,self.xlist)
            for r in rs:
                if r is self.sockfd:
                    c, addr = r.accept()
                    self.rlist.append(c)
                else:                    
                    self.handle(r)  # 处理请求
    def handle(self, connfd):        
        request = connfd.recv(4096)   # 接收HTTP请求       
        if not request:    # 客户端断开
            self.rlist.remove(connfd)
            connfd.close()
            return       
        request_line = request.splitlines()[0]    # 提取请求内容 (字节串按行分割)
        info = request_line.decode().split(' ')[1]
        print(connfd.getpeername(), ':', info)
        if info=='/' or info[-5:]=='.html': #将请求内容分为1，请求网页2，其他
            self.get_html(connfd, info)
        else:
            self.get_data(connfd, info)    
    def get_html(self, connfd, info):  # 返回网页
        if info == '/':            
            filename = self.dir + "/index.html"   # 请求主页
        else:
            filename = self.dir + info
        try:
            fd = open(filename)
        except Exception:    # 网页不存在           
            response = "HTTP/1.1 404 Not Found\r\n"
            response += 'Content-Type:text/html\r\n'
            response += '\r\n'
            response += '<h1>Sorry....</h1>'
        else:    # 网页存在           
            response = "HTTP/1.1 200 OK\r\n"
            response += 'Content-Type:text/html\r\n'
            response += '\r\n'
            response += fd.read()
        finally:            
            connfd.send(response.encode())  # 将响应发送给浏览器   
    def get_data(self, connfd, info):     # 其他数据
        response = "HTTP/1.1 200 OK\r\n"
        response += 'Content-Type:text/html\r\n'
        response += '\r\n'
        response += "<h1>Waiting for httpserver 3.0</h1>"
        connfd.send(response.encode())
if __name__ == "__main__":   # 用户使用HTTPServer
    #通过 HTTPServer类快速搭建服务，展示自己的网页   
    HOST = '0.0.0.0'   # 用户决定的参数
    PORT = 8000
    DIR = './static'  # 网页存储位置
    httpd = HTTPServer(HOST, PORT, DIR)  # 实例化对象
    httpd.serve_forever()  # 启动服务