python epoll web静态服务器简单实现

最新推荐文章于 2024-10-18 00:00:00 发布

CSND_likai

最新推荐文章于 2024-10-18 00:00:00 发布

阅读量142

点赞数

分类专栏： python web 文章标签： python

本文链接：https://blog.csdn.net/CSND_likai/article/details/105410365

版权

python web 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

最近整理非阻塞io的一些东西 select和epoll 两种方式下面是使用epoll 的实现方式

import socket
import select
import re
import sys


class WSGIServer():
    def __init__(self, port, documents_root):
        self.server_socket = socket.socket()
        # 2. 绑定本地信息
        self.server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        self.server_socket.bind(("", port))
        # 3. 设置一个监听套接字
        self.server_socket.listen(128)
        self.documents_root = documents_root
        self.epoll = select.epoll()
        self.epoll.register(self.server_socket.fileno(), select.EPOLLIN|select.EPOLLET)
        self.fd_socket = dict()

    def run_server(self):

        while True:
            print("hello")
            epoll_list = self.epoll.poll()  # 获取监测的到的请求
            print("world")
            for fd, event in epoll_list:
                if fd == self.server_socket.fileno():
                    client_socket, addr = self.server_socket.accept()
                    client_socket.setblocking(False)
                    self.epoll.register(client_socket.fileno(), select.EPOLLIN | select.EPOLLET)
                    self.fd_socket[client_socket.fileno()] = client_socket
                elif event == select.EPOLLIN:
                    request = self.fd_socket[fd].recv(1024).decode("utf-8")
                    if request:
                        self.deal_with_request(request, self.fd_socket[fd])
                    else:
                        self.epoll.unregister(fd)
                        self.fd_socket[fd].close()
                        del self.fd_socket[fd]

    def deal_with_request(self, request, client_socket):
        """为这个浏览器服务器"""

        if not request:
            return

        request_lines = request.splitlines()
        for i, line in enumerate(request_lines):
            print(i, line)

        # 提取请求的文件(index.html)
        # GET /a/b/c/d/e/index.html HTTP/1.1
        ret = re.match(r"([^/]*)([^ ]+)", request_lines[0])
        if ret:
            print("正则提取数据:", ret.group(1))
            print("正则提取数据:", ret.group(2))
            file_name = ret.group(2)
            if file_name == "/":
                file_name = "/index.html"

        # 读取文件数据
        try:
            f = open(self.documents_root + file_name, "rb")
        except:
            response_body = "file not found, 请输入正确的url"

            response_header = "HTTP/1.1 404 not found\r\n"
            response_header += "Content-Type: text/html; charset=utf-8\r\n"
            response_header += "Content-Length: %d\r\n" % len(response_body)
            response_header += "\r\n"

            # 将header返回给浏览器
            client_socket.send(response_header.encode('utf-8'))

            # 将body返回给浏览器
            client_socket.send(response_body.encode("utf-8"))
        else:
            content = f.read()
            f.close()

            response_body = content

            response_header = "HTTP/1.1 200 OK\r\n"
            response_header += "Content-Length: %d\r\n" % len(response_body)
            response_header += "\r\n"

            # 将数据返回给浏览器
            client_socket.send(response_header.encode("utf-8") + response_body)


DOCUMENTS_ROOT = "./html"


def main():
    """控制web服务器整体"""
    # python3 xxxx.py 7890
    if len(sys.argv) == 2:
        port = sys.argv[1]
        if port.isdigit():
            port = int(port)
    else:
        print("运行方式如: python3 xxx.py 7890")
        return

    print("http服务器使用的port:%s" % port)
    http_server = WSGIServer(port, DOCUMENTS_ROOT)
    http_server.run_server()


if __name__ == "__main__":
    main()