Python簡單實現Web服務器
阿新 • • 發佈:2018-04-08
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服務器