Как мне преобразовать из _io.BytesIO в байтовоподобный объект в python3.6?

Question

Я использую эту функцию для распаковки тела или HTTP-ответа, если он сжат с помощью gzip, compress или deflate.

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.