灵应
1ea8fd3989
CI/CD Pipeline / Deploy Staging (push) Has been skipped
CI/CD Pipeline / Staging E2E Tests (push) Has been skipped
CI/CD Pipeline / Build Production Runtime Images (push) Has been skipped
CI/CD Pipeline / Deploy Production (push) Has been skipped
CI/CD Pipeline / Production Browser E2E (push) Has been skipped
CI/CD Pipeline / Frontend Lint (push) Failing after 45h41m20s
CI/CD Pipeline / Validate Code Quality And Tests (push) Failing after 45h41m20s
- Schema: 新增 asset_select_mode + asset_select_count 字段 - Domain: GenerationTask 新增 asset_select_mode 字段 - Application: Command/UseCase 透传 asset_select_mode - Route: _select_assets_from_library() 辅助函数 + 创建任务集成 - SQLAlchemy: Model/Repository 映射 asset_select_mode - Alembic: 032 号迁移 - Worker: _download_library_assets() 支持 asset_ids 过滤 - 单测: 15 个测试覆盖三种模式 + 边界情况
83 lines
2.6 KiB
Python
83 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
import sys
|
|
from dataclasses import dataclass, field
|
|
from datetime import datetime, timezone
|
|
|
|
if sys.version_info >= (3, 11):
|
|
from enum import StrEnum
|
|
else:
|
|
from enum import Enum
|
|
|
|
class StrEnum(str, Enum):
|
|
pass
|
|
|
|
|
|
from uuid import uuid4
|
|
|
|
|
|
class GenerationTaskStatus(StrEnum):
|
|
PENDING = "pending"
|
|
RUNNING = "running"
|
|
COMPLETED = "completed"
|
|
FAILED = "failed"
|
|
CANCELLED = "cancelled"
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class GenerationTask:
|
|
id: str
|
|
project_id: str
|
|
asset_library_id: str
|
|
strategy_id: str = ""
|
|
voice_library_id: str = ""
|
|
template_id: str = ""
|
|
asset_ids: list[str] = field(default_factory=list)
|
|
title_ids: list[str] = field(default_factory=list)
|
|
voice_ids: list[str] = field(default_factory=list)
|
|
status: GenerationTaskStatus = GenerationTaskStatus.PENDING
|
|
progress: float = 0.0
|
|
result_count: int = 0
|
|
error_message: str = ""
|
|
started_at: datetime | None = None
|
|
completed_at: datetime | None = None
|
|
source_edit_plan_id: str = ""
|
|
created_by_user_id: str = ""
|
|
asset_select_mode: str = ""
|
|
created_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
|
|
|
|
@classmethod
|
|
def create(
|
|
cls,
|
|
project_id: str,
|
|
asset_library_id: str,
|
|
*,
|
|
strategy_id: str = "",
|
|
voice_library_id: str = "",
|
|
template_id: str = "",
|
|
asset_ids: list[str] | None = None,
|
|
title_ids: list[str] | None = None,
|
|
voice_ids: list[str] | None = None,
|
|
created_by_user_id: str = "",
|
|
source_edit_plan_id: str = "",
|
|
asset_select_mode: str = "",
|
|
) -> "GenerationTask":
|
|
if not project_id.strip() and not template_id.strip():
|
|
raise ValueError("project_id 或 template_id 至少需要提供一个")
|
|
if not asset_library_id.strip() and not (asset_ids or title_ids or voice_ids):
|
|
raise ValueError("asset_library_id 或 asset_ids/title_ids/voice_ids 至少需要提供一个")
|
|
return cls(
|
|
id=uuid4().hex,
|
|
project_id=project_id.strip(),
|
|
asset_library_id=asset_library_id.strip(),
|
|
strategy_id=strategy_id.strip(),
|
|
voice_library_id=voice_library_id.strip(),
|
|
template_id=template_id.strip(),
|
|
asset_ids=list(asset_ids) if asset_ids else [],
|
|
title_ids=list(title_ids) if title_ids else [],
|
|
voice_ids=list(voice_ids) if voice_ids else [],
|
|
created_by_user_id=created_by_user_id.strip(),
|
|
source_edit_plan_id=source_edit_plan_id.strip(),
|
|
asset_select_mode=asset_select_mode,
|
|
)
|