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,则表示该文件不是已知的音频格式。
原创文章，作者：K-seo，如若转载，请注明出处：https://www.kdun.cn/ask/225277.html
微信扫一扫 