实用的skill入门指南：从零构建你的第一个自动化脚本

2次阅读

没有评论

共计 2562 个字符，预计需要花费 7 分钟才能阅读完成。

在日常开发中，我们经常需要处理一些重复性的任务，比如批量重命名文件、整理文件夹结构、处理大量数据等。这些任务虽然简单，但如果每次都手动操作，不仅效率低下，还容易出错。这时候，自动化脚本就显得尤为重要。

Python 作为一种简单易学的编程语言，非常适合用来编写自动化脚本。通过 Python 脚本，我们可以将这些重复性的任务自动化，从而节省大量时间，提升工作效率。

在开始编写脚本之前，我们需要确保我们的开发环境已经准备就绪。以下是环境配置的步骤：

安装 Python 3.x：可以从 Python 官网下载并安装最新版本的 Python。
安装必要的库：对于文件操作，Python 内置的 os 和shutil库已经足够。如果需要更复杂的功能，可以使用 glob 或pathlib库。

安装完成后，可以通过以下命令检查 Python 版本：

python --version

下面，我们将一步步实现一个简单的文件批量重命名脚本。这个脚本的功能是将指定文件夹中的所有文件按照一定的规则重命名。

import os

def batch_rename(folder_path, prefix):
    """
    批量重命名文件夹中的文件
    :param folder_path: 文件夹路径
    :param prefix: 文件名前缀
    """
    # 获取文件夹中的所有文件
    files = os.listdir(folder_path)

    # 遍历文件并重命名
    for index, file in enumerate(files):
        # 获取文件的完整路径
        old_path = os.path.join(folder_path, file)

        # 如果是文件而不是文件夹
        if os.path.isfile(old_path):
            # 获取文件扩展名
            file_ext = os.path.splitext(file)[1]

            # 构建新的文件名
            new_name = f"{prefix}_{index}{file_ext}"
            new_path = os.path.join(folder_path, new_name)

            # 重命名文件
            os.rename(old_path, new_path)
            print(f"Renamed {file} to {new_name}")

if __name__ == "__main__":
    folder_path = input("请输入文件夹路径：")
    prefix = input("请输入文件名前缀：")
    batch_rename(folder_path, prefix)

os.listdir(folder_path)：获取指定文件夹中的所有文件和文件夹列表。
os.path.isfile(old_path)：判断路径是否为文件。
os.path.splitext(file)[1]：获取文件的扩展名。
os.rename(old_path, new_path)：重命名文件。

在实际使用中，可能会遇到一些错误，以下是一些常见错误及其解决方案：

文件夹路径错误 ：如果输入的文件夹路径不存在，程序会抛出FileNotFoundError。可以在调用os.listdir 前检查路径是否存在。

if not os.path.exists(folder_path):
    print("文件夹路径不存在！")
    exit()

权限不足：如果程序没有权限访问或修改文件夹中的文件，会抛出PermissionError。确保程序有足够的权限。
文件名冲突 ：如果新文件名与已有文件名冲突，os.rename 会抛出FileExistsError。可以在重命名前检查新文件名是否已存在。

if os.path.exists(new_path):
    print(f"文件 {new_name} 已存在，跳过重命名")
    continue

虽然这个脚本已经足够简单高效，但在处理大量文件时，还可以进一步优化：

多线程处理：如果文件数量非常多，可以使用多线程来加速重命名过程。

from concurrent.futures import ThreadPoolExecutor

def rename_file(args):
    old_path, new_path = args
    os.rename(old_path, new_path)
    print(f"Renamed {os.path.basename(old_path)} to {os.path.basename(new_path)}")

def batch_rename(folder_path, prefix):
    files = os.listdir(folder_path)
    args_list = []

    for index, file in enumerate(files):
        old_path = os.path.join(folder_path, file)
        if os.path.isfile(old_path):
            file_ext = os.path.splitext(file)[1]
            new_name = f"{prefix}_{index}{file_ext}"
            new_path = os.path.join(folder_path, new_name)
            args_list.append((old_path, new_path))

    with ThreadPoolExecutor() as executor:
        executor.map(rename_file, args_list)