mirror of
https://github.com/python/cpython.git
synced 2025-08-04 00:48:58 +00:00
Replace duplicate code in http.server with call to http.client.parse_headers().
This commit is contained in:
parent
ef9f48e578
commit
e6fdd04b37
2 changed files with 2 additions and 17 deletions
|
@ -88,6 +88,7 @@ import io
|
|||
import os
|
||||
import sys
|
||||
import cgi
|
||||
import http.client
|
||||
import time
|
||||
import socket # For gethostbyaddr()
|
||||
import shutil
|
||||
|
@ -312,20 +313,7 @@ class BaseHTTPRequestHandler(socketserver.StreamRequestHandler):
|
|||
self.command, self.path, self.request_version = command, path, version
|
||||
|
||||
# Examine the headers and look for a Connection directive.
|
||||
|
||||
# MessageClass wants to see strings rather than bytes.
|
||||
# But a TextIOWrapper around self.rfile would buffer too many bytes
|
||||
# from the stream, bytes which we later need to read as bytes.
|
||||
# So we read the correct bytes here, as bytes, then use StringIO
|
||||
# to make them look like strings for MessageClass to parse.
|
||||
headers = []
|
||||
while True:
|
||||
line = self.rfile.readline()
|
||||
headers.append(line)
|
||||
if line in (b'\r\n', b'\n', b''):
|
||||
break
|
||||
hfile = io.StringIO(b''.join(headers).decode('iso-8859-1'))
|
||||
self.headers = email.parser.Parser(_class=self.MessageClass).parse(hfile)
|
||||
self.headers = http.client.parse_headers(self.rfile)
|
||||
|
||||
conntype = self.headers.get('Connection', "")
|
||||
if conntype.lower() == 'close':
|
||||
|
@ -524,7 +512,6 @@ class BaseHTTPRequestHandler(socketserver.StreamRequestHandler):
|
|||
protocol_version = "HTTP/1.0"
|
||||
|
||||
# MessageClass used to parse headers
|
||||
import http.client
|
||||
MessageClass = http.client.HTTPMessage
|
||||
|
||||
# Table mapping response codes to messages; entries have the
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue