Як перевірити, чи файл є дійсним файлом зображення?

Зараз я використовую PIL.

from PIL import Image
try:
    im=Image.open(filename)
    # do stuff
except IOError:
    # filename not an image file

Однак, хоча це достатньо охоплює більшість випадків, деякі файли зображень, такі як xcf, svg та psd, не виявляються. Файли Psd видає виняток OverflowError.

Чи є якимось чином і я міг би їх включити?

Багато разів перші пара символів будуть чарівним числом для різних форматів файлів. Ви можете перевірити це на додаток до перевірки винятків вище.

Я щойно знайшов вбудований модуль imghdr . З документації python:

Модуль imghdr визначає тип зображення, що міститься у файлі чи байтовому потоці.

Ось як це працює:

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

Використання модуля набагато краще, ніж повторне виконання аналогічних функцій

На додаток до того, що пропонує Брайан, ви можете використовувати метод підтвердження PIL, щоб перевірити, чи файл порушений.

ім.перевірити ()

Спроби визначити, чи файл порушений, не фактично розшифровуючи дані зображення. Якщо цей метод виявить якісь проблеми, він може спричинити винятки. Цей метод працює лише на щойно відкрите зображення; якщо зображення вже завантажено, результат не визначений. Крім того, якщо вам потрібно завантажити зображення після використання цього методу, потрібно знову відкрити файл зображення. Атрибути

Додатково до PILперевірки зображення ви також можете додати перевірку розширення імені файлу таким чином:

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

Зауважте, що це перевіряє, чи має ім'я файлу дійсне розширення зображення, воно насправді не відкриває зображення, щоб побачити, чи це дійсне зображення, тому вам потрібно додатково використовувати PILабо одну з бібліотек, запропоновану в інших відповідях.

Оновлення

Я також реалізував таке рішення в своєму скрипті Python тут, на GitHub .

Я також переконався, що пошкоджені файли (jpg) часто не є "зламаними" зображеннями, тобто пошкоджений файл зображення іноді залишається законним файлом зображення, оригінальне зображення втрачається або змінюється, але ви все ще можете завантажити його без помилок. Але, усічення файлів викликає завжди помилки.

Закінчити оновлення

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

У випадку, якщо ви прагнете виявити також пошкоджені зображення, @Nadia Alramli правильно пропонує im.verify()метод, але це не виявляє всіх можливих дефектів зображення , наприклад, im.verifyне виявляє усічені зображення (що більшість глядачів часто завантажують із сірою областю).

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

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 (сучасний процесор 2,5 ГГц x86_64).

Для інших форматів psd , xcf , .. ви можете використовувати Imagemagick обгортку Wand , код такий:

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

Але, з моїх експериментів, Wand не виявляє усічені зображення, я думаю, що він завантажує відсутні частини, як сіру область, не вимагаючи.

Я помічаю, що Imagemagick має зовнішню ідентифікацію команди, яка могла б виконати роботу, але я не знайшов способу викликати цю функцію програмно, і я не перевіряв цей маршрут.

Я пропоную завжди проводити попередню перевірку, перевіряти розмір файлів, щоб він не дорівнював нулю (або дуже малий), дуже дешева ідея:

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

В Linux ви можете використовувати python-magic ( http://pypi.python.org/pypi/python-magic/0.1 ), який використовує libmagic для ідентифікації форматів файлів.

AFAIK, libmagic заглядає у файл і намагається розповісти вам більше про нього, а не лише про формат, як, наприклад, розміри растрових зображень, версію формату тощо. Отже, ви можете бачити це як поверхневий тест на "дійсність".

Для інших визначень "дійсних" можливо вам доведеться написати власні тести.

Ви можете використати прив'язки Python до лібмагічних, пітон-магічних, а потім перевірити типи mime. Це не скаже вам, чи файли пошкоджені чи неушкоджені, але він повинен мати можливість визначати, який тип зображення це.

Ну, я не знаю про внутрішню частину PSD, але я, звичайно, знаю, що фактично svg не є файлом зображення, - він заснований на xml, так що він, по суті, є звичайний текстовий файл.

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

Установка

python -m pip install filetype

Переваги

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

Приклад рішення

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

Чи перевіряє розширення файлів прийнятне чи ви намагаєтеся підтвердити, що дані представляють файл зображення?

Якщо ви можете перевірити розширення файлу, звичайний вираз або просте порівняння можуть задовольнити вимогу.

format = [".jpg",".png",".jpeg"]
 for (path,dirs,files) in os.walk(path):
     for file in files:
         if file.endswith(tuple(format)):
             print(path)
             print ("Valid",file)
         else:
             print(path)
             print("InValid",file)