如何在python中强制http.client发送分块编码的http正文？

Question

我想发送分块的HTTP主体来测试我自己的HTTP服务器。 所以我编写了这个python代码：

import http.client

body = 'Hello World!' * 80

conn = http.client.HTTPConnection("some.domain.com")
url = "/some_path?arg=true_arg"

conn.request("POST", url, body, {"Transfer-Encoding":"chunked"})

resp = conn.getresponse()
print(resp.status, resp.reason)

我希望HTTP请求的主体是transferrd分块的， 但是我用Wireshark捕获网络包，HTTP请求的主体不被分块传输。

如何在python中通过http.client lib传输分块体？

Answer 1

好的，我明白了。

首先，编写我自己的分块编码函数。

然后使用putrequest（）、putheader（）、endheaders（）和send（）而不是request（）

import http.client

def chunk_data(data, chunk_size):
    dl = len(data)
    ret = ""
    for i in range(dl // chunk_size):
        ret += "%s\r\n" % (hex(chunk_size)[2:])
        ret += "%s\r\n\r\n" % (data[i * chunk_size : (i + 1) * chunk_size])

    if len(data) % chunk_size != 0:
        ret += "%s\r\n" % (hex(len(data) % chunk_size)[2:])
        ret += "%s\r\n" % (data[-(len(data) % chunk_size):])

    ret += "0\r\n\r\n"
    return ret


conn = http.client.HTTPConnection(host)
url = "/some_path"
conn.putrequest('POST', url)
conn.putheader('Transfer-Encoding', 'chunked')
conn.endheaders()
conn.send(chunk_data(body, size_per_chunk).encode('utf-8'))

resp = conn.getresponse()
print(resp.status, resp.reason)
conn.close()

Answer 2

我建议，如果您已经知道数据的大小，比如给定的answer中的数据大小，那么您可以设置Content-Length并一次性将其全部发送回去，这就像您对conn.send的单个调用所做的那样。

当您不知道数据有多大（例如动态生成的内容）时，分块传输编码最有用。我修改了你的代码来说明：

import httplib

def write_chunk(conn, data):
    conn.send("%s\r\n" % hex(len(data))[2:])
    conn.send("%s\r\n" % data)

def dynamically_generate_data():
    for i in range(80):
        yield "hello world"

conn = httplib.HTTPConnection("localhost")
url = "/some_path"
conn.putrequest('POST', url)
conn.putheader('Transfer-Encoding', 'chunked')
conn.endheaders()

for new_chunk in dynamically_generate_data():
    write_chunk(conn, new_chunk)
conn.send('0\r\n')

resp = conn.getresponse()
print(resp.status, resp.reason)
conn.close()