fix: 改进

src/heurams/interface/screens/nucreator.py

@@ -13,6 +13,9 @@ from textual.containers import ScrollableContainer
 from textual.screen import Screen
 
 from heurams.services.version import ver
+import toml
+from pathlib import Path
+from heurams.context import config_var
 
 
 class NucleonCreatorScreen(Screen):
@@ -39,7 +42,6 @@ class NucleonCreatorScreen(Screen):
                 except Exception as e:
                     templates.append(f"无描述模板 ({i.name})")
                     print(e)
-        print(templates)
         return templates
 
     def compose(self) -> ComposeResult:
@@ -50,7 +52,7 @@ class NucleonCreatorScreen(Screen):
                 "> 提示: 你可能注意到当选中文本框时底栏和操作按键绑定将被覆盖  \n只需选中(使用鼠标或 Tab)选择框即可恢复底栏功能"
             )
             yield Markdown("1. 键入单元集名称")
-            yield Input(placeholder="单元集名称")
+            yield Input(placeholder="单元集名称", id="name_input")
             yield Markdown(
                 "> 单元集名称不应与现有单元集重复.  \n> 新的单元集文件将创建在 ./nucleon/你输入的名称.toml"
             )
@@ -61,12 +63,12 @@ class NucleonCreatorScreen(Screen):
 古诗词模板单元集 ({ver})
 英语词汇和短语模板单元集 ({ver})
 """
-            yield Select.from_values(LINES, prompt="选择类型")
+            yield Select.from_values(LINES, prompt="选择类型", id="template_select")
             yield Markdown("> 新单元集的版本号将和主程序版本保持同步")
             yield Label(f"\n")
             yield Markdown("3. 输入常见附加元数据 (可选)")
-            yield Input(placeholder="作者")
-            yield Input(placeholder="内容描述")
+            yield Input(placeholder="作者", id="author_input")
+            yield Input(placeholder="内容描述", id="desc_input")
             yield Button(
                 "新建空白单元集",
                 id="submit_button",
@@ -87,4 +89,81 @@ class NucleonCreatorScreen(Screen):
     def on_button_pressed(self, event) -> None:
         event.stop()
         if event.button.id == "submit_button":
-            pass
+            # 获取输入值
+            name_input = self.query_one("#name_input")
+            template_select = self.query_one("#template_select")
+            author_input = self.query_one("#author_input")
+            desc_input = self.query_one("#desc_input")
+            
+            name = name_input.value.strip() # type: ignore
+            author = author_input.value.strip() # type: ignore
+            desc = desc_input.value.strip() # type: ignore
+            selected = template_select.value # type: ignore
+            
+            # 验证
+            if not name:
+                self.notify("单元集名称不能为空", severity="error")
+                return
+            
+            # 获取配置路径
+            config = config_var.get()
+            nucleon_dir = Path(config["paths"]["nucleon_dir"])
+            template_dir = Path(config["paths"]["template_dir"])
+            
+            # 检查文件是否已存在
+            nucleon_path = nucleon_dir / f"{name}.toml"
+            if nucleon_path.exists():
+                self.notify(f"单元集 '{name}' 已存在", severity="error")
+                return
+            
+            # 确定模板文件
+            if selected is None:
+                self.notify("请选择一个模板", severity="error")
+                return
+            # selected 是描述字符串，格式如 "描述 (filename.toml)"
+            # 提取文件名
+            import re
+            match = re.search(r'\(([^)]+)\)$', selected)
+            if not match:
+                self.notify("模板选择格式无效", severity="error")
+                return
+            template_filename = match.group(1)
+            template_path = template_dir / template_filename
+            if not template_path.exists():
+                self.notify(f"模板文件不存在: {template_filename}", severity="error")
+                return
+            
+            # 加载模板
+            try:
+                with open(template_path, 'r', encoding='utf-8') as f:
+                    template_data = toml.load(f)
+            except Exception as e:
+                self.notify(f"加载模板失败: {e}", severity="error")
+                return
+            
+            # 更新元数据
+            metadata = template_data.get("__metadata__", {})
+            attribution = metadata.get("attribution", {})
+            if author:
+                attribution["author"] = author
+            if desc:
+                attribution["desc"] = desc
+            attribution["name"] = name
+            # 可选: 设置版本
+            attribution["version"] = ver
+            metadata["attribution"] = attribution
+            template_data["__metadata__"] = metadata
+            
+            # 确保 nucleon_dir 存在
+            nucleon_dir.mkdir(parents=True, exist_ok=True)
+            
+            # 写入新文件
+            try:
+                with open(nucleon_path, 'w', encoding='utf-8') as f:
+                    toml.dump(template_data, f)
+            except Exception as e:
+                self.notify(f"保存单元集失败: {e}", severity="error")
+                return
+            
+            self.notify(f"单元集 '{name}' 创建成功")
+            self.app.pop_screen()