如何用Python程序修改扩展名

如何用Python程序修改扩展名

用Python程序修改文件的扩展名非常简单，只需几行代码就能实现。使用os模块、使用glob模块、批量修改。本文将详细介绍如何使用Python来修改文件的扩展名，并展示一些实用的示例代码。

一、使用os模块

Python的os模块提供了一些基本的操作系统接口，可以用来修改文件扩展名。os模块中的rename()函数可以重命名文件或目录，包括修改文件的扩展名。

import os
def change_extension(directory, old_ext, new_ext):
    for filename in os.listdir(directory):
        if filename.endswith(old_ext):
            base = os.path.splitext(filename)[0]
            new_name = base + new_ext
            os.rename(os.path.join(directory, filename), os.path.join(directory, new_name))
directory = '/path/to/your/directory'
old_extension = '.txt'
new_extension = '.md'
change_extension(directory, old_extension, new_extension)

在上面的代码中，我们首先导入os模块，然后定义了一个change_extension函数，该函数接受三个参数：目录、旧扩展名和新扩展名。通过os.listdir()函数，我们可以获取目录中的所有文件名，并使用os.path.splitext()函数来分割文件名和扩展名。最后，我们使用os.rename()函数来重命名文件。

示例解释

假设我们有一个目录，里面包含一些.txt文件，我们希望将这些文件的扩展名改为.md。我们只需将目录路径、旧扩展名和新扩展名传递给change_extension函数即可。程序会自动遍历目录中的所有文件，并将扩展名为.txt的文件修改为.md。

二、使用glob模块

glob模块提供了一种方便的方式来查找符合特定模式的文件名。与os模块不同，glob模块允许使用通配符来匹配文件名，这使得批量修改文件扩展名变得更加方便。

import os
import glob
def change_extension_glob(directory, old_ext, new_ext):
    pattern = os.path.join(directory, f'*{old_ext}')
    for filepath in glob.glob(pattern):
        base = os.path.splitext(filepath)[0]
        new_filepath = base + new_ext
        os.rename(filepath, new_filepath)
directory = '/path/to/your/directory'
old_extension = '.jpg'
new_extension = '.png'
change_extension_glob(directory, old_extension, new_extension)

在上面的代码中，我们首先导入glob和os模块，然后定义了一个change_extension_glob函数，该函数接受三个参数：目录、旧扩展名和新扩展名。通过os.path.join()函数，我们可以构建一个模式字符串，该字符串用于匹配目录中的所有旧扩展名文件。然后，我们使用glob.glob()函数来查找所有符合模式的文件，并使用os.path.splitext()函数来分割文件名和扩展名。最后，我们使用os.rename()函数来重命名文件。

示例解释

假设我们有一个目录，里面包含一些.jpg文件，我们希望将这些文件的扩展名改为.png。我们只需将目录路径、旧扩展名和新扩展名传递给change_extension_glob函数即可。程序会自动查找目录中的所有.jpg文件，并将扩展名修改为.png。

三、批量修改

在实际应用中，我们可能需要批量修改多个目录中的文件扩展名，这时可以编写一个更通用的函数来处理。

import os
def batch_change_extension(directories, old_ext, new_ext):
    for directory in directories:
        for filename in os.listdir(directory):
            if filename.endswith(old_ext):
                base = os.path.splitext(filename)[0]
                new_name = base + new_ext
                os.rename(os.path.join(directory, filename), os.path.join(directory, new_name))
directories = ['/path/to/your/directory1', '/path/to/your/directory2']
old_extension = '.avi'
new_extension = '.mp4'
batch_change_extension(directories, old_extension, new_extension)

在上面的代码中，我们定义了一个batch_change_extension函数，该函数接受三个参数：目录列表、旧扩展名和新扩展名。通过遍历目录列表，我们可以批量修改多个目录中的文件扩展名。

示例解释

假设我们有两个目录，里面包含一些.avi文件，我们希望将这些文件的扩展名改为.mp4。我们只需将目录路径列表、旧扩展名和新扩展名传递给batch_change_extension函数即可。程序会自动遍历目录列表中的每个目录，并将扩展名为.avi的文件修改为.mp4。

四、处理异常情况

在实际应用中，可能会遇到一些异常情况，例如文件不存在、权限不足等。为了提高程序的健壮性，我们可以添加一些异常处理机制。

import os
def safe_change_extension(directory, old_ext, new_ext):
    for filename in os.listdir(directory):
        if filename.endswith(old_ext):
            base = os.path.splitext(filename)[0]
            new_name = base + new_ext
            try:
                os.rename(os.path.join(directory, filename), os.path.join(directory, new_name))
                print(f'Successfully renamed {filename} to {new_name}')
            except Exception as e:
                print(f'Error renaming {filename}: {e}')
directory = '/path/to/your/directory'
old_extension = '.doc'
new_extension = '.pdf'
safe_change_extension(directory, old_extension, new_extension)