python判断文件类型的函数

Python怎么判断文件类型

在日常使用中，我们经常需要判断一个文件的类型，例如图片、视频、音频等，Python提供了多种方法来实现这个功能，本文将介绍两种常用的方法：通过文件扩展名判断和通过文件内容判断。

通过文件扩展名判断

1、获取文件扩展名

要判断一个文件的类型，首先需要获取其扩展名，在Python中，可以使用os.path模块的splitext()函数来获取文件的扩展名。

import os
def get_file_extension(file_path):
    file_name, file_extension = os.path.splitext(file_path)
    return file_extension[1:]

2、根据扩展名判断文件类型

获取到文件的扩展名后，可以通过预先定义好的字典来匹配对应的文件类型，以下是一个简单的示例：

def get_file_type(file_extension):
    file_types = {
        '.jpg': '图片',
        '.png': '图片',
        '.jpeg': '图片',
        '.gif': '图片',
        '.bmp': '图片',
        '.mp4': '视频',
        '.avi': '视频',
        '.mkv': '视频',
        '.flv': '视频',
        '.mov': '视频',
        '.wmv': '视频',
        '.mp3': '音频',
        '.wav': '音频',
        '.flac': '音频',
    }
    return file_types.get(file_extension, '其他')

通过文件内容判断

1、读取文件前4个字节

有些文件类型的特征是固定的，例如JPEG图片的前4个字节为FFD8FF，我们可以先读取文件的前4个字节，然后根据这些字节来判断文件类型，以下是一个简单的示例：

def get_file_type_by_content(file_path):
    with open(file_path, 'rb') as f:
        header = f.read(4)
        if header == b'\xff\xd8xff\xe0':
            return 'JPEG图片'
        elif header == b'RIFF':
            return 'WAV音频'
         其他文件类型的判断逻辑可以类似地添加在这里
        else:
            return '未知文件类型'

2、实现更复杂的文件类型判断逻辑

如果需要支持更多的文件类型，可以在上述示例的基础上进行扩展，可以定义一个包含各种音频编码格式的字典，然后根据文件头中的信息来判断具体的音频格式，以下是一个简化的示例：

def get_file_type_by_content(file_path):
    with open(file_path, 'rb') as f:
        header = f.read(4)
        if header == b'RIFF':
            f.seek(4)   跳过WAVE标识符和块大小字段
            chunk_id = f.read(4)   RIFF块ID('WAVE')或fmt块ID('fmt ')或data块ID('data')或其他非标准块ID('?')
            if chunk_id == b'WAVE':   WAVE文件头开始于“WAVE”标识符之后的4字节位置，接下来的数据块以“fmt ”开头，每个数据块都以“data”开头，其他块可能具有不同的格式，我们需要检查下一个块是否为“fmt ”，如果是，则继续查找“fmt ”块以确定音频编码格式，否则，表示该文件不是WAVE格式，对于其他非标准块ID,可以根据实际情况进行类似的处理，如果没有找到匹配的块ID,则表示该文件不是已知的音频格式。