【python初级】检测文件或字节流获得图片的类型

冷不防 2022-10-12 04:13 148阅读 0赞

### 【python初级】检测文件或字节流获得图片的类型 ###

*  1、背景
 *  2、用法

# 1、背景 #

windows下，我们通过图片的后缀知道文件是什么类型的图片，  
但请注意，后缀并不能真正的代表文件的类型，比如，你可以把一个png的图片的后缀改成jpg;

关于 imghdr是什么  
它是python内置模块，用来检测图片类型；

能够支持的检测的图片类型：  
详见：imghdr.py

tests = []
    
    def test_jpeg(h, f):
        """JPEG data in JFIF or Exif format"""
        if h[6:10] in (b'JFIF', b'Exif'):
            return 'jpeg'
    
    tests.append(test_jpeg)
    
    def test_png(h, f):
        if h.startswith(b'\211PNG\r\n\032\n'):
            return 'png'
    
    tests.append(test_png)
    
    def test_gif(h, f):
        """GIF ('87 and '89 variants)"""
        if h[:6] in (b'GIF87a', b'GIF89a'):
            return 'gif'
    
    tests.append(test_gif)
    
    def test_tiff(h, f):
        """TIFF (can be in Motorola or Intel byte order)"""
        if h[:2] in (b'MM', b'II'):
            return 'tiff'
    
    tests.append(test_tiff)
    
    def test_rgb(h, f):
        """SGI image library"""
        if h.startswith(b'\001\332'):
            return 'rgb'
    
    tests.append(test_rgb)
    
    def test_pbm(h, f):
        """PBM (portable bitmap)"""
        if len(h) >= 3 and \
            h[0] == ord(b'P') and h[1] in b'14' and h[2] in b' \t\n\r':
            return 'pbm'
    
    tests.append(test_pbm)
    
    def test_pgm(h, f):
        """PGM (portable graymap)"""
        if len(h) >= 3 and \
            h[0] == ord(b'P') and h[1] in b'25' and h[2] in b' \t\n\r':
            return 'pgm'
    
    tests.append(test_pgm)
    
    def test_ppm(h, f):
        """PPM (portable pixmap)"""
        if len(h) >= 3 and \
            h[0] == ord(b'P') and h[1] in b'36' and h[2] in b' \t\n\r':
            return 'ppm'
    
    tests.append(test_ppm)
    
    def test_rast(h, f):
        """Sun raster file"""
        if h.startswith(b'\x59\xA6\x6A\x95'):
            return 'rast'
    
    tests.append(test_rast)
    
    def test_xbm(h, f):
        """X bitmap (X10 or X11)"""
        if h.startswith(b'#define '):
            return 'xbm'
    
    tests.append(test_xbm)
    
    def test_bmp(h, f):
        if h.startswith(b'BM'):
            return 'bmp'
    
    tests.append(test_bmp)
    
    def test_webp(h, f):
        if h.startswith(b'RIFF') and h[8:12] == b'WEBP':
            return 'webp'
    
    tests.append(test_webp)
    
    def test_exr(h, f):
        if h.startswith(b'\x76\x2f\x31\x01'):
            return 'exr'
    
    tests.append(test_exr)

即支持的格式如下：

<table> 
 <thead> 
  <tr> 
   <th>jpeg</th> 
   <th>JPEG date in JFIF or Exif</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td>png</td> 
   <td>Portable Network Graphics</td> 
  </tr> 
  <tr> 
   <td>gif</td> 
   <td>GIF 87a and 89a Files</td> 
  </tr> 
  <tr> 
   <td>tiff</td> 
   <td>TIFF Files</td> 
  </tr> 
  <tr> 
   <td>rgb</td> 
   <td>SGI ImgLib Files</td> 
  </tr> 
  <tr> 
   <td>pbm</td> 
   <td>Portable Bitmap Files</td> 
  </tr> 
  <tr> 
   <td>pgm</td> 
   <td>Portable Graymap Files</td> 
  </tr> 
  <tr> 
   <td>ppm</td> 
   <td>Portalble Pixmap Files</td> 
  </tr> 
  <tr> 
   <td>rast</td> 
   <td>Sun Raster Files</td> 
  </tr> 
  <tr> 
   <td>xbm</td> 
   <td>X Bitmap Files</td> 
  </tr> 
  <tr> 
   <td>bmp</td> 
   <td>BMP Files</td> 
  </tr> 
  <tr> 
   <td>webp</td> 
   <td>WebP Files</td> 
  </tr> 
  <tr> 
   <td>exr</td> 
   <td>OpenEXR Files</td> 
  </tr> 
 </tbody> 
</table>

# 2、用法 #

import imghdr
    
    # ---------- 2.1文件对象 -----------------------
    pic_type=imghdr.what("123")
    print(">>>图片类型：{0}".format(pic_type)) #jpeg
    # --------------------------------------------------
    
    # ------------- 2.2 字节流数组 -------------------
    import imghdr
    import urllib.request
    url="https://img2.baidu.com/it/u=2370411164,323094399&fm=15&fmt=auto&gp=0.jpg"
    response=urllib.request.urlopen(url)
    pic_type=imghdr.what(None,response.read())
    print(">>>图片类型：{0}".format(pic_type)) #jpeg
    # ----------------------------------------------------

**用法的说明：**

imghdr.what，这个方法最多接受两个参数，返回值是一个字符串，是上面表格中的一个，表示检测到的此图片的类型。

1个参数情况：  
当需要对文件进行检测的时候只传入1个参数即可，  
即一个文件对象，文件对象用来对本地文件做检测；

2个参数的情况：  
当需要对一个字节流检测的时候；  
第一个参数传None，第二个参数传入字节流即可(当第二个参数被指定的时候第一个参数就会被忽略掉)；