Как проверить, является ли файл действительным файлом изображения?

Часто первая пара символов будет магическим числом для различных форматов файлов. Вы можете проверить это в дополнение к проверке исключений выше.

Я только что нашел встроенный модуль imghdr. Из документации Python:

Модуль imghdr определяет тип изображения, содержащегося в файле или потоке байтов.

Вот как это работает:

>>> import imghdr
>>> imghdr.what('/tmp/bass')
'gif'

Использование модуля намного лучше, чем повторная реализация аналогичной функциональности

В дополнение к тому, что предлагает Брайан, вы можете использовать метод PIL verify. чтобы проверить, не поврежден ли файл.

im.verify ()

Пытается определить, поврежден ли файл, без фактического декодирования данных изображения. Если этот метод обнаруживает какие-либо проблемы, он вызывает соответствующие исключения. Этот метод работает только с вновь открытым изображением; если изображение уже было загружено, результат не определен. Кроме того, если вам нужно загрузить изображение после использования этого метода, вы должны повторно открыть файл изображения. Атрибуты

В дополнение к проверке PIL изображений вы также можете добавить проверку расширения имени файла следующим образом:

filename.lower().endswith(('.png', '.jpg', '.jpeg', '.tiff', '.bmp', '.gif'))

Обратите внимание, что это только проверяет, имеет ли имя файла допустимое расширение изображения, на самом деле оно не открывает изображение, чтобы увидеть, является ли оно действительным изображением, поэтому вам необходимо дополнительно использовать PIL или одну из библиотек, предложенных в других ответах.

Обновить

Я также реализовал следующее решение в моем скрипте Python, здесь, на GitHub.

Я также подтвердил, что поврежденные файлы (jpg) часто не являются «битыми» изображениями, то есть поврежденный файл изображения иногда остается законным файлом изображения, исходное изображение потеряно или изменено, но вы все равно можете загрузить его без ошибок. Но усечение файла всегда вызывает ошибки.

Конец обновления

Вы можете использовать модуль Python Pillow (PIL) с большинством форматов изображений, чтобы проверить, является ли файл действительным и неповрежденным файлом изображения.

В случае, если вы стремитесь обнаружить также битые изображения, @Nadia Alramli правильно предлагает метод im.verify(), но этот не обнаруживает всех возможных дефектов изображения, например, im.verify не обнаруживает усеченные изображения (что большинство зрителей часто загружают с серым участком).

Pillow также может обнаруживать дефекты этого типа, но вы должны применить манипуляции с изображением или декодировать / перекодировать изображение или запустить проверку. Напоследок предлагаю использовать этот код:

try:
  im = Image.load(filename)
  im.verify() #I perform also verify, don't know if he sees other types o defects
  im.close() #reload is necessary in my case
  im = Image.load(filename) 
  im.transpose(PIL.Image.FLIP_LEFT_RIGHT)
  im.close()
except: 
  #manage excetions here

В случае дефектов изображения этот код вызовет исключение. Учтите, что im.verify примерно в 100 раз быстрее, чем обработка изображений (и я думаю, что переворот - одно из самых дешевых преобразований). С помощью этого кода вы собираетесь проверить набор изображений со скоростью около 10 МБ / с со стандартной подушкой или 40 МБ / с с модулем Pillow-SIMD (современный процессор x86_64 2,5 ГГц).

Для других форматов psd, xcf, .. вы можете использовать Imagemagick оболочку Wand, код выглядит следующим образом :

im = wand.image.Image(filename=filename)
temp = im.flip;
im.close()

Но, судя по моим экспериментам, Wand не обнаруживает усеченные изображения, я думаю, что он загружает недостающие части в виде серой области без запроса.

Я красный, что Imagemagick имеет внешнюю команду identify, которая может выполнить задание, но я не нашел способа вызвать эту функцию программно, и я не тестировали этот маршрут.

Я предлагаю всегда выполнять предварительную проверку, проверять, чтобы размер файла не был нулевым (или очень маленьким), это очень дешевая идея:

statfile = os.stat(filename)
filesize = statfile.st_size
if filesize == 0:
  #manage here the 'faulty image' case

Один из вариантов - использовать пакет filetype.

Установка

python -m pip install filetype

Преимущества

1. Быстро: работает, загружая только первые несколько байтов вашего изображения (проверьте магическое число)
2. Поддерживает различные типы пантомимы: изображения, видео, шрифты, аудио, архивы.

Пример

filetype ›= 1.0.7

import filetype

filename = "/path/to/file.jpg"

if filetype.is_image(filename):
    print(f"{filename} is a valid image...")
elif filetype.is_video(filename):
    print(f"{filename} is a valid video...")

filetype ‹= 1.0.6

import filetype

filename = "/path/to/file.jpg"

if filetype.image(filename):
    print(f"{filename} is a valid image...")
elif filetype.video(filename):
    print(f"{filename} is a valid video...")

Дополнительная информация об официальном репо: https://github.com/h2non/filetype.py

В Linux вы можете использовать python-magic, который использует libmagic для определения форматов файлов.

AFAIK, libmagic изучает файл и пытается рассказать вам больше, чем просто формат, например размеры растрового изображения, версию формата и т. Д. Так что вы можете рассматривать это как поверхностный тест на достоверность.

Для других определений допустимости вам, возможно, придется написать свои собственные тесты.

Вы можете использовать привязки Python к libmagic, python-magic, а затем проверить типы mime. . Это не скажет вам, повреждены ли файлы или нет, но он сможет определить, какой это тип изображения.