Python TPCServer rfile.read bloki

Question

Piszę prostą obsługę żądań SocketServer.TCPServer (StreamRequestHandler), która przechwyci żądanie, wraz z nagłówkami i treścią wiadomości. To służy do fałszowania serwera HTTP, który możemy wykorzystać do testowania.

Nie mam problemu z przechwytywaniem linii lub nagłówków.

Jeśli spróbuję pobrać więcej z rfile niż istnieje, kod blokuje się. Jak mogę pobrać całą treść żądania, nie znając jej rozmiaru? Innymi słowy, nie mam nagłówka Content-Size.

Oto fragment tego, co mam teraz:

def _read_request_line(self): 
    server.request_line = self.rfile.readline().rstrip('\r\n') 

def _read_headers(self): 
    headers = [] 
    for line in self.rfile: 
     line = line.rstrip('\r\n') 
     if not line: 
      break 
     parts = line.split(':', 1) 
     header = (parts[0].strip(), parts[0].strip()) 
     headers.append(header) 
    server.request_headers = headers 

def _read_content(self): 
    server.request_content = self.rfile.read() # blocks 

Answer 1

komentarz Keitha jest poprawna. Oto jak to wygląda

 length = int(self.headers.getheader('content-length')) 
    data = self.rfile.read(length)