Python判断上传文件类型

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

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‘)