今天遇到一个小问题,如何探测图片的文件类型。总结如下:
python magicmagic 模块的github地址在这, 可以用来方便地探测文件类型,使用示例如下,可以方便地探测一个文件或是字节流文件。
>>> import magic >>> magic.from_file("testdata/test.pdf") 'PDF document, version 1.2' >>> magic.from_buffer(open("testdata/test.pdf").read(1024)) 'PDF document, version 1.2' >>> magic.from_file("testdata/test.pdf", mime=True) 'application/pdf' imghdrpython提供的标准库 imghdr 可以用来确定图片的类型,参见这里。
API如下:
简单的例子如下:
>>> import imghdr >>> imghdr.what('bass.gif') 'gif'可能的返回类型如下:
'rgb': SGI ImgLib Files
'gif': GIF 87a and 89a Files
'pbm': Portable Bitmap Files
'pgm': Portable Graymap Files
'ppm': Portable Pixmap Files
'tiff': TIFF Files
'rast': Sun Raster Files
'xbm': X Bitmap Files
'jpeg': JPEG data in JFIF or Exif formats
'bmp': BMP files
'png': Portable Network Graphics
PIL.Image使用PIL也可以检测图片类型,例子如下:
from PIL import Image img = Image.open(filename) print(img.format) # 'JPEG'而且可以用 PIL 方便地转化图片格式,它会自动检测后缀以合适的格式保存:
import Image im = Image.open('test.jpg') im.save('test.tiff') # or 'test.tif' # 或者手动指定格式 import Image im = Image.open("file.png") im.save("file.jpg", "JPEG")