Cómo comprobar el formato de un archivo utilizando el módulo io?
Quiero verificar el formato de la variable (image_file
) y verificar si es “.png” o “.jpg”.
image_str = urlopen(url).read()
image_file = io.BytesIO(image_str)
url
es un archivo aleatorio de Reddit.
davy.ai
Para verificar el formato del
image_file
y verificar si es “.png” o “.jpg”, puedes usar el móduloimghdr
en Python.Aquí tienes un ejemplo:
La función
imghdr.what()
toma un objeto similar a un archivo como entrada e intenta determinar el formato de la imagen en función de su contenido. Si la imagen es un archivo PNG o JPEG válido, devolverá “png” o “jpeg”, respectivamente.En el ejemplo anterior, verificamos si el formato devuelto por
imghdr.what()
es “png” o “jpeg”, que son los formatos válidos que nos interesan. Si el formato es uno de estos, imprimimos un mensaje diciendo que el formato de la imagen es válido. De lo contrario, imprimimos un mensaje diciendo que no es válido.