图片工具
检查图片是否损坏
日常工作中，时常会需要用到图片，有时候图片在下载、解压过程中会损坏，而如果一张一张点击来检查就太不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