python操作文件--批量修改文件名,替换文件/文件夹名中的字符串(已使用递归算法完善代码,可通用任何文件目录)

python操作文件--批量修改文件名,替换文件/文件夹名中的字符串(已使用递归算法完善代码,可通用任何文件目录)

最近搞了一些课程资源,但是每个文件后面都有+vx的广告就比较烦,于是写了个python操作文件批量修改文件名,替换文件/文件夹名中的字符串程序,顺手复习一下好久好久大概有2,3年没写过的python

递归遍历获取文件目录的字典,再遍历字典根据传入的参数(想替换的字符串,新字符串,文件目录字典)修改文件名

import os


# 将传入参数的文件路径中的所有文件路径递归遍历返回到传入的all_files字典中
def show_files(path, all_files):
    # 首先遍历当前目录所有文件及文件夹
    file_list = os.listdir(path)
    # 准备循环判断每个元素是否是文件夹还是文件，是文件的话，把名称传入list，是文件夹的话，递归
    for file in file_list:
        # 利用os.path.join()方法取得路径全名，并存入cur_path变量，否则每次只能遍历一层目录
        cur_path = os.path.join(path, file)
        # 判断是否是文件夹
        if os.path.isdir(cur_path):
            show_files(cur_path, all_files)
        else:
            all_files.setdefault(path, []).append(file)
            # all_files.append(os.path.join(path, file))
    return all_files


# 传入要替换文件/文件夹中包含的字符串,要替换为的字符串,和文件dict
def modify_file_name(old_str, new_str, all_files, root_src):
    file_count = 0
    modify_folder_count = 0
    modify_file_count = 0
    # 先修改文件名
    for folder in all_files:
        for file in all_files[folder]:
            if old_str in file:
                new_name = file.replace(old_str, new_str)
                os.rename(os.path.join(folder, file), os.path.join(folder, new_name))
                print(file + "已成功修改为: " + new_name)
                modify_file_count += 1
            file_count += 1
    # 再修改文件夹名
    for folder in all_files:
        if old_str in folder:
            os.rename(folder, folder.replace(old_str, new_str))
            modify_folder_count += 1
    return [file_count, modify_folder_count, modify_file_count]


if __name__ == '__main__':
    src = "E:\\我的文件\\教程"
    # 传入空的dict接收文件名
    get_all_files = show_files(src, {})
    count_list = modify_file_name("【更多IT教程 微信xxx】", "", get_all_files, src)
    print("文件夹数量: " + str(len(get_all_files)))
    print("文件数量: " + str(count_list[0]))
    print("成功修改文件夹: " + str(count_list[1]))
    print("成功修改文件: " + str(count_list[2]))

这是改好的截图,之前每个后面都有"+vx xxx"字符串

然后每个文件夹中有一堆文件(图片中已经是修改好的了)