chengma/cmshoppe
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: rename bundled cover prompt template

Browse Source
This commit is contained in:
chengma
2026-07-17 18:24:58 +08:00
parent d1cd50f5c5
commit d853db1c8a
8 changed files with 93 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_gui.py
+3 -3
View File
@@ -3290,7 +3290,7 @@ class GuiTests(TempDirMixin, unittest.TestCase):
["另存为", "重命名", "删除"],
[action.text() for action in tab.title_template_actions_menu.actions()],
)
self.assertEqual("papa1", tab.cover_template_combo.currentText())
self.assertEqual("cover_prompt", tab.cover_template_combo.currentText())
self.assertIn("商品标题:{新标题}", tab.cover_prompt_edit.toPlainText())
self.assertEqual("保存模板", tab.save_cover_template_button.text())
self.assertEqual("模板操作", tab.cover_template_actions_button.text())
@@ -3934,11 +3934,11 @@ class GuiTests(TempDirMixin, unittest.TestCase):
self.addCleanup(tab.close)
self.assertIn("蝦皮台灣站", tab.title_prompt_edit.toPlainText())
self.assertEqual("papa1", tab.cover_template_combo.currentText())
self.assertEqual("cover_prompt", tab.cover_template_combo.currentText())
self.assertIn("商品标题:{新标题}", tab.cover_prompt_edit.toPlainText())
self.assertTrue(os.path.exists(title_prompt_path))
self.assertTrue(os.path.exists(os.path.join(title_templates_dir, "默认.txt")))
self.assertTrue(os.path.exists(os.path.join(cover_prompts_dir, "papa1.txt")))
self.assertTrue(os.path.exists(os.path.join(cover_prompts_dir, "cover_prompt.txt")))
self.assert_removed(temp_dir)
tests/test_packaging.py
+3
View File
@@ -32,6 +32,9 @@ class PackagingTests(unittest.TestCase):
self.assertIn("datas=default_prompt_datas", normalized)
self.assertIn("COLLECT(", spec)
self.assertIn('name="cmshopee"', normalized)
cover_prompt = self.read_text("app/default_prompts/cover/cover_prompt.txt")
self.assertIn("商品标题:{新标题}", cover_prompt)
self.assertFalse((REPO_ROOT / "app/default_prompts/cover/papa1.txt").exists())
suite_prompt = self.read_text("app/default_prompts/product_suite/base.txt")
self.assertIn("生成目标：{套图名称},{生成目标}", suite_prompt)
self.assertIn("{生成目标}", suite_prompt)
tests/test_prompts.py
+16 -2
View File
@@ -120,10 +120,10 @@ class PromptTests(TempDirMixin, unittest.TestCase):
prompts.load_title_prompt(title_path),
prompts.load_title_template("默认", title_templates_dir),
)
self.assertEqual(["papa1"], prompts.list_cover_templates(cover_dir))
self.assertEqual(["cover_prompt"], prompts.list_cover_templates(cover_dir))
self.assertIn(
"商品标题:{新标题}",
prompts.load_cover_template("papa1", cover_dir),
prompts.load_cover_template("cover_prompt", cover_dir),
)
self.assert_removed(temp_dir)
@@ -153,6 +153,20 @@ class PromptTests(TempDirMixin, unittest.TestCase):
self.assert_removed(temp_dir)
def test_ensure_default_prompts_keeps_legacy_papa1_template_unchanged(self):
with self.make_temp_dir() as temp_dir:
title_path = os.path.join(temp_dir, "title_prompt.txt")
cover_dir = os.path.join(temp_dir, "prompts", "cover")
prompts.save_cover_template("papa1", "历史封面提示词", cover_dir)
prompts.ensure_default_prompts(title_path, cover_dir)
self.assertEqual(["papa1"], prompts.list_cover_templates(cover_dir))
self.assertEqual("历史封面提示词", prompts.load_cover_template("papa1", cover_dir))
self.assertFalse(os.path.exists(os.path.join(cover_dir, "cover_prompt.txt")))
self.assert_removed(temp_dir)
def test_ensure_default_prompts_accepts_explicit_title_templates_dir(self):
with self.make_temp_dir() as temp_dir:
title_path = os.path.join(temp_dir, "custom", "title_prompt.txt")