pyhton面试题:写一个函数根据文件的开头8个字节的信息,判断该文件是不是png文件

假设公司有一批图片文件,不小心被管理人员把扩展名都去掉了。

这批图片文件中有的是png文件,有的是jpg文件。

png文件的开头一定是 这样的 89 50 4e 47 0d 0a 1a 0a 8个字节

现在要求你写一个函数,参数是图片文件的路径,函数根据文件的开头8个字节的信息,判断该文件是不是png文件。

如果是,打印出 png, 否则打印出 jpg。

import binascii
import os
def isPng(path):
with open(path, 'rb') as f:
a = f.read()
hexstr = binascii.b2a_hex(a[0:8])
if hexstr == "89504e470d0a1a0a":
return os.rename(path,f"{path}.png")
else:
return os.rename(path,f"{path}.jpg")
posted @ 2020-08-08 15:38  小刚blogs  阅读(406)  评论(0)    收藏  举报