python判断文件类型_Python判断上传文件类型的简单示例

对python这个高级语言感兴趣的小伙伴，下面一起跟随编程之家 jb51.cc的小编两巴掌来看看吧！

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

代码如下

# @param Python判断上传文件类型

# @author 编程之家 jb51.cc|www.jb51.cc

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('512pic.jpg')

# End www.jb51.cc

常见文件格式的文件头

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

JPEG (jpg) FFD8FF

PNG (png) 89504E47

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) 44656C69766572792D646174653A

Outlook Express (dbx) CFAD12FEC5FD746F

Outlook (pst) 2142444E

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

MS Access (mdb) 5374616E64617264204A

总结

如果觉得编程之家网站内容还不错，欢迎将编程之家网站推荐给程序员好友。

本图文内容来源于网友网络收集整理提供，作为学习参考使用，版权属于原作者。

如您喜欢交流学习经验，点击链接加入交流1群：1065694478(已满)交流2群：163560250