1
2
3
4
5

import re

file_name = "C:\\Users\Dell\\Desktop\\11111111111\\test.php.txt.asp"
fstr = re.search(r'.*\.(.*)$', file_name, re.I).group(1)
print(fstr)	# "asp"

1
2
3
4
5

import os

file_name = "/home/test/test.json.mp3.jsp"
fstr = os.path.basename(file_name).split('.')[-1].lower()
print(fstr)	# "jsp"

1

pip install filetype

1
2
3
4
5
6
7
8
9

# Linux下安装:
pip install python-magic

# OSX下安装(需要先安装libmagic依赖项):
brew install libmagic
pip install python-magic

# Windows下安装:
pip install python-magic-bin 0.4.14

  1
  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
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101

# coding = utf-8
# 适用于windows和linux系统环境
# 检测结果：magic库更可靠更详细，filetype会出现误检测和无结果。
# 可以检测出视频、音频、图片、压缩文件、PDF文件等被修改后缀的原格式。
# 注意：Windows下检测路径末尾可以不加'\'，但是Linux下需要在末尾加'/'否则会报错

import os

import filetype
import magic


def fdirTraverser(file_dir):
    """
    简单的文件目录遍历器
    file_dir：待检测目录地址
    """
    for root, dirs, files in os.walk(file_dir):
        print("当前目录路径：%s\n" % root)
        print("当前路径下所有子目录：%s\n" % dirs)
        print("当前路径下所有非目录子文件：%s\n" % files)


def getSpecFile(file_dir, file_extension):
    """
    获取当前路径下除子目录外所有指定后缀的文件名列表（绝对路径）
    file_dir：待检测目录地址
    file_extension：指定文件后缀名
    """
    aa = []
    for root, dirs, files in os.walk(file_dir):
        for file in files:
            # os.path.splitext()函数将路径拆分为文件名+扩展名
            if os.path.splitext(file)[1] == file_extension:
                aa.append(os.path.join(root, file))
        return aa


def getAllFile(path, list_name):
    """
    递归获取当前路径下所有文件名（绝对路径，包括所有子目录下的文件）
    path: 待检测目录地址
    list_name: 存储主目录下所有文件名称的列表（绝对路径）
    """

    b = os.listdir(path)
    # print(b)
    for file in b:
        file_path = os.path.join(path, file)
        if os.path.isdir(file_path):
            getAllFile(file_path, list_name)
        else:
            list_name.append(file_path)


def ck_ftype(path_list):
    """使用python的ftype和magic库对比判断文件类型"""
    for i in path_list:
        print("++++*******************START********************++++")
        print("待检测的文件为：%s\n" % i)
        # 使用filetype库
        kind = filetype.guess(i)
        print("[+]--------filetype--------[+]")
        print("filetype库检测的文件类型：")
        if kind is None:
            print('[-] 使用filetype库未检测出文件类型！\n')
        else:
            print('文件扩展名: %s，文件MIME类型：%s' % (kind.extension, kind.mime))
            print("[+]------------------------[+]\n")

        # 使用magic库
        # bb = magic.from_file(i)
        # cc = magic.from_file(i, mime=True)
        bb = magic.from_buffer(open(i, "rb").read(2048))
        cc = magic.from_buffer(open(i, "rb").read(2048), mime=True)
        print("[+]------------magic------------[+]")
        print("magic库检测的文件类型：%s，MIME类型：%s" % (bb, cc))
        print("[+]-----------------------------[+]\n")
        print("++++******************END*********************++++\n\n")


if __name__ == '__main__':
    # 支持linux和windows
    # path = "/home/mycode/test_files/"
    path = input("请输入要查询的文件路径：")

    # 简单的文件、目录遍历
    fdirTraverser(path)

    # 获取当前路径下除子目录外所有指定后缀的文件名列表（绝对路径）
    file_extension = ".png"
    file_list = getSpecFile(path, file_extension)
    print("file_list：%s\n" % file_list)

    # 递归获取当前路径下所有文件名（绝对路径，包括所有子目录下的文件）
    aa = []
    getAllFile(path, aa)
    print("待检测所有文件列表: %s\n" % aa)

    # 使用python的ftype和magic库对比判断文件类型
    ck_ftype(aa)