使用python判断jpeg图片的完整性实例

Python  /  管理员 发布于 7年前   139

用扩展名判断文件格式非常简单，但是有可能是错误的。 jpeg文件有固定的文件头，其文件头的格式如下：

Start Marker | JFIF Marker | Header Length | Identifier 0xff, 0xd8  | 0xff, 0xe0 |  2-bytes  | "JFIF\0"

所以可以通过文件头的方式快速判断文件格式：

def is_jpg(filename):  data = open(filename,'rb').read(11)  if data[:4] != '\xff\xd8\xff\xe0' and data[:4]!='\xff\xd8\xff\xe1':     return False  if data[6:] != 'JFIF\0' and data[6:] != 'Exif\0':     return False  return True

也可以通过PIL类库来做判断：

from PIL import Imagedef is_jpg(filename):  try:    i=Image.open(filename)    return i.format =='JPEG'  except IOError:    return Fals

应用场景：判断image文件夹中的jpeg文件是否完整，代码如下：

#coding=utf-8#summary: 判断图片的有效性import ioimport os from PIL import Image#判断文件是否为有效（完整）的图片#输入参数为文件路径#会出现漏检的情况def IsValidImage(pathfile): bValid = True try:  Image.open(pathfile).verify() except:  bValid = False return bValid  def is_valid_jpg(jpg_file):   """判断JPG文件下载是否完整   """   if jpg_file.split('.')[-1].lower() == 'jpg':     with open(jpg_file, 'rb') as f:       f.seek(-2, 2)       return f.read() == '\xff\xd9' #判定jpg是否包含结束字段   else:     return True #利用PIL库进行jpeg格式判定，但有些没有结束字段的文件检测不出来def is_jpg(filename):  try:    i=Image.open(filename)    return i.format =='JPEG'  except IOError:    return False allfiles=os.listdir('image')log_file=open('img_lossinfo.txt','w')log = open('img_r.txt','w')log_w=open('img_w.txt','w')log1=open('img_jpeg.txt','w')log2=open('img_notjpg.txt','w')for i in allfiles:#if 1:if i[-4:]=='.jpg':f=os.path.join('image',i)value=IsValidImage(f)if not value:log_file.write(i+'\n')if is_valid_jpg(f):print flog.write(i+'\n')else:log_w.write(i+'\n')if is_jpg(f):log1.write(i+'\n')else:log2.write(i+'\n')

以上这篇使用python判断jpeg图片的完整性实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。