217 lines
8.1 KiB
Python
217 lines
8.1 KiB
Python
"""司命台 · 人物模板管理聚合服务（api-spec §19 扩展）。
|
||
|
||
系统模板 CRUD，所有写操作记审计并入起居注。
|
||
路由直接调用此服务层函数（函数内部自管 ``AsyncSessionLocal``）。
|
||
"""
|
||
|
||
from app.core.config import settings
|
||
from app.core.errors import AppError
|
||
from app.core.ids import new_character_id
|
||
from app.core.time import utc_now
|
||
from app.db.session import AsyncSessionLocal
|
||
from app.models.character import Character
|
||
from app.repositories import characters as character_repo
|
||
from app.repositories.json_utils import dumps, loads_dict
|
||
from app.schemas.admin_character import (
|
||
AdminCreateTemplateRequest,
|
||
AdminTemplateSummaryResponse,
|
||
AdminUpdateTemplateRequest,
|
||
)
|
||
from app.services.audit import record_audit_event
|
||
|
||
|
||
def _template_to_response(template: Character) -> AdminTemplateSummaryResponse:
|
||
profile = loads_dict(template.profile_json)
|
||
model_policy = loads_dict(template.model_policy_json)
|
||
# 从 profile 中推断 is_default（无独立字段，保留为 False 未映射）
|
||
return AdminTemplateSummaryResponse(
|
||
key=template.template_key or "",
|
||
name=template.name,
|
||
display_name=template.display_name,
|
||
language=template.language,
|
||
privacy_level=template.privacy_level,
|
||
is_default=False,
|
||
profile=profile,
|
||
model_policy=model_policy,
|
||
created_at=template.created_at.isoformat() if template.created_at else "",
|
||
updated_at=template.updated_at.isoformat() if template.updated_at else "",
|
||
)
|
||
|
||
|
||
async def admin_list_templates() -> list[AdminTemplateSummaryResponse]:
|
||
"""列出所有系统模板（管理端完整视图）。"""
|
||
async with AsyncSessionLocal() as session:
|
||
templates = await character_repo.list_system_templates(session)
|
||
return [_template_to_response(t) for t in templates]
|
||
|
||
|
||
async def admin_get_template(key: str) -> AdminTemplateSummaryResponse:
|
||
"""获取单个系统模板。"""
|
||
async with AsyncSessionLocal() as session:
|
||
template = await character_repo.get_system_template_by_key(session, key)
|
||
if template is None:
|
||
raise AppError("not_found", f"模板 '{key}' 不存在。")
|
||
return _template_to_response(template)
|
||
|
||
|
||
async def admin_create_template(
|
||
payload: AdminCreateTemplateRequest,
|
||
) -> AdminTemplateSummaryResponse:
|
||
"""创建系统模板。"""
|
||
async with AsyncSessionLocal() as session:
|
||
# 检查是否已存在同 key 模板
|
||
existing = await character_repo.get_system_template_by_key(session, payload.key)
|
||
if existing is not None:
|
||
raise AppError("conflict", f"模板 key '{payload.key}' 已存在。")
|
||
|
||
profile = dict(payload.profile)
|
||
model_policy = dict(payload.model_policy)
|
||
model_policy.setdefault("default_chat_model_hint", settings.default_model_hint)
|
||
|
||
character = Character(
|
||
id=new_character_id(),
|
||
name=payload.name,
|
||
display_name=payload.display_name,
|
||
status="active",
|
||
language=payload.language,
|
||
profile_json=dumps(profile),
|
||
model_policy_json=dumps(model_policy),
|
||
privacy_level=payload.privacy_level,
|
||
owner_scope="system",
|
||
is_system_template=True,
|
||
source_template_id=None,
|
||
template_key=payload.key,
|
||
)
|
||
session.add(character)
|
||
await session.flush()
|
||
|
||
# birthday 兜底
|
||
await _ensure_birthday(character)
|
||
|
||
await record_audit_event(
|
||
session,
|
||
action="admin.character_template.created",
|
||
character_id=character.id,
|
||
actor="admin",
|
||
decision="auto",
|
||
input_summary=f"管理员创建系统模板 '{payload.key}'",
|
||
resource_type="character_template",
|
||
resource_id=character.id,
|
||
)
|
||
|
||
await session.commit()
|
||
await session.refresh(character)
|
||
return _template_to_response(character)
|
||
|
||
|
||
async def admin_update_template(
|
||
key: str,
|
||
payload: AdminUpdateTemplateRequest,
|
||
) -> AdminTemplateSummaryResponse:
|
||
"""更新系统模板（部分更新）。"""
|
||
async with AsyncSessionLocal() as session:
|
||
template = await character_repo.get_system_template_by_key(session, key)
|
||
if template is None:
|
||
raise AppError("not_found", f"模板 '{key}' 不存在。")
|
||
|
||
changed = False
|
||
update_fields = payload.model_dump(exclude_unset=True)
|
||
|
||
if "name" in update_fields and template.name != update_fields["name"]:
|
||
template.name = update_fields["name"]
|
||
changed = True
|
||
if "display_name" in update_fields and template.display_name != update_fields["display_name"]:
|
||
template.display_name = update_fields["display_name"]
|
||
changed = True
|
||
if "language" in update_fields and template.language != update_fields["language"]:
|
||
template.language = update_fields["language"]
|
||
changed = True
|
||
if "privacy_level" in update_fields and template.privacy_level != update_fields["privacy_level"]:
|
||
template.privacy_level = update_fields["privacy_level"]
|
||
changed = True
|
||
|
||
if "profile" in update_fields:
|
||
current_profile = loads_dict(template.profile_json)
|
||
current_profile.update(update_fields["profile"])
|
||
template.profile_json = dumps(current_profile)
|
||
changed = True
|
||
|
||
if "model_policy" in update_fields:
|
||
current_policy = loads_dict(template.model_policy_json)
|
||
current_policy.update(update_fields["model_policy"])
|
||
template.model_policy_json = dumps(current_policy)
|
||
changed = True
|
||
|
||
if changed:
|
||
await record_audit_event(
|
||
session,
|
||
action="admin.character_template.updated",
|
||
character_id=template.id,
|
||
actor="admin",
|
||
decision="auto",
|
||
input_summary=f"管理员更新系统模板 '{key}'",
|
||
resource_type="character_template",
|
||
resource_id=template.id,
|
||
)
|
||
template.updated_at = utc_now()
|
||
await session.commit()
|
||
await session.refresh(template)
|
||
|
||
return _template_to_response(template)
|
||
|
||
|
||
async def admin_delete_template(key: str) -> None:
|
||
"""删除系统模板。"""
|
||
async with AsyncSessionLocal() as session:
|
||
template = await character_repo.get_system_template_by_key(session, key)
|
||
if template is None:
|
||
raise AppError("not_found", f"模板 '{key}' 不存在。")
|
||
|
||
# 检查是否有关联的子人物
|
||
await _check_template_in_use(session, template)
|
||
|
||
await record_audit_event(
|
||
session,
|
||
action="admin.character_template.deleted",
|
||
character_id=template.id,
|
||
actor="admin",
|
||
decision="auto",
|
||
input_summary=f"管理员删除系统模板 '{key}'",
|
||
resource_type="character_template",
|
||
resource_id=template.id,
|
||
)
|
||
|
||
await session.delete(template)
|
||
await session.commit()
|
||
|
||
|
||
async def _check_template_in_use(session, template: Character) -> None:
|
||
"""检查模板是否被用户人物引用，是则阻止删除。"""
|
||
from sqlalchemy import func, select
|
||
|
||
count = await session.scalar(
|
||
select(func.count())
|
||
.select_from(Character)
|
||
.where(
|
||
Character.source_template_id == template.id,
|
||
Character.is_system_template.is_(False),
|
||
)
|
||
)
|
||
if count and count > 0:
|
||
raise AppError(
|
||
"conflict",
|
||
f"模板 '{template.template_key}' 已被 {count} 个人物引用，无法删除。"
|
||
"请先确保无用户人物引用此模板后再试。",
|
||
)
|
||
|
||
|
||
async def _ensure_birthday(character: Character) -> None:
|
||
"""为模板创建兜底 birthday（profile 中无 birthday 时用 created_at）。"""
|
||
profile = loads_dict(character.profile_json)
|
||
birthday = profile.get("birthday")
|
||
if isinstance(birthday, str) and birthday.strip():
|
||
return
|
||
birthday_iso = character.created_at.strftime("%Y-%m-%d")
|
||
profile["birthday"] = birthday_iso
|
||
character.profile_json = dumps(profile)
|