部分移植

src/heurams/kernel/particles/atom.py

@@ -1,13 +1,17 @@
-from electron import Electron
-from nucleon import Nucleon
+from .electron import Electron
+from .nucleon import Nucleon
+import pathlib
+import typing
+import toml
+import json
 
 class Atom():
     """
-    一个静态类, 包含一个原子对象的所有信息:
+    统一处理一系列对象的所有信息与持久化:
         关联电子 (算法数据)
         关联核子 (内容数据)
         关联轨道 (策略数据)
-        关联路径 ()
+        以及关联路径
     """
 
     def __init__(self, ident = ""):
@@ -15,15 +19,36 @@ class Atom():
         self.register = {
             "nucleon": None,
             "nucleon_path": None,
+            "nucleon_fmt": "toml",
             "electron": None,
             "electron_path": None,
-            "orbital": None,            
+            "electron_fmt": "json",
+            "orbital": None,
+            "orbital_path": None, # 允许设置为 None, 此时使用 nucleon 文件内的推荐配置
+            "orbital_fmt": "toml",
         }
 
     def link(self, key, value):
-        self.register[key] = value
-    
+        if key in self.register.keys():
+            self.register[key] = value
+        else:
+            raise ValueError("不受支持的原子元数据链接操作")
     
+    def persist(self, key):
+        path: pathlib.Path | None = self.register[key + "_path"]
+        if isinstance(path, pathlib.Path):
+            path = typing.cast(pathlib.Path, path)
+            path.parent.mkdir(parents=True, exist_ok=True)
+            if self.register[key + "_fmt"] == "toml":
+                with open(path, "w") as f:
+                    toml.dump(self.register[key], f)
+            elif self.register[key + "_fmt"] == "json":
+                with open(path, "w") as f:
+                    json.dump(self.register[key], f)
+            else:
+                raise KeyError("不受支持的持久化格式")
+        else:
+            raise TypeError("对未初始化的路径对象操作")
 
     @staticmethod
     def placeholder():