Audit Bot
ea9e536740
方案A: 后端扩展3处API改动，支持前端模板中心化模型。
1. POST /api/v1/generation/tasks — project_id改为可选，新增template_id/asset_ids/title_ids/voice_ids
2. GET /api/v1/generation/tasks + GET /api/v1/tasks — 新增用户级列表（跨project）
3. POST /api/v1/tasks/{task_id}/retry + POST /api/v1/generation/tasks/{task_id}/retry — 简化重试
改动涉及：
- Domain: GenerationTask新增6个字段，放宽create()校验
- Application: CreateGenerationTaskCommand新增字段
- Ports: GenerationTaskRepository新增list_by_user()
- Adapter: SQLAlchemy模型+仓储实现新字段和list_by_user
- Schema: 双模式校验（project模式/模板模式）
- Routes: generation_tasks + task_center双路由注册
- Migration: 015_add_generation_task_extensions
向后兼容：所有旧端点和参数不变。
105 lines
3.9 KiB
Python
105 lines
3.9 KiB
Python
from sqlalchemy.orm import Session
|
|
|
|
from packages.adapters.sqlalchemy_impl.models import GenerationTaskModel
|
|
from packages.domain import GenerationTask
|
|
|
|
|
|
def _to_domain(model: GenerationTaskModel) -> GenerationTask:
|
|
"""Convert ORM model to domain entity."""
|
|
return GenerationTask(
|
|
id=model.id,
|
|
project_id=model.project_id,
|
|
strategy_id=model.strategy_id,
|
|
asset_library_id=model.asset_library_id,
|
|
voice_library_id=model.voice_library_id,
|
|
edit_plan_id=model.edit_plan_id,
|
|
template_id=model.template_id,
|
|
asset_ids=list(model.asset_ids or []),
|
|
title_ids=list(model.title_ids or []),
|
|
voice_ids=list(model.voice_ids or []),
|
|
status=model.status,
|
|
progress=model.progress,
|
|
result_count=int(model.result_count or 0),
|
|
error_message=model.error_message,
|
|
started_at=model.started_at,
|
|
completed_at=model.completed_at,
|
|
created_by_user_id=model.created_by_user_id,
|
|
created_at=model.created_at,
|
|
)
|
|
|
|
|
|
class SQLAlchemyGenerationTaskRepository:
|
|
def __init__(self, session: Session):
|
|
self.session = session
|
|
|
|
def create(self, task: GenerationTask) -> GenerationTask:
|
|
model = GenerationTaskModel(
|
|
id=task.id,
|
|
project_id=task.project_id,
|
|
strategy_id=task.strategy_id,
|
|
asset_library_id=task.asset_library_id,
|
|
voice_library_id=task.voice_library_id,
|
|
edit_plan_id=task.edit_plan_id,
|
|
template_id=task.template_id,
|
|
asset_ids=task.asset_ids,
|
|
title_ids=task.title_ids,
|
|
voice_ids=task.voice_ids,
|
|
status=task.status,
|
|
progress=task.progress,
|
|
result_count=task.result_count,
|
|
error_message=task.error_message,
|
|
started_at=task.started_at,
|
|
completed_at=task.completed_at,
|
|
created_by_user_id=task.created_by_user_id,
|
|
created_at=task.created_at,
|
|
)
|
|
self.session.add(model)
|
|
self.session.commit()
|
|
return task
|
|
|
|
def get(self, task_id: str) -> GenerationTask | None:
|
|
model = self.session.query(GenerationTaskModel).filter(GenerationTaskModel.id == task_id).first()
|
|
if model is None:
|
|
return None
|
|
return _to_domain(model)
|
|
|
|
def list_by_project(self, project_id: str) -> list[GenerationTask]:
|
|
models = (
|
|
self.session.query(GenerationTaskModel)
|
|
.filter(GenerationTaskModel.project_id == project_id)
|
|
.order_by(GenerationTaskModel.created_at.desc())
|
|
.all()
|
|
)
|
|
return [_to_domain(m) for m in models]
|
|
|
|
def list_by_user(self, user_id: str) -> list[GenerationTask]:
|
|
models = (
|
|
self.session.query(GenerationTaskModel)
|
|
.filter(GenerationTaskModel.created_by_user_id == user_id)
|
|
.order_by(GenerationTaskModel.created_at.desc())
|
|
.all()
|
|
)
|
|
return [_to_domain(m) for m in models]
|
|
|
|
def update(self, task: GenerationTask) -> GenerationTask:
|
|
model = self.session.query(GenerationTaskModel).filter(GenerationTaskModel.id == task.id).first()
|
|
if model is None:
|
|
raise ValueError(f"GenerationTask {task.id} not found")
|
|
model.project_id = task.project_id
|
|
model.asset_library_id = task.asset_library_id
|
|
model.strategy_id = task.strategy_id
|
|
model.voice_library_id = task.voice_library_id
|
|
model.edit_plan_id = task.edit_plan_id
|
|
model.template_id = task.template_id
|
|
model.asset_ids = task.asset_ids
|
|
model.title_ids = task.title_ids
|
|
model.voice_ids = task.voice_ids
|
|
model.status = task.status
|
|
model.progress = task.progress
|
|
model.result_count = task.result_count
|
|
model.error_message = task.error_message
|
|
model.started_at = task.started_at
|
|
model.completed_at = task.completed_at
|
|
self.session.commit()
|
|
return task
|