静态Web服务器-返回指定页面数据

前面说过返回固定页面数据返回固定页面，这次返回客户端指定页面数据

使用捕获异常的方法来判断客户端请求内容**不存在时返回404

import socket

def main():
    # 创建服务端套接字
    tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    # 设置端口号复用, 程序退出,端口号立即释放
    tcp_server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, True)
    # 绑定端口号
    tcp_server_socket.bind(('', 8000))
    # 设置监听
    tcp_server_socket.listen(128)
    # 循环等待接受客户端的连接请求
    while True:
        # 等待接受客户端的连接请求
        new_socket, ip_port = tcp_server_socket.accept()
        # 接收数据
        recv_data = new_socket.recv(4096)
        # 判断客户端是否发送数据
        if len(recv_data) == 0:
            print('客户端断开连接了')
            # 关闭服务于客户端的套接字
            new_socket.close()
            continue
        # 对二进制数据进行解码
        recv_content = recv_data.decode('utf-8')
        print(recv_content)
        # 以空格进行分割
        request_list = recv_content.split(' ', 2)
        # 对请求数据切片获取路径
        request_path = request_list[1]
        print(request_path)
        # 判断返回的是否是根目录,如果是则返回首页数据
        if request_path == '/':
            request_path = '/index.html'
        try:
            # 打开发送内容,此时是以'rb'方式打开,打开内容为二进制
            with open('static' + request_path, 'rb') as file:
                # 读取内容
                file_data = file.read()

        except Exception as e:
            # 请求资源不存在,返回404数据
            # 响应行
            response_line = 'HTTP/1.1 404 NOT FOUND\r\n'
            # 响应头
            response_header = 'Server: PWS/1.1\r\n'
            # 打开发送内容,此时是以'rb'方式打开,打开内容为二进制
            with open('static/error.html', 'rb') as file:
                # 读取内容
                file_data = file.read()
            # 响应体,此时的响应体是二进制
            response_body = file_data
            # 将数据封装成http响应报文格式
            response = (response_line +
                        response_header +
                        '\r\n').encode('utf-8') + response_body

            # 发送响应报文数据
            new_socket.send(response)

        else:
            # 响应行
            response_line = 'HTTP/1.1 200 OK\r\n'
            # 响应头
            response_header = 'Server: PWS/1.1\r\n'
            # 响应体,此时的响应体是二进制
            response_body = file_data
            # 将数据封装成http响应报文格式
            response = (response_line +
                        response_header +
                        '\r\n').encode('utf-8') + response_body
            # 发送响应报文数据
            new_socket.send(response)
        finally:
            # 关闭服务于客户端的套接字
            new_socket.close()

if __name__ == '__main__':
    main()

使用导入os模块的方法判断客户端请求内容不存在时返回404

示例：os.path.exists()

import socket
import os

def main():
    # 创建服务端套接字
    tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    # 设置端口号复用, 程序退出,端口号立即释放
    tcp_server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, True)
    # 绑定端口号
    tcp_server_socket.bind(('', 8000))
    # 设置监听
    tcp_server_socket.listen(128)
    # 循环等待接受客户端的连接请求
    while True:
        # 等待接受客户端的连接请求
        new_socket, ip_port = tcp_server_socket.accept()
        # 接收数据
        recv_data = new_socket.recv(4096)
        # 判断客户端是否发送数据
        if len(recv_data) == 0:
            print('客户端断开连接了')
            # 关闭服务于客户端的套接字
            new_socket.close()
            continue
        # 对二进制数据进行解码
        recv_content = recv_data.decode('utf-8')
        print(recv_content)
        # 以空格进行分割
        request_list = recv_content.split(' ', 2)
        # 对请求数据切片获取路径
        request_path = request_list[1]
        print(request_path)
        # 判断返回的是否是根目录,如果是则返回首页数据
        if request_path == '/':
            request_path = '/index.html'

        value = os.path.exists('static' + request_path)
        if value == True:
            # 打开发送内容,此时是以'rb'方式打开,打开内容为二进制
            with open('static' + request_path, 'rb') as file:
                # 读取内容
                file_data = file.read()

            # 响应行
            response_line = 'HTTP/1.1 200 OK\r\n'
            # 响应头
            response_header = 'Server: PWS/1.1\r\n'
            # 响应体,此时的响应体是二进制
            response_body = file_data
            # 将数据封装成http响应报文格式
            response = (response_line +
                        response_header +
                        '\r\n').encode('utf-8') + response_body

            # 发送响应报文数据
            new_socket.send(response)

            # 关闭服务于客户端的套接字
            new_socket.close()
        else:
            # 请求资源不存在,返回404数据
            # 响应行
            response_line = 'HTTP/1.1 404 NOT FOUND\r\n'
            # 响应头
            response_header = 'Server: PWS/1.1\r\n'
            # 打开发送内容,此时是以'rb'方式打开,打开内容为二进制
            with open('static/error.html', 'rb') as file:
                # 读取内容
                file_data = file.read()
            # 响应体,此时的响应体是二进制
            response_body = file_data
            # 将数据封装成http响应报文格式
            response = (response_line +
                        response_header +
                        '\r\n').encode('utf-8') + response_body

            # 发送响应报文数据
            new_socket.send(response)
            # 关闭服务于客户端的套接字
            new_socket.close()

if __name__ == '__main__':
    main()