05_服務wsgi協議的伺服器.py

import socket

import re
import multiprocessing
import 多程序http伺服器.mini_frame05




class WebServer(object):


    def __init__(self):
        # 1. 建立套接字
        self.tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.tcp_server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)


        # 2. 繫結
        self.tcp_server_socket.bind(("", 7890))


        # 3. 變為監聽套接字
        self.tcp_server_socket.listen(128)


    def __del__(self):
        # 6. 關閉監聽套接字
        self.tcp_server_socket.close()


    def service_client(self, new_socket):
        """為客戶端返回資料"""


        # 1.接收瀏覽器傳送過來的request請求(符合http協議的格式) ，即http請求
        # GET /index.html HTTP/1.1
        request = new_socket.recv(1024).decode("utf-8")


        request_lines = request.splitlines()
        print("")
        print(">" * 20)
        print(request_lines)


        # 2.通過正則解析 GET /index.html HTTP/1.1 獲取檔名 即 /index.html
        file_name = ""
        ret = re.match(r"[^/]+(/[^ ]*)", request_lines[0])
        if ret:
            file_name = ret.group(1)
            if file_name == "/":
                file_name = "/index.html"
        if not file_name.endswith('.py'):  # 靜態資源
            # 3. 通過檔名 /index.html 開啟檔案讀取資料後 組織符合http格式的response
            # 即應答報文 header + body 給瀏覽器
            try:
                f = open("./html" + file_name, "rb")
            except:
                response = "HTTP/1.1 404 NOT FOUND\r\n"
                response += "\r\n"
                response += "------file not found-----"
                new_socket.send(response.encode("utf-8"))
            else:
                # 3.1 準備傳送給瀏覽器的資料---response_header
                response_header = "HTTP/1.1 200 OK\r\n"
                response_header += "\r\n"
                # 3.2 準備傳送給瀏覽器的資料---response_body
                response_body = f.read()
                f.close()


                # 將response header傳送給瀏覽器
                new_socket.send(response_header.encode("utf-8"))
                # 將response body傳送給瀏覽器
                new_socket.send(response_body)
        else:
            # 傳參字典
            env = dict()
            env["url"] = file_name
            # 呼叫介面
            response_body = 多程序http伺服器.mini_frame05.application(env, self.set_respose_header)
            # 需要函式呼叫後才有例項屬性，設定應答頭
            response_header = "HTTP/1.1 %s\r\n" % self.statue
            # response_header += "%s:%s\r\n" % (self.headers[0][0], self.headers[0][1])
            for i in self.headers:
                response_header += "%s:%s\r\n" % (i[0], i[1])
            response_header += "\r\n"
            # 設定應答體
            response = response_header + response_body


            new_socket.send(response.encode("utf-8"))
        # 關閉套接
        new_socket.close()


    def set_respose_header(self, statue, headers):
        self.statue = statue
        self.headers = headers


    def run_for_ever(self):
        """用來完成整體的控制"""


        while True:
            # 4. 等待新客戶端的連結
            new_socket, client_addr = self.tcp_server_socket.accept()


            # 5. 為這個客戶端服務
            p = multiprocessing.Process(target=self.service_client, args=(new_socket,))
            p.start()


            new_socket.close()




def main():
    """業務流程"""
    my_web_server = WebServer()
    my_web_server.run_for_ever()


if __name__ == "__main__":

    main()

mini_frame05.py

# 處理動態資源
def index():
    return "index is show"




def center():
    return "center is show"




def profile():
    return "profile is show"




# 符合wsgi協議的介面
# environ（字典）:儲存請求資訊 index.py
# start_response（函式的引用）: 響應頭response_header
# 1.狀態碼 2.響應頭列表headers
# return: 響應體response_body
def application(environ, start_response):
    file_name = environ["url"]
    start_response('200 OK', [('Content-Type', 'text/html;charset=utf-8')])
    if file_name == "/index.py":
        return index()
    elif file_name == "/center.py":
        return center()
    elif file_name == "/profile.py":
        return profile()
    else:
        return "%s 沒有發現對應的函式" % file_name