feat(gui): replace AI studio with product suite

2026-07-14 09:53:13 +08:00
parent fb873aae90
commit bc115ba0d7
27 changed files with 3453 additions and 81 deletions
@@ -0,0 +1,172 @@
+import os
+import sys
+import unittest
+
+os.environ.setdefault("QT_QPA_PLATFORM", "offscreen")
+sys.path.insert(0, os.path.dirname(__file__))
+
+from _helpers import TempDirMixin
+
+from app import accounts, image_studio, image_studio_images
+from app import gui
+
+if gui.QT_IMPORT_ERROR is not None:
+    raise unittest.SkipTest("PySide6 未安装")
+
+from PySide6.QtGui import QImage
+from PySide6.QtWidgets import QApplication, QLabel, QPushButton
+
+from app.gui.tabs.product_suite import ProductSuiteTab, SuiteResultCard
+
+
+class ProductSuiteGuiTests(TempDirMixin, unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.app = QApplication.instance() or QApplication([])
+
+    def tearDown(self):
+        for widget in QApplication.topLevelWidgets():
+            widget.close()
+            widget.deleteLater()
+        self.app.processEvents()
+
+    def _config(self, temp_dir):
+        return {
+            "chrome_path": "chrome.exe",
+            "user_data_root": os.path.join(temp_dir, "chrome_user_data_dir"),
+            "image_dir": os.path.join(temp_dir, "images"),
+            "db_path": os.path.join(temp_dir, "cmshopee.db"),
+            "debug_port_range": [9222, 9260],
+            "config_path": os.path.join(temp_dir, "config.json"),
+            "cmhub_config_path": os.path.join(temp_dir, "cmhub.json"),
+        }
+
+    def _write_image(self, path):
+        image = QImage(40, 30, QImage.Format_RGB32)
+        image.fill(0xFF336699)
+        self.assertTrue(image.save(path))
+
+    def test_tab_builds_suite_controls_without_old_detail_workspace(self):
+        with self.make_temp_dir() as temp_dir:
+            config = self._config(temp_dir)
+            accounts.create_account("主店", "alias-a", debug_port=9222, config=config)
+            tab = ProductSuiteTab(config=config, db_path=config["db_path"])
+            self.addCleanup(tab.close)
+
+            self.assertEqual("productSuiteTab", tab.objectName())
+            self.assertEqual(1, tab.task_tabs.count())
+            self.assertEqual("套图任务 1", tab.task_tabs.tabText(0))
+            self.assertEqual("alias-a", tab.account_combo.currentData())
+            self.assertEqual("Shopee", tab.platform_combo.currentData())
+            self.assertEqual("中国台湾", tab.country_combo.currentData())
+            self.assertEqual("繁体中文", tab.language_combo.currentData())
+            self.assertEqual("1:1", tab.ratio_combo.currentData())
+            self.assertEqual("合计 5 张", tab.category_total_label.text())
+            self.assertEqual("生成套图（5）", tab.generate_button.text())
+
+            visible_text = " ".join(
+                [widget.text() for widget in tab.findChildren(QLabel)]
+                + [widget.text() for widget in tab.findChildren(QPushButton)]
+            )
+            self.assertNotIn("详情图", visible_text)
+            self.assertNotIn("AI工场", visible_text)
+            self.assertIn("白底图", visible_text)
+            self.assertIn("场景图", visible_text)
+            self.assertIn("卖点图", visible_text)
+
+            tab.add_custom_category()
+            self.assertFalse(tab.custom_category_edit.isHidden())
+            tab.custom_category_edit.setText("尺寸图")
+            tab._commit_custom_category()
+            self.assertTrue(tab.custom_category_edit.isHidden())
+            self.assertIn("尺寸图", tab._displayed_state.settings["categories"])
+            self.assertEqual("尺寸图", tab._displayed_state.active_category)
+
+        self.assert_removed(temp_dir)
+
+    def test_task_tabs_keep_independent_prompt_and_settings(self):
+        with self.make_temp_dir() as temp_dir:
+            config = self._config(temp_dir)
+            accounts.create_account("主店", "alias-a", debug_port=9222, config=config)
+            tab = ProductSuiteTab(config=config, db_path=config["db_path"])
+            self.addCleanup(tab.close)
+
+            tab.prompt_edit.setPlainText("任务一卖点")
+            first_state = tab._displayed_state
+            tab.ratio_combo.setCurrentIndex(tab.ratio_combo.findData("3:4"))
+            second_state = tab.add_task(inherit=True)
+
+            self.assertEqual(2, tab.task_tabs.count())
+            self.assertEqual("任务一卖点", second_state.prompt)
+            self.assertEqual("3:4", second_state.settings["ratio"])
+            tab.prompt_edit.setPlainText("任务二卖点")
+            tab.ratio_combo.setCurrentIndex(tab.ratio_combo.findData("16:9"))
+            tab.task_tabs.setCurrentIndex(0)
+
+            self.assertIs(first_state, tab._displayed_state)
+            self.assertEqual("任务一卖点", tab.prompt_edit.toPlainText())
+            self.assertEqual("3:4", tab.ratio_combo.currentData())
+
+        self.assert_removed(temp_dir)
+
+    def test_project_settings_and_result_history_use_existing_backend(self):
+        with self.make_temp_dir() as temp_dir:
+            config = self._config(temp_dir)
+            account = accounts.create_account("主店", "alias-a", debug_port=9222, config=config)
+            project = image_studio.create_or_get_project(
+                account,
+                item_id="51100639510",
+                path=config["db_path"],
+            )
+            source_path = os.path.join(temp_dir, "source.png")
+            self._write_image(source_path)
+            source = image_studio_images.import_original_files(
+                project.id,
+                [source_path],
+                path=config["db_path"],
+                config=config,
+            )["assets"][0]
+            job = image_studio.create_job(
+                project.id,
+                source_asset_id=source.id,
+                job_type="场景图",
+                prompt="场景卖点",
+                path=config["db_path"],
+            )
+            image_studio.update_job_status(
+                job.id,
+                "failed",
+                error="上游超时 https://example.invalid/private",
+                path=config["db_path"],
+            )
+
+            tab = ProductSuiteTab(config=config, db_path=config["db_path"])
+            self.addCleanup(tab.close)
+            state = tab._displayed_state
+            state.account_alias = "alias-a"
+            state.item_id = "51100639510"
+            state.project_id = project.id
+            state.current_job_ids = [job.id]
+            tab._load_state(state)
+
+            self.assertEqual([source.id], tab.original_list.asset_ids())
+            self.assertEqual("共 1 张 · 成功 0 张", tab.result_summary_label.text())
+            cards = tab.findChildren(SuiteResultCard)
+            self.assertEqual(1, len(cards))
+            self.assertNotIn(
+                "https://",
+                " ".join(label.text() for label in cards[0].findChildren(QLabel)),
+            )
+
+            state.settings["ratio"] = "4:3"
+            state.prompt = "持久化卖点"
+            tab._persist_state(state)
+            stored = image_studio.get_project(project.id, path=config["db_path"])
+            self.assertEqual("持久化卖点", stored.draft_prompt)
+            self.assertEqual("4:3", image_studio.project_suite_settings(stored)["ratio"])
+
+        self.assert_removed(temp_dir)
+
+
+if __name__ == "__main__":
+    unittest.main()