从文件名构建嵌套字典的实用指南

import os
from pathlib import Path
import numpy as np
import xarray as xr  # 假设需要处理 xarray 数据

def build_nested_dictionary(folder_path, fileNamesAndTypes_check):
    """
    从指定文件夹中的文件名构建嵌套字典。

    Args:
        folder_path (str): 包含文件的文件夹路径。
        fileNamesAndTypes_check (dict): 文件名（不带扩展名）到文件类型的映射。

    Returns:
        dict: 构建的嵌套字典。
    """

    vz_reloaded = {}  # 初始化主字典

    files = [f for f in os.listdir(folder_path) if not f.startswith('.') and f != 'fileNamesAndTypes.txt']  # 过滤文件

    for file in files:
        file_path = Path(folder_path, file)
        fileNoExt = Path(file).stem  # 去除扩展名
        keys = fileNoExt.split('_')
        if 'vz' in keys:
            keys.remove('vz')  # 移除根名称

        reloadedType = fileNamesAndTypes_check[fileNoExt] # 获取文件类型

        try:
            if reloadedType == str:
                with open(file_path, 'rb') as f:
                    contents = f.read()
            elif reloadedType == xr.core.dataset.Dataset:
                contents = xr.open_dataset(file_path)
            elif reloadedType == xr.core.dataarray.DataArray:
                contents = xr.open_dataarray(file_path)
            elif reloadedType == int:
                contents = np.load(file_path).item() # 或者其他加载方式
            elif reloadedType == list:
                contents = np.load(file_path).tolist()
            elif reloadedType == np.ndarray:
                contents = np.load(file_path)
            else:
                contents = None
                print(f"Unsupported file type: {reloadedType} for file {file}")


            # 动态构建嵌套字典
            current_dict = vz_reloaded
            for i, key in enumerate(keys):
                if i == len(keys) - 1:
                    current_dict[key] = contents  # 最后一个键，赋值
                else:
                    if key not in current_dict:
                        current_dict[key] = {}  # 如果键不存在，创建新的字典
                    current_dict = current_dict[key]  # 进入下一层

        except Exception as e:
            print(f"Error processing file {file}: {e}")

    return vz_reloaded


# 示例用法
# 假设 fc.selected 是包含 'vz_save' 文件夹的路径
fc_selected = "./data"  # 替换为你的实际路径
folder_path = Path(fc_selected, 'vz_save')

# 假设 fileNamesAndTypes_check 已经定义
fileNamesAndTypes_check = {
    "ERA_Neural_Water_Forecast": str,
    "ERA_Polynomial_Reof": np.ndarray,
    # ... 其他文件名和类型
}

# 创建一个模拟的 vz_save 文件夹和一些文件
if not os.path.exists(folder_path):
    os.makedirs(folder_path)

# 创建一些模拟文件
with open(Path(folder_path, "ERA_Neural_Water_Forecast.txt"), "w") as f:
    f.write("This is a test string.")
np.save(Path(folder_path, "ERA_Polynomial_Reof.npy"), np.array([1, 2, 3]))

vz_reloaded = build_nested_dictionary(folder_path, fileNamesAndTypes_check)
print(vz_reloaded)