xiaoxia-saas/apps/api/app/api/routes/_default_template.py

"""P0 fix: 共享的默认模板自动兜底逻辑。"""

from __future__ import annotations

import logging
from typing import Optional

from sqlalchemy.orm import Session

logger = logging.getLogger(__name__)


def get_or_create_default_template_id(db: Session, user_id: str) -> Optional[str]:
    """为用户查找一个有效模板；若不存在则自动创建默认配音模板。"""
    from packages.adapters.sqlalchemy_impl.models import TemplateClipConfigModel, TemplateModel
    from packages.adapters.sqlalchemy_impl.template_repository import SQLAlchemyTemplateRepository
    from packages.application.template.commands import CreateTemplateCommand, SegmentCommand
    from packages.application.template.use_cases import CreateTemplateUseCase

    existing = (
        db.query(TemplateModel)
        .filter(TemplateModel.user_id == user_id, TemplateModel.is_active.is_(True))
        .order_by(TemplateModel.created_at.asc())
        .first()
    )
    if existing is not None:
        has_seg = (
            db.query(TemplateClipConfigModel.id).filter(TemplateClipConfigModel.template_id == existing.id).first()
        )
        if has_seg:
            return existing.id

    try:
        repo = SQLAlchemyTemplateRepository(db)
        cmd = CreateTemplateCommand(
            user_id=user_id,
            name="默认配音模板",
            mode="voice_over",
            category="default",
            tags=[],
            title_config={},
            subtitle_config={},
            bgm_config={},
            estimated_duration=0.0,
            segments=[SegmentCommand(segment_order=0, duration_min=1.0, duration_max=30.0)],
        )
        tpl = CreateTemplateUseCase(repo).execute(cmd)
        db.commit()
        logger.info("[default-template] 自动创建默认模板: user=%s tpl=%s", user_id, tpl.id)
        return tpl.id
    except Exception:
        db.rollback()
        existing2 = (
            db.query(TemplateModel)
            .filter(TemplateModel.user_id == user_id, TemplateModel.is_active.is_(True))
            .order_by(TemplateModel.created_at.asc())
            .first()
        )
        if existing2 is not None:
            has_seg2 = (
                db.query(TemplateClipConfigModel.id).filter(TemplateClipConfigModel.template_id == existing2.id).first()
            )
            if has_seg2:
                return existing2.id
        logger.exception("[default-template] 自动创建默认模板失败: user=%s", user_id)
        return None