老男孩python学习自修第二十一天【socket】

1. 使用python编写一个静态的web服务器，能够处理静态页面的http请求

原理：

a. 使用socket进行服务端和浏览器之间的通信

b. 使用多线程处理多个客户端浏览器的请求

c. 解析用户请求并使用http协议发送响应数据

实现：

# coding:UTF-8

import socket
import re

from multiprocessing import Process

def handle_request(client_socket):
    # 定义html服务器根目录
    HTML_ROOT_DIR = "./html"

    # 接收用户的请求
    request_data = client_socket.recv(1024)

    # 解析用户的请求
    request_lines = request_data.splitlines()
    request_first_line = request_lines[0].decode("UTF-8")
    # 使用正则表达式取出请求的文件名
    file_name = re.match(r"\w+ +(/[^ ]*) ", request_first_line).group(1)
    # 默认请求的文件名为index.html
    if "/" == file_name:
        file_name = "/index.html"
    file_path = HTML_ROOT_DIR + file_name

    # 读取请求资源的内容, 并构造响应报文
    try:
        file = open(file_path, "rb")
    except IOError:
        response_first_line = "HTTP/1.1 404 Not Found\r\n"
        response_headers = "Server: My Web Static Server\r\n"
        response_body = "file is not found"
        response = response_first_line + response_headers + "\r\n" + response_body
    else:
        data = file.read()
        file.close()
        response_first_line = "HTTP/1.1 200 OK\r\n"
        response_headers = "Server: My Web Static Server\r\n"
        response_body = data.decode("UTF-8")
        response = response_first_line + response_headers + "\r\n" + response_body

    # 发送响应报文
    client_socket.send(bytes(response, encoding="UTF-8"))

    # 关闭socket连接资源
    client_socket.close()

if __name__ == "__main__":
    '''这是主函数程序运行入口，如果该文件被当做模块导入则不执行'''
    # 创建一个服务端的socket
    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    # 设置服务端socket重复利用端口地址
    server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    # 绑定服务端socket的端口
    server_socket.bind(("",8001))
    # 设置服务端socket最大同时响应128个客户端请求
    server_socket.listen(128)

    # 服务端永久等待客户端的请求， 来一个客户端请求则创建一个新的线程响应一个请求
    while True:
        client_socket, client_address = server_socket.accept()
        print('["%s", "%s"]用户已经连接上服务器！' % client_address)
        handler = Process(target=handle_request, args=(client_socket,))
        handler.start()
        client_socket.close()