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'>