Python文件批量修改

import os
from pathlib import Path


def menu_one():
    folder_path = input("请输入需要批量修改文件名的路径：")
    content = input("请输入需要去掉的内容：")
    file_names = os.listdir(folder_path)
    for file_name in file_names:
        new_file_name = file_name.replace(content, '')
        os.rename(os.path.join(folder_path, file_name), os.path.join(folder_path, new_file_name))
        print(f'{file_name} to {new_file_name}')


def menu_two():
    path = input("请输入需要批量修改文件名的路径：")
    prefix = input("请输入需要在文件名前添加的内容：")
    for file_name in os.listdir(path):
        # 重命名文件
        os.rename(os.path.join(path, file_name), os.path.join(path, prefix + file_name))
        print(f'{file_name} to {prefix + file_name}')


def menu_three():
    folder_path = input("请输入需要批量修改文件名的路径：")
    old = input("请输入修改前的内容：")
    new = input("请输入修改后的内容：")
    file_names = os.listdir(folder_path)
    for file_name in file_names:
        new_file_name = file_name.replace(old, new)
        os.rename(os.path.join(folder_path, file_name), os.path.join(folder_path, new_file_name))
        print(f'{file_name} to {new_file_name}')


def menu_four():
    folder_path = input("请输入需要文件名需要批量添加后缀的路径：")
    ext = input("请输入需要添加的后缀（.exe）")
    folder_path = Path(folder_path)
    for file_path in folder_path.rglob('*'):
        if file_path.is_file():
            new_file_name = f"{file_path}.{ext}"
            print(f"{file_path} to {new_file_name}")
            os.rename(file_path, new_file_name)


def menu():
    while True:
        print("1、删除文件名中指定的内容")
        print("2、文件名前面增加内容")
        print("3、修改文件名中指定内容")
        print("4、批量添加文件名后缀")
        mode = eval(input("输入对应序号："))
        if mode == 1:
            menu_one()
        elif mode == 2:
            menu_two()
        elif mode == 3:
            menu_three()
        elif mode == 4:
            menu_four()
        else:
            print("选项错误")


if __name__ == '__main__':
    menu()