python - 从 '_io.BytesIO' 转换为 python3.6 中的类字节对象？

Question

如果使用 gzip、compress 或 deflate 压缩 HTTP 响应主体，我将使用此函数对其进行解压缩。

def uncompress_body(self, compression_type, body):
    if compression_type == 'gzip' or compression_type == 'compress':
        return zlib.decompress(body)
    elif compression_type == 'deflate':
        compressor = zlib.compressobj(9, zlib.DEFLATED, -zlib.MAX_WBITS)
        compressed = compressor.compress(body)
        compressed += compressor.flush()
        return base64.b64encode(compressed)

    return body

但是 python 会抛出此错误消息。

TypeError: a bytes-like object is required, not '_io.BytesIO'

在这一行:

return zlib.decompress(body)

本质上，我如何从“_io.BytesIO”转换为类似字节的对象？

Answer 1

这是一个类似文件的对象。阅读它们:

>>> b = io.BytesIO(b'hello')
>>> b.read()
b'hello'

如果来自body 的数据太大而无法读入内存，您需要重构代码并使用zlib.decompressobj。而不是 zlib.decompress。