🐛 使用文件名作为预设键保证唯一性

Author: shadow3aaa

src/nonebot_plugin_nyaturingtest/presets.py

@@ -56,14 +56,14 @@ class RolePreset:
     ],
 )
 
-PRESETS = []
+PRESETS: dict[str, RolePreset] = {}
 
 
 def _load_presets_from_directory(directory: str = "./nya_presets"):
     # 如果文件夹不存在就创建并且写入例子(_猫娘预设)
     if not os.path.exists(directory):
         os.makedirs(directory)
-        with open(os.path.join(directory, "猫娘预设.json"), "w", encoding="utf-8") as f:
+        with open(os.path.join(directory, "喵喵.json"), "w", encoding="utf-8") as f:
             json.dump(_猫娘预设.__dict__, f, ensure_ascii=False, indent=4)
     for filename in os.listdir(directory):
         if filename.endswith(".json"):
@@ -72,7 +72,7 @@ def _load_presets_from_directory(directory: str = "./nya_presets"):
                 with open(path, encoding="utf-8") as f:
                     data = json.load(f)
                     preset = RolePreset(**data)
-                    PRESETS.append(preset)
+                    PRESETS[filename] = preset
             except Exception as e:
                 logger.warning(f"无法加载预设 {filename}: {e}")
 

src/nonebot_plugin_nyaturingtest/session.py

@@ -318,24 +318,24 @@ def presets(self) -> list[str]:
         """
         获取可选预设
         """
-        return [f"{preset.name} {preset.role}" for preset in PRESETS if not preset.hidden]
+        return [f"{filename}: {preset.name} {preset.role}" for filename, preset in PRESETS.items() if not preset.hidden]
 
-    def load_preset(self, preset_name: str) -> bool:
+    def load_preset(self, filename: str) -> bool:
         """
         加载预设
         """
-        for preset in PRESETS:
-            if preset.name == preset_name:
-                self.reset()
-                self.set_role(preset.name, preset.role)
-                self.long_term_memory_knowledge.add_texts(preset.knowledges)
-                self.long_term_memory_relationships.add_texts(preset.relationships)
-                self.long_term_memory_events.add_texts(preset.events)
-                self.long_term_memory_self.add_texts(preset.bot_self)
-                logger.info(f"加载预设：{preset_name}成功")
-                return True
-        logger.error(f"不存在的预设：{preset_name}")
-        return False
+        if filename not in PRESETS.keys():
+            logger.error(f"不存在的预设：{filename}")
+            return False
+        preset = PRESETS[filename]
+        self.reset()
+        self.set_role(preset.name, preset.role)
+        self.long_term_memory_knowledge.add_texts(preset.knowledges)
+        self.long_term_memory_relationships.add_texts(preset.relationships)
+        self.long_term_memory_events.add_texts(preset.events)
+        self.long_term_memory_self.add_texts(preset.bot_self)
+        logger.info(f"加载预设：{filename} 成功")
+        return True
 
     def status(self) -> str:
         """