Phát hiện xem tệp có phải là hình ảnh trong Python

Question

Có cách nào chung để phát hiện xem tệp có là hình ảnh hay không (jpg, bmp, png, v.v.)

Hoặc đang tạo danh sách các phần mở rộng tệp và làm một so sánh từng người một theo cách duy nhất?

Answer 1

Bạn nên sử dụng thư viện cho việc này. Lưu ý rằng phần mở rộng! = Loại tệp, vì bạn có thể thay đổi phần mở rộng thành tệp .jpg, mở nó bằng sơn và sơn sẽ diễn giải nó như một jpeg (ví dụ). Bạn nên kiểm tra How to find the mime type of a file in python?.

Answer 2

Giả sử:

>>> files = {"a_movie.mkv", "an_image.png", "a_movie_without_extension", "an_image_without_extension"} 

Và họ là bộ phim và hình ảnh thích hợp tập tin trong thư mục script.

Bạn có thể sử dụng mô-đun mimetypes dựng sẵn nhưng sẽ không hoạt động nếu không có tiện ích mở rộng.

>>> import mimetypes 
>>> {file: mimetypes.guess_type(file) for file in files} 
{'a_movie_without_extension': (None, None), 'an_image.png': ('image/png', None), 'an_image_without_extension': (None, None), 'a_movie.mkv': (None, None)} 

Hoặc gọi lệnh unix file. Này hoạt động mà không cần mở rộng, nhưng không phải trong Windows:

>>> import subprocess 
>>> def find_mime_with_file(path): 
...  command = "/usr/bin/file -i {0}".format(path) 
...  return subprocess.Popen(command, shell=True, stdout=subprocess.PIPE).communicate()[0].split()[1] 
... 
>>> {file: find_mime_with_file(file) for file in files} 
{'a_movie_without_extension': 'application/octet-stream;', 'an_image.png': 'image/png;', 'an_image_without_extension': 'image/png;', 'a_movie.mkv': 'application/octet-stream;'} 

Hoặc bạn cố gắng để mở nó với PIL, và kiểm tra lỗi, nhưng cần PIL cài đặt:

>>> from PIL import Image 
>>> def check_image_with_pil(path): 
...  try: 
...   Image.open(path) 
...  except IOError: 
...   return False 
...  return True 
... 
>>> {file: check_image_with_pil(file) for file in files} 
{'a_movie_without_extension': False, 'an_image.png': True, 'an_image_without_extension': True, 'a_movie.mkv': False} 

Hoặc, vì đơn giản, như bạn nói rằng, chỉ cần kiểm tra tiện ích mở rộng, đó là cách tốt nhất tôi nghĩ.

>>> extensions = {".jpg", ".png", ".gif"} #etc 
>>> {file: any(file.endswith(ext) for ext in extensions) for file in files} 
{'a_movie_without_extension': False, 'an_image.png': True, 'an_image_without_extension': False, 'a_movie.mkv': False}