feat: add selected task reset actions

2026-06-30 11:50:49 +08:00
parent 580a7a570d
commit f7ea9ec0fa
6 changed files with 412 additions and 8 deletions
@@ -115,6 +115,58 @@ class DbTests(TempDirMixin, unittest.TestCase):

        self.assert_removed(temp_dir)

+
+    def test_reset_generated_and_apply_status_keep_local_history(self):
+        with self.make_temp_dir() as temp_dir:
+            db_path = os.path.join(temp_dir, "cmshopee.db")
+            db.init_db(db_path)
+            batch_id = db.create_batch(["input.xlsx"], path=db_path)
+            db.insert_tasks(
+                batch_id,
+                [
+                    {
+                        "source_file_abs": os.path.join(temp_dir, "input.xlsx"),
+                        "source_sheet": "Sheet1",
+                        "source_row": 2,
+                        "account_name": "shop",
+                        "alias": "alias",
+                        "item_id": "51100639510",
+                    }
+                ],
+                path=db_path,
+            )
+            task = db.list_tasks(batch_id=batch_id, path=db_path)[0]
+            new_cover = os.path.join(temp_dir, "new.jpg")
+            with open(new_cover, "wb") as fh:
+                fh.write(b"jpeg")
+
+            db.set_collected(task.id, "旧标题", "old.jpg", path=db_path)
+            db.set_generated(task.id, "新标题", new_cover, path=db_path)
+            db.set_applied(task.id, True, path=db_path)
+
+            reset_apply = db.reset_apply_status(task.id, path=db_path)
+            self.assertEqual("applied", reset_apply["before"].stage)
+            after_apply = reset_apply["after"]
+            self.assertEqual("generated", after_apply.stage)
+            self.assertEqual("pending", after_apply.status)
+            self.assertEqual("新标题", after_apply.new_title)
+            self.assertEqual(new_cover, after_apply.new_cover_path)
+            self.assertEqual(1, after_apply.committed)
+
+            reset_generated = db.reset_generated(task.id, path=db_path)
+            after_generated = reset_generated["after"]
+            self.assertEqual(new_cover, reset_generated["new_cover_path"])
+            self.assertIsNone(reset_generated["deleted_file"])
+            self.assertTrue(os.path.exists(new_cover))
+            self.assertEqual("collected", after_generated.stage)
+            self.assertEqual("success", after_generated.status)
+            self.assertIsNone(after_generated.new_title)
+            self.assertIsNone(after_generated.new_cover_path)
+            self.assertIsNone(after_generated.last_error)
+            self.assertEqual(1, after_generated.committed)
+
+        self.assert_removed(temp_dir)
+
    def test_duplicate_task_and_invalid_update_raise_clear_errors(self):
        with self.make_temp_dir() as temp_dir:
            db_path = os.path.join(temp_dir, "cmshopee.db")