Python利用pillow判断图片完整性

Matthewl

#encoding: utf-8　　
#author: walker
　　
#date: 2016-07-26
　　
#summary: 判断图片的有效性
　　

　　
import io
　　
import imghdr
　　
from os import PathLike
　　
from PIL import Image
　　

　　

　　
#判断文件是否为有效（完整）的图片
　　
#输入参数为文件路径，或文件对象
　　
def IsValidImage(file):
　　
    bValid = True
　　
    if isinstance(file, (str, PathLike)):
　　
        fileObj = open(file, 'rb')
　　
    else:
　　
        fileObj = file
　　

　　
    buf = fileObj.read()
　　
    if buf[6:10] in (b'JFIF', b'Exif'):     #jpg图片
　　
        if not buf.rstrip(b'\0\r\n').endswith(b'\xff\xd9'):
　　
            bValid = False
　　
    else:
　　
        try:
　　
            Image.open(fileObj).verify()
　　
        except:
　　
            bValid = False
　　

　　
    return bValid
　　

　　

　　
#判断文件是否为有效（完整）的图片
　　
#输入参数为bytes，如网络请求返回的二进制数据
　　
def IsValidImage4Bytes(buf):
　　
    bValid = True
　　
    try:
　　
        Image.open(io.BytesIO(buf)).verify()
　　
    except:
　　
        bValid = False
　　

　　
    return bValid
　　

　　

　　
#判断文件是否为有效（完整）的图片
　　
#输入参数为bytes，如网络请求返回的二进制数据
　　
def IsValidImage4Bytes(buf):
　　
    bValid = True
　　
    if buf[6:10] in (b'JFIF', b'Exif'):     #jpg图片
　　
        if not buf.rstrip(b'\0\r\n').endswith(b'\xff\xd9'):
　　
            bValid = False
　　
    else:
　　
        try:
　　
            Image.open(io.BytesIO(buf)).verify()
　　
        except:
　　
            bValid = False
　　

　　
    return bValid