python通过文件头判断文件类型

下面是Python通过文件头判断文件类型的完整实例教程。

1. 什么是文件头

文件头（File Header）指的是文件的开头几个字节，包含了文件的一些基本信息。不同类型的文件，在文件头中包含的信息不同，因此可以通过读取文件头来判断文件的类型。

2. 用Python判断文件类型的方法

Python中可以通过读取文件头来判断文件类型，具体方法如下：

import os

def get_file_type(file_path):
    with open(file_path, "rb") as f:
        # 读取文件头的前4个字节
        head = f.read(4)
        if head[:3] == b"\xFF\xD8\xFF":      # JPEG/jpg文件
            return "JPEG"
        elif head[:4] == b"\x89\x50\x4E\x47": # PNG文件
            return "PNG"
        elif head[:2] == b"\x4D\x5A":        # EXE文件
            return "EXE"
        elif head[:2] == b"\xFF\x4F":        # JPG文件
            return "JPG"
        elif head[:2] == b"\x46\x4C":        # FLV文件
            return "FLV"
        else:
            return "unknown"

以上代码实现了通过读取文件头来判断文件类型的功能。这里通过open函数以二进制模式打开文件，然后读取文件头的前4个字节，根据不同的文件类型的文件头来判断文件类型。

3. 示例说明

下面给出两个示例说明。

示例1

假如我们有一张图片文件，要判断它的文件类型。文件名为test.jpg，文件的路径为/home/user/test.jpg，使用以上代码可以得到如下结果：

file_path = "/home/user/test.jpg"
file_type = get_file_type(file_path)
print(file_type)

输出结果为：

JPEG

这说明test.jpg是一张JPEG图片文件。

示例2

假如我们有一个音频文件，要判断它的文件类型。文件名为test.mp3，文件的路径为/home/user/test.mp3，使用以上代码可以得到如下结果：

file_path = "/home/user/test.mp3"
file_type = get_file_type(file_path)
print(file_type)

输出结果为：

unknown

这说明test.mp3的文件类型无法通过当前代码判断出来，可能需要增加对应的判断条件来支持MP3文件的判断。