python - 将大文件从 URL 直接流式传输到 gzip 文件

Question

我想将一个大文件直接流式传输到 gzip 文件中，而不是将其全部下载到内存中然后压缩。这就是我已经走了多远(不起作用)。我知道如何在 python 中下载文件并保存，我知道如何压缩一个文件，这是行不通的流部分。

注意:此链接的 csv 并不大，它只是一个示例网址。

    import requests
    import zlib

    url = f"http://samplecsvs.s3.amazonaws.com/Sacramentorealestatetransactions.csv"

    with requests.get(url, stream=True) as r:
        compressor = zlib.compressobj()
        with open(save_file_path, 'wb') as f:
            f.write(compressor.compress(r.raw))

Answer 1

好吧，我明白了:

with requests.get(url, stream=True, verify=False) as r:
    if save_file_path.endswith('gz'):
        compressor = zlib.compressobj(9, zlib.DEFLATED, zlib.MAX_WBITS | 16)
        with open(save_file_path, 'wb') as f:
            for chunk in r.iter_content(chunk_size=1024*1024):
                f.write(compressor.compress(chunk))
            f.write(compressor.flush())
    else:
        with open(save_file_path, 'wb') as f:
            shutil.copyfileobj(r.raw, f)