Python中如何检测文件类型，python-magic类库的使用-CFANZ编程社区

Python-magic是一个Python的第三方库，用于检测和识别文件类型。它使用libmagic库，可以根据文件的内容确定文件的类型，而不是根据文件的扩展名来确定文件类型。在这篇教程中，我们将学习如何使用Python-magic库来检测文件类型。

安装Python-magic

在使用Python-magic之前，需要先安装它。可以使用pip工具来安装它，使用以下命令：

pip install python-magic

导入Python-magic

安装完成后，需要导入Python-magic模块。可以使用以下代码导入：

import magic

检测文件类型

要检测文件类型，首先需要创建一个Magic对象，可以使用以下代码创建：

magic_obj = magic.Magic()

创建Magic对象时，可以传递参数来指定不同的检测模式，例如：

magic_obj = magic.Magic(mime=True)

上面的代码将创建一个Magic对象，它将返回文件的MIME类型。

现在，我们可以使用Magic对象来检测文件类型，可以使用以下代码：

file_type = magic_obj.from_file('example.txt')
print(file_type)

上面的代码将检测名为example.txt的文件的类型，并打印输出。

检测文件内容

有时，我们可能需要检测文件的内容而不是文件的名称。可以使用以下代码检测文件内容：

with open('example.txt', 'rb') as f:
    file_content = f.read()
    file_type = magic_obj.from_buffer(file_content)
    print(file_type)

上面的代码将打开名为example.txt的文件，并读取文件内容。然后，它使用Magic对象来检测文件内容的类型，并打印输出。

检测目录中的所有文件类型

有时，我们可能需要检测目录中所有文件的类型。可以使用以下代码检测目录中的所有文件类型：

import os

directory = '/path/to/directory'

for filename in os.listdir(directory):
    file_path = os.path.join(directory, filename)
    if os.path.isfile(file_path):
        with open(file_path, 'rb') as f:
            file_content = f.read()
            file_type = magic_obj.from_buffer(file_content)
            print(f"{filename}: {file_type}")

上面的代码将遍历目录中的所有文件，并使用Magic对象检测每个文件的类型，并打印输出。