编写一个简单的Web静态服务器

import socket
import re
import struct

from multiprocessing import Process

#静态文件根目录
HTML_ROOT_DIR = "./html"

class HTTPServer(object):
    """Web服务器对象"""
    def __init__(self):
        self.server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

    def start(self):
        """启动Web服务器"""
        self.server_socket.listen(128)
        while True:
            client_socket, client_address = self.server_socket.accept()
            print("[%s,%s]用户连接上了" % client_address)
            handle_client_process = Process(target=self.handle_client, args=(client_socket,))
            handle_client_process.start()
            client_socket.close()

    def handle_client(self, client_socket):
        """处理客户端请求"""
        # 获取客户端请求数据
        request_data = client_socket.recv(1024)
        request_lines = request_data.splitlines()

        # 解析请求报文'GET / HTTP/1.1'
        request_start_line = request_lines[0]
        file_name = re.match(r"\w+ +(/[^ ]*)", request_start_line.decode("utf-8")).group(1)

        if file_name == "/":
            file_name = "/index.html"
        # 打开文件,读取内容

        try:
            file = open(HTML_ROOT_DIR + file_name, "rb")
        except IOError:
            response_start_line = "HTTP/1.1 200 OK\r\n"
            response_headers = "Server:My server\r\n"
            response_body = "The file is not found!"
        else:
            file_data = file.read()
            file.close()

            # 构造响应数据
            response_start_line = "HTTP/1.1 200 OK\r\n"
            response_headers = "Server:My server\r\n"
            response_body = file_data.decode("utf-8")

        response = response_start_line + response_headers + "\r\n" + response_body

        # 向客户端返回响应数据
        client_socket.send(bytes(response, "utf-8"))

        # 处理请求结束,关闭连接
        client_socket.close()

    def bind(self, port):
        self.server_socket.bind(("", port))

def main():
    http_server = HTTPServer()
    http_server.bind(8080)
    http_server.start()

if __name__ == "__main__":
    main()

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/lx1315998513/article/details/88085284