python准确判断文件类型

#! /usr/bin/python

# pythontab提醒您注意中文编码问题，指定编码为utf-8

# -*- coding: utf-8 -*-

importstruct

# 支持文件类型

# 用16进制字符串的目的是可以知道文件头是多少字节

# 各种文件头的长度不一样，少则2字符，长则8字符

deftypeList():

return{

"FFD8FF":"JPEG",

"89504E47":"PNG"}

# 字节码转16进制字符串

defbytes2hex(bytes):

num=len(bytes)

hexstr=u""

foriinrange(num):

t=u"%x"%bytes[i]

iflen(t)%2:

hexstr+=u"0"

hexstr+=t

returnhexstr.upper()

# 获取文件类型

deffiletype(filename):

binfile=open(filename,'rb')# 必需二制字读取

tl=typeList()

ftype='unknown'

forhcodeintl.keys():

numOfBytes=len(hcode)/2# 需要读多少字节

binfile.seek(0)# 每次读取都要回到文件头，不然会一直往后读取

hbytes=struct.unpack_from("B"*numOfBytes, binfile.read(numOfBytes))# 一个 "B"表示一个字节

f_hcode=bytes2hex(hbytes)

iff_hcode==hcode:

ftype=tl[hcode]

break

binfile.close()

returnftype

if__name__=='__main__':

printfiletype('./test.jpg')