Python利用pillow判断图片完整性

pip install pillow

#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