¿Convertir de ‘_io.BytesIO’ a un objeto similar a bytes en python3.6?

2 minutos de lectura

Avatar de usuario de Dan
Dan

Estoy usando esta función para descomprimir el cuerpo de una respuesta HTTP si está comprimido con gzip, compress o 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

Sin embargo, Python arroja este mensaje de error.

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

en esta línea:

return zlib.decompress(body)

Esencialmente, ¿cómo convierto de ‘_io.BytesIO’ a un objeto similar a bytes?

Es un objeto similar a un archivo. Léelos:

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

Si los datos que vienen de body es demasiado grande para leer en la memoria, querrá refactorizar su código y usar zlib.decompressobj en lugar de zlib.decompress.

  • Cuando piensas demasiado en una solución básica 😀

    – Cirilo N.

    7 de enero de 2021 a las 13:29

En caso de que escriba en el objeto primero, asegúrese de restablecer la transmisión antes de leer:

>>> b = io.BytesIO()
>>> image = PIL.Image.open(path_to_image)
>>> image.save(b, format="PNG")
>>> b.seek(0)
>>> b.read()
b'\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x06\xcf\x00\x00\x03W\x08\x02\x00'

o obtener directamente los datos con getvalue

>>> b.getvalue()
b'\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x06\xcf\x00\x00\x03W\x08\x02\x00'

b = io.BytesIO()
image = PIL.Image.open(path_to_image) # ie 'image.png'
image.save(b, format="PNG")
b.getbuffer().tobytes() # b'\x89PNG\r\n\x1a\n\x00 ...

  • Por favor agregue contexto en cuanto a por qué esta es una respuesta

    – S3DEV

    28 de marzo de 2022 a las 15:24

  • Una ventaja de esta respuesta es que no necesita restablecer la transmisión con b.seek para obtener los bytes completos que se escribieron en él.

    – Chris Collett

    28/04/2022 a las 20:20

Sé que es tarde para la respuesta, pero si alguien encuentra el mismo problema, el siguiente fragmento de código podría ser útil.

from io import BytesIO


data = BytesIO(b'Some data')

bytes_data = data.getvalue()

print(bytes_data, type(bytes_data)

Producción

b'Some data' <class 'bytes'>

¿Ha sido útil esta solución?