1. 程式人生 > >Python簡單實現Web服務器

Python簡單實現Web服務器

Python 簡單web服務器實現

import socket import re import gevent from gevent import monkey monkey.patch_all()# 識別等待時間,讓協程切換 def client_handler(client_socket): ‘‘‘接收客戶端鏈接請求,響應對應的的數據‘‘‘ # 接收數據 request_data = client_socket.recv(4096) # 判斷是否接收到數據 if not request_data: print("客戶端已經斷開鏈接") client_socket.close() return # 對接收到的客戶端請求數據進行解碼 request_str_data = request_data.decode() #對請求的報文進行分割,分割出一個請求各行數列表 data_list = request_str_data.split("\r\n") # 拿到請求行數據,請求行數據是列表第0個元素 request_line = data_list[0] # 通過正則匹配到我們請求的文件路徑 result = re.match(r"\w+\s+(\S+)", request_line) # 判斷匹配的請求文件路徑是否存在 if not result: print("請求路徑不存在") client_socket.close() return path_info = result.group(1) print("用戶請求信息%s" % str(path_info)) # 設置請求域名默認跳轉首頁 if path_info == "/": # 指定首頁地址 path_info = "/index.html" # 響應頭 response_header = "Server: PWS1.0\r\n" try: # 響應體,打開客戶端請求的數據 with open("./html" + path_info, "rb") as file: file_data = file.read() except Exception as e: # 構造請求錯誤響應報文 response_line = "HTTP/1.1 404 NOT FOUND\r\n" response_body = "EROOR!!! %s".center(800) %(e) # 拼接響應報文 response_data = response_line + response_header + "\r\n" + response_body # 給客戶端發送響應報文 client_socket.send(response_data.encode()) else: # 構造請求成功響應報文 response_line = "HTTP/1.1 200 OK\r\n" response_body = file_data response_data = (response_line + response_header + "\r\n").encode() + response_body # 發送響應報文 client_socket.send(response_data) finally: # 關閉套接字 client_socket.close() # 創建主函數,定義套接字 def main(): # 創建套接字,指定IP和數據報類型 server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # 設置端口復用 server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, True) # 設置綁定,監聽,接收鏈接 server_socket.bind(("", 4433)) server_socket.listen(128) # 服務器是面向多客戶,循環接收客戶端請求鏈接 while True: client_socket, client_address = server_socket.accept() print("服務端接收到來自%s的鏈接請求" % str(client_address)) # 處理鏈接請求 # client_handler(client_socket) # 創建協程實現多任務 g1 = gevent.spawn(client_handler, client_socket) # 保持主進程存活(阻塞主進程,等待協程g1執行完再退出) # g1.join() # 程序入口 if __name__ == ‘__main__‘: main()

面向對象封裝上面代碼

import socket
import re
import gevent
from gevent import monkey
import sys

monkey.patch_all()# 識別等待時間,讓協程切換

class HTTPServer(object):

    def __init__(self, port):
        """完成實例對象的初始化操作"""
        # 創建套接字,指定IP和數據報類型
        server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

        # 設置端口復用
        server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, True)
        # 設置綁定,監聽,接收鏈接
        server_socket.bind(("", 4433))
        server_socket.listen(128)

        # 套接字對象的引用
        self.server_socket = server_socket

    # 套接字的鏈接等待
    def start(self):
        # 服務器是面向多客戶,循環接收客戶端請求鏈接
        while True:
            client_socket, client_address = self.server_socket.accept()
            print("服務端接收到來自%s的鏈接請求" % str(client_address))
            # 處理鏈接請求
            # client_handler(client_socket)
            # 創建協程實現多任務
            g1 = gevent.spawn(self.client_handler, client_socket)
            # 保持主進程存活(阻塞主進程,等待協程g1執行完再退出)
            #g1.join()

    def client_handler(self, client_socket):
        ‘‘‘接收客戶端鏈接請求,響應對應的的數據‘‘‘
        # 接收數據
        request_data = client_socket.recv(4096)
        # 判斷是否接收到數據
        if not request_data:
            print("客戶端已經斷開鏈接")
            client_socket.close()
            return

        # 對接收到的客戶端請求數據進行解碼
        request_str_data = request_data.decode()

        #對請求的報文進行分割,分割出一個請求各行數列表
        data_list = request_str_data.split("\r\n")

        # 拿到請求行數據,請求行數據是列表第0個元素
        request_line = data_list[0]

        # 通過正則匹配到我們請求的文件路徑
        result = re.match(r"\w+\s+(\S+)", request_line)

        # 判斷匹配的請求文件路徑是否存在
        if not result:
            print("請求路徑不存在")
            client_socket.close()
            return
        path_info = result.group(1)
        print("用戶請求信息%s" % str(path_info))

        # 設置請求域名默認跳轉首頁
        if path_info == "/":
            # 指定首頁地址
            path_info = "/index.html"

        # 響應頭
        response_header = "Server: PWS1.0\r\n"
        try:
            # 響應體,打開客戶端請求的數據
            with open("./html" + path_info, "rb") as file:
                file_data = file.read()
        except Exception as e:
            # 構造請求錯誤響應報文
            response_line = "HTTP/1.1 404 NOT FOUND\r\n"
            response_body = "EROOR!!! %s".center(800) %(e)

            # 拼接響應報文
            response_data = response_line + response_header + "\r\n" + response_body
            # 給客戶端發送響應報文
            client_socket.send(response_data.encode())

        else:
            # 構造請求成功響應報文
            response_line = "HTTP/1.1 200 OK\r\n"
            response_body = file_data
            response_data = (response_line + response_header + "\r\n").encode() + response_body
            # 發送響應報文
            client_socket.send(response_data)

        finally:
            # 關閉套接字
            client_socket.close()

# 創建主函數,定義套接字,設置命令行自定義端口運行
def main():
        # 判斷輸入命令參數是否符合要求
    if len(sys.argv) != 2:
        print("正確打開方式: python3 運行程序.py 端口號")
        return

    if not sys.argv[1].isdigit():
        print("正確打開方式: python3 運行程序.py 端口號")
        return

    port = int(sys.argv[1])
    http_server = HTTPServer(port)
    http_server.start()

# 程序入口
if __name__ == ‘__main__‘:
    main()

Python簡單實現Web服務器