Формат изображения на imgur

Question

Я играю в python, пытаясь загрузить некоторые изображения из imgur. Я использовал urrlib и urllib.retrieve, но вам нужно указать расширение при сохранении файла. Это не проблема для большинства сообщений, поскольку ссылка имеет, например, .jpg, но я не уверен, что делать, когда расширение не существует. Мой вопрос в том, есть ли какой-либо способ определить формат изображения файла перед его загрузкой. Вопрос в основном конкретный, но я бы не возражал против решения для большинства сайтов для хостинга изображений.

Заранее спасибо

Answer 1

Вы можете использовать imghdr.what(filename[, h]) в Python 2.7 и Python 3, чтобы определить тип изображения.

Прочтите here для получения дополнительной информации, если вы используете Python 2.7.

here Read для получения дополнительной информации, если вы не используете Python 3.

Answer 2

Предполагая, что изображение имеет никакого расширения файла, нет никакого способа, чтобы определить, какой тип это, прежде чем загрузить его. Все форматы изображений устанавливают начальные байты на определенное значение. Чтобы проверить эти «магические» начальные байты, проверьте https://github.com/ahupp/python-magic - он соответствует начальным байтам относительно известных форматов изображений.

Код ниже загружает изображение с imgur и определяет, какой тип файла он есть.

import magic 
import requests 
import shutil 

r = requests.get('http://i.imgur.com/yed5Zfk.gif', stream=True) ##Download picture 
if r.status_code == 200: 
    with open('~/Desktop/picture', 'wb') as f: 
     r.raw.decode_content = True 
     shutil.copyfileobj(r.raw, f) 

print magic.from_file('~/Desktop/picture') ##Determine type 

## Prints: 'GIF image data, version 89a, 360 x 270'