如何根据用户输入重命名分散在整个程序中的文件路径实例？

import os

def rename_files(user_input, folder_path):
    for root, dirs, files in os.walk(folder_path):
        for file in files:
            file_path = os.path.join(root, file)
            if need_rename(file_path):
                new_file_path = get_new_file_path(file_path, user_input)
                os.rename(file_path, new_file_path)

def need_rename(file_path):
    # 判断文件路径是否需要重命名的条件，可以根据实际需求进行修改
    # 这里以文件名包含特定字符串为例
    if "old" in file_path:
        return True
    return False

def get_new_file_path(file_path, user_input):
    # 获取新的文件路径，将文件名替换为用户输入的新文件名
    file_dir = os.path.dirname(file_path)
    file_name = os.path.basename(file_path)
    new_file_name = file_name.replace("old", user_input)
    new_file_path = os.path.join(file_dir, new_file_name)
    return new_file_path

# 调用示例
user_input = input("请输入新文件名：")
folder_path = "/path/to/your/folder"
rename_files(user_input, folder_path)