import socket
import re
import multiprocessing


# Http服務器類
class HttpServer:
 # 初始化屬性
 def __init__(self):
  # 創(chuàng)建套接字
  self.tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  # 設置地址重用
  self.tcp_server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, True)
  # 綁定端口
  self.tcp_server_socket.bind(("", 8080))
  # 設置監(jiān)聽，最多建立128個連接
  self.tcp_server_socket.listen(128)

 def start(self):
  while True:
   # 接收客戶端信息
   client_socket, ip_port = self.tcp_server_socket.accept()
   print('新用戶:', ip_port, '已連接')
   # 創(chuàng)建新進程，給客戶端返回響應報文
   t1 = multiprocessing.Process(target=self.recv_msg, args=(client_socket, ip_port))
   # 啟動進程
   t1.start()
   # 關閉客戶端套接字(多進程不共享資源，故先后需要關閉兩次套接字)
   client_socket.close()

 # 發(fā)送消息方法
 @staticmethod
 def recv_msg(client_socket, ip_port):
  # 接收客戶端請求報文
  client_text = client_socket.recv(1024)
  if not client_text:
   print("客戶端", ip_port, "已下線")
   client_socket.close()
   return
  # 請求報文解碼
  client_text_str = client_text.decode()
  print(client_text_str)
  # 將請求報文切割
  client_list = client_text_str.split("\r\n")
  # 正則匹配出請求路徑
  ret = re.search(r"\s(.*)\s", client_list[0])
  print(ret.group())
  # 匹配不到則請求報文錯誤
  if not ret:
   print("用戶請求報文格式錯誤!")
   client_socket.close()
   return
  # 得到路徑
  path_info = ret.group(1)
  # 設置默認路徑
  if path_info == '/':
   path_info = '/index.html'
  print("接收到用戶請求：", path_info)
  # 響應頭
  response_header = "Server:Python-Web1.0\r\n"
  # 響應空行
  response_blank = "\r\n"
  try:
   # 讀取指定路徑文件，并且返回
   with open('web'+path_info, "rb") as file:
    # 讀取文件的二進制數(shù)據(jù)
    response_content = file.read()
  # 找不到路徑則返回404錯誤
  except Exception as e:
   # 開始拼接響應數(shù)據(jù)
   response_line = "HTTP/1.1 404 Not Found\r\n"
   response_content = "Error !!! %s" % str(e)
   response_content = response_content.encode()
  else:
   # 開始拼接響應數(shù)據(jù)
   response_line = "HTTP/1.1 200 OK\r\n"
  # 拼接響應頭
  response_data = (response_line + response_header + response_blank).encode() + response_content
  # 發(fā)送數(shù)據(jù)
  client_socket.send(response_data)
  # 關閉socket
  client_socket.close()


if __name__ == '__main__':
 http1 = HttpServer()
 http1.start()