如何递归抓取特定文件并将其相对路径存储在字典中？

import os

def recursive_file_search(directory, file_extension, result_dict):
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith(file_extension):
                file_path = os.path.join(root, file)
                relative_path = os.path.relpath(file_path, directory)
                result_dict[relative_path] = file_path

        for dir in dirs:
            new_directory = os.path.join(root, dir)
            result_dict = recursive_file_search(new_directory, file_extension, result_dict)

    return result_dict

# 示例用法
directory = "/path/to/directory"  # 指定目录
file_extension = ".txt"  # 指定文件类型
result = {}  # 初始化结果字典
result = recursive_file_search(directory, file_extension, result)
print(result)