diff --git a/scripts/tier2/artifacts/result_migration_small_files_20260617/apply_edit.py b/scripts/tier2/artifacts/result_migration_small_files_20260617/apply_edit.py new file mode 100644 index 00000000..809a6d85 --- /dev/null +++ b/scripts/tier2/artifacts/result_migration_small_files_20260617/apply_edit.py @@ -0,0 +1,9 @@ +with open('src/app_controller.py', 'r', encoding='utf-8') as f: + content = f.read() +old = " def load_context_preset(self, name: str) -> models.ContextPreset:\n presets = self.context_preset_manager.load_all(self.project)\n if name not in presets:\n raise KeyError(f\"Context preset '{name}' not found.\")\n preset = presets[name]" +new = " def load_context_preset(self, name: str) -> models.ContextPreset:\n presets_result = self.context_preset_manager.load_all(self.project)\n if not presets_result.ok:\n raise RuntimeError(f\"Failed to load context presets: {presets_result.errors}\")\n presets = presets_result.data\n if name not in presets:\n raise KeyError(f\"Context preset '{name}' not found.\")\n preset = presets[name]" +assert old in content, 'old not found' +content = content.replace(old, new) +with open('src/app_controller.py', 'w', encoding='utf-8') as f: + f.write(content) +print('Done') diff --git a/src/app_controller.py b/src/app_controller.py index 7b7c5c13..8496a9ce 100644 --- a/src/app_controller.py +++ b/src/app_controller.py @@ -2977,7 +2977,10 @@ class AppController: self._save_active_project() def load_context_preset(self, name: str) -> models.ContextPreset: - presets = self.context_preset_manager.load_all(self.project) + presets_result = self.context_preset_manager.load_all(self.project) + if not presets_result.ok: + raise RuntimeError(f"Failed to load context presets: {presets_result.errors}") + presets = presets_result.data if name not in presets: raise KeyError(f"Context preset '{name}' not found.") preset = presets[name] diff --git a/tests/test_context_presets_manager.py b/tests/test_context_presets_manager.py index eb06ca36..3e1805e0 100644 --- a/tests/test_context_presets_manager.py +++ b/tests/test_context_presets_manager.py @@ -21,7 +21,9 @@ def project_dict(): def test_manager_load_all(project_dict): manager = ContextPresetManager() - presets = manager.load_all(project_dict) + result = manager.load_all(project_dict) + assert result.ok, f"load_all failed: {result.errors}" + presets = result.data assert "test_preset" in presets assert len(presets["test_preset"].files) == 2 assert presets["test_preset"].files[0].path == "file1.py"