python通过文件头判断文件类型
发布时间：2020-05-23 21:10:27 所属栏目：Python 来源：互联网
导读：对于提供上传的服务器，需要对上传的文件进行过滤。本文为大家提供了python通过文件头判断文件类型的方法，避免不必要的麻烦。
|
对于提供上传的服务器，需要对上传的文件进行过滤。 本文为大家提供了python通过文件头判断文件类型的方法，避免不必要的麻烦。 分享代码如下
import struct
# 支持文件类型
# 用16进制字符串的目的是可以知道文件头是多少字节
# 各种文件头的长度不一样，少半2字符，长则8字符
def typeList():
return {
"52617221": EXT_RAR,"504B0304": EXT_ZIP}
# 字节码转16进制字符串
def bytes2hex(bytes):
num = len(bytes)
hexstr = u""
for i in range(num):
t = u"%x" % bytes[i]
if len(t) % 2:
hexstr += u"0"
hexstr += t
return hexstr.upper()
# 获取文件类型
def filetype(filename):
binfile = open(filename,'rb') # 必需二制字读取
tl = typeList()
ftype = 'unknown'
for hcode in tl.keys():
numOfBytes = len(hcode) / 2 # 需要读多少字节
binfile.seek(0) # 每次读取都要回到文件头，不然会一直往后读取
hbytes = struct.unpack_from("B"*numOfBytes,binfile.read(numOfBytes)) # 一个 "B"表示一个字节
f_hcode = bytes2hex(hbytes)
if f_hcode == hcode:
ftype = tl[hcode]
break
binfile.close()
return ftype
if __name__ == '__main__':
print filetype(Your-file-path)
常见文件格式的文件头 文件格式 文件头(十六进制) 以上就是本文的全部内容，希望对大家的学习有所帮助。 （编辑：安卓应用网） 【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容! |