Python判断上传文件类型

当前位置:

首页 > 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('pythontab.jpg')

栏目列表