📜  py3 识别文件扩展名 (1)

📅  最后修改于: 2023-12-03 15:03:54.077000             🧑  作者: Mango

Py3 识别文件扩展名

在编程中,我们往往需要对不同的文件类型进行处理,所以需要识别文件的扩展名。在Python中,识别文件扩展名可以通过以下两种方式实现。

使用os.path.splitext方法

Python标准库中的os模块提供了一种获取文件扩展名的方法,即os.path.splitext()方法。该方法接收一个文件路径作为参数,返回一个包含文件名和扩展名的元组。

下面是一个示例代码,演示了如何使用os.path.splitext()方法获取文件扩展名:

import os

filename = 'example.txt'
name, ext = os.path.splitext(filename)
print(name)  # 输出 example
print(ext)   # 输出 .txt
使用第三方库mimetypes

Python的标准库中还有另外一个模块叫做mimetypes,该模块可以让我们通过文件名或扩展名获取MIME类型。

下面是一个示例代码,演示了如何使用mimetypes模块获取文件扩展名:

import mimetypes

filename = 'example.txt'
mime_type, encoding = mimetypes.guess_type(filename)
print(mime_type)  # 输出 text/plain
print(encoding)   # 输出 None

除了使用guess_type()方法,mimetypes模块还提供了其他方法用于获取MIME类型。

以上就是Python中识别文件扩展名的两种实现方式,使用os.path.splitext更简单、原生;而mimetypes则是更灵活、可获取MIME类型。根据实际情况选择即可。