视频1 视频21 视频41 视频61 视频文章1 视频文章21 视频文章41 视频文章61 推荐1 推荐3 推荐5 推荐7 推荐9 推荐11 推荐13 推荐15 推荐17 推荐19 推荐21 推荐23 推荐25 推荐27 推荐29 推荐31 推荐33 推荐35 推荐37 推荐39 推荐41 推荐43 推荐45 推荐47 推荐49 关键词1 关键词101 关键词201 关键词301 关键词401 关键词501 关键词601 关键词701 关键词801 关键词901 关键词1001 关键词1101 关键词1201 关键词1301 关键词1401 关键词1501 关键词1601 关键词1701 关键词1801 关键词1901 视频扩展1 视频扩展6 视频扩展11 视频扩展16 文章1 文章201 文章401 文章601 文章801 文章1001 资讯1 资讯501 资讯1001 资讯1501 标签1 标签501 标签1001 关键词1 关键词501 关键词1001 关键词1501 专题2001
Python判断上传文件类型
2020-11-27 14:28:38 责编:小OO
文档


在开发上传服务时,经常需要对上传的文件进行过滤。

本文为大家提供了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')

常见文件格式的文件头

文件格式 文件头(十六进制)

JPEG (jpg) FFD8FF

PNG (png) 504E47

GIF (gif) 47494638

TIFF (tif) 49492A00

Windows Bitmap (bmp) 424D

CAD (dwg) 41433130

Adobe Photoshop (psd) 38425053

Rich Text Format (rtf) 7B5C727466

XML (xml) 3C3F786D6C

HTML (html) 68746D6C3E

Email [thorough only] (eml) 44656C69766572792D6174653A

Outlook Express (dbx) CFAD12FEC5FD746F

Outlook (pst) 2142444E

MS Word/Excel (xls.or.doc) D0CF11E0

MS Access (mdb) 5374616E6172204A

下载本文
显示全文
专题