feat(data-process): 完善后台生成与失败重试

backend/app/modules/data_process/schema_cli.py

@@ -7,6 +7,18 @@ from urllib.parse import urlsplit
 
 from app.modules.data_process.store import DataProcessStore
 
+REQUIRED_TASK_COLUMNS = (
+    "generation_run_id",
+    "results_confirmed",
+    "workflow_step",
+    "preview_status",
+    "preview_progress",
+    "preview_run_id",
+    "preview_failure_reason",
+    "preview_total_files",
+    "preview_completed_files",
+)
+
 
 def _target_label(database_url: str) -> str:
     parsed = urlsplit(database_url)
@@ -18,14 +30,13 @@ def _schema_ready(store: DataProcessStore) -> bool:
     with store.connect() as conn:
         row = conn.execute(
             """
-            SELECT EXISTS (
-                SELECT 1
-                FROM information_schema.columns
-                WHERE table_schema=current_schema()
-                  AND table_name='data_process_tasks'
-                  AND column_name='generation_run_id'
-            ) AS ready
-            """
+            SELECT COUNT(*) = %s AS ready
+            FROM information_schema.columns
+            WHERE table_schema=current_schema()
+              AND table_name='data_process_tasks'
+              AND column_name = ANY(%s)
+            """,
+            (len(REQUIRED_TASK_COLUMNS), list(REQUIRED_TASK_COLUMNS)),
         ).fetchone()
     return bool(row and row["ready"])