图片工具

检查图片是否损坏

日常工作中，时常会需要用到图片，有时候图片在下载、解压过程中会损坏，而如果一张一张点击来检查就太不Cool了，因此我想大家都需要一个检查脚本；

测试图片，0.jpg是正常的，broke.jpg是手动删掉一点内容后异常的：

脚本运行结果：

代码如下：

	# 从本地判断图片是否损坏

	def is_valid_image(path):

		'''

		检查文件是否损坏

		'''

		try:

			bValid = True

			fileObj = open(path, 'rb')  # 以二进制形式打开

			buf = fileObj.read()

			if not buf.startswith(b'\xff\xd8'):  # 是否以\xff\xd8开头

			    bValid = False

			elif buf[6:10] in (b'JFIF', b'Exif'):  # “JFIF”的ASCII码

			    if not buf.rstrip(b'\0\r\n').endswith(b'\xff\xd9'):  # 是否以\xff\xd9结尾

			        bValid = False

			else:

			    try:

			        Image.open(fileObj).verify()

			    except Exception as e:

			        bValid = False

			        print(e)

		except Exception as e:

			return False

		return bValid

	flag1=is_valid_image(r'valid/0.jpg')

	print(flag1)

	flag1=is_valid_image(r'valid/broke.jpg')

	print(flag1)

	print ''

通过该脚本可以自动的对图片进行校验，后续是直接删除还是将正常、损坏分开就交给大家发挥啦；

图片后缀与实际类型匹配检验

我相信很多同学都有和我一样的习惯，在jpg不满足要求是，手动改为png，实际上大多数情况下，这种方式是可行的，但是在类型为gif等时，是无法直接打开的，这个需求的来源是我通过itchat做的自动微信内容备份工具

在下载聊天中的图片时，经常会下载到gif但是实际为jpg或者png的情况，这就导致这些图片无法直接展示，且需要手动改回实际类型，因此有了下面这个脚本；

类型校验转换前：

校验log：

校验及转换结果：

代码如下：

	def is_type_wrong(path):

		'''

		检查文件后缀是否与实际对应，例如实际是jpg，后缀是gif，导致打不开

		'''

		print path

		real_type = path[path.rfind('.')+1:]

		print real_type

		if path.lower().endswith('.gif') or path.lower().endswith('.jpg') or path.lower().endswith('.png'):

			header = []

			with open(path, 'rb') as f:

			    while(len(header)<5):

			        header.append(f.read(1))

			print header

			tmp = real_type

			if (header[0] == '\x47' and header[1] and '\x49' and header[2] == '\x46' and header[3] == '\x38'):

			    tmp = 'gif'

			if (header[0] == '\xff' and header[1] == '\xd8'):

			    tmp = 'jpg'

			if (header[0] == '\x89' and header[1] == '\x50' and header[2] == '\x4e' and header[3] == '\x47' and header[4] == '\x0D'):

			    tmp = 'png'

			print tmp

			if real_type != tmp:

			    return True,tmp

		return False,real_type

	if __name__ == '__main__':

		is_wrong,real_type=is_type_wrong('type/1.gif')

		if is_wrong:

			os.system('cp type/1.gif type/1.'+real_type)

		is_wrong,real_type=is_type_wrong('type/2.gif')

		if is_wrong:

			os.system('cp type/2.gif type/2.'+real_type)

		is_wrong,real_type=is_type_wrong('type/3.gif')

		if is_wrong:

			os.system('cp type/3.gif type/3.'+real_type)

		is_wrong,real_type=is_type_wrong('type/4.gif')

		if is_wrong:

			os.system('cp type/4.gif type/4.'+real_type)

通过该脚本，可以自动的对图片的后缀以及其实际类型进行校验，配合linux的cp、mv等命令很容易的实现图片类型修正的功能，还是挺有用的感觉；

小结

实际上这两个脚本的运行都是依赖于图片文件自身具备的格式，对其固定格式进行检查，实现完整性、正确性的检验，大家也可以尝试这进行手动的修改类型，比如jpg改为png，手动损坏一个图片文件，比如直接txt打开后删掉一段即可来试试看哈；

最后

大家可以到我的Github上看看有没有其他需要的东西，目前主要是自己做的机器学习项目、Python各种脚本工具、数据分析挖掘项目以及Follow的大佬、Fork的项目等：

https://github.com/NemoHoHaloAi

巴特西

检查图片是否损坏、图片后缀是否与实际图片类型对应 - Python

图片工具

检查图片是否损坏

图片后缀与实际类型匹配检验

小结

最后

最新文章

热门文章