5

我想发送分块的 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 请求的正文被传输分块，但是我用 Wireshark 捕获网络包，HTTP 请求的正文没有被分块传输。

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

4

2 回答 2

8

好的我明白了。

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

然后使用 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()
于 2012-02-11T07:23:20.157 回答
2

我建议，如果您已经知道数据的大小，就像给定的答案一样，您可以设置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()
于 2014-10-31T11:50:23.490 回答