from .nucleon import Nucleon
from .electron import Electron
import heurams.services.hasher as hasher
import pathlib
import toml
import json
from copy import deepcopy

def load_nucleon(path: pathlib.Path, fmt = "toml"):
    with open(path, "r") as f:
        dictdata = dict()
        dictdata = toml.load(f) # type: ignore
        lst = list()
        nested_data = dict()
        # 修正 toml 解析器的不管嵌套行为
        for key, value in dictdata.items():
            if "__metadata__" in key: # 以免影响句号
                if '.' in key:
                    parts = key.split('.')
                    current = nested_data
                    for part in parts[:-1]:
                        if part not in current:
                            current[part] = {}
                        current = current[part]
                    current[parts[-1]] = value
            else:
                nested_data[key] = value
        # print(nested_data)
        for item, attr in nested_data.items():
            if item == "__metadata__":
                continue
            lst.append((Nucleon(hasher.hash(item), attr, deepcopy(nested_data['__metadata__'])), deepcopy(nested_data["__metadata__"]["orbital"])))
        return lst

def load_electron(path: pathlib.Path, fmt = "json") -> dict:
    """从文件路径加载电子对象

    Args:
        path (pathlib.Path): 路径
        fmt (str): 文件格式(可选, 默认 json)
    
    Returns:
        dict: 键名是电子对象名称, 值是电子对象
    """
    with open(path, "r") as f:
        dictdata = dict()
        dictdata = json.load(f) # type: ignore
        dic = dict()
        for item, attr in dictdata.items():
            dic[item] = (Electron(hasher.hash(item), attr))
        return dic