初始化提交

backend/app/repositories/personality.py

@@ -0,0 +1,84 @@
+"""人格数据访问。"""
+
+from sqlalchemy import func, select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.models.personality import PersonalityProposal, PersonalityVersion
+
+
+async def get_personality(session: AsyncSession, personality_id: str) -> PersonalityVersion | None:
+    return await session.get(PersonalityVersion, personality_id)
+
+
+async def get_latest_personality(
+    session: AsyncSession, character_id: str
+) -> PersonalityVersion | None:
+    result = await session.execute(
+        select(PersonalityVersion)
+        .where(PersonalityVersion.character_id == character_id)
+        .order_by(PersonalityVersion.version.desc())
+        .limit(1)
+    )
+    return result.scalar_one_or_none()
+
+
+async def list_personality_versions(
+    session: AsyncSession,
+    character_id: str,
+    *,
+    page: int = 1,
+    page_size: int = 20,
+) -> tuple[list[PersonalityVersion], int]:
+    """按版本号倒序分页返回某人物的人格版本历史，返回 (条目, 总数)。"""
+    cond = PersonalityVersion.character_id == character_id
+    total = int(
+        (
+            await session.execute(select(func.count()).select_from(PersonalityVersion).where(cond))
+        ).scalar_one()
+    )
+    result = await session.execute(
+        select(PersonalityVersion)
+        .where(cond)
+        .order_by(PersonalityVersion.version.desc())
+        .offset((page - 1) * page_size)
+        .limit(page_size)
+    )
+    return list(result.scalars().all()), total
+
+
+async def add_proposal(session: AsyncSession, proposal: PersonalityProposal) -> PersonalityProposal:
+    session.add(proposal)
+    await session.flush()
+    return proposal
+
+
+async def get_proposal(session: AsyncSession, proposal_id: str) -> PersonalityProposal | None:
+    return await session.get(PersonalityProposal, proposal_id)
+
+
+async def list_proposals(
+    session: AsyncSession,
+    *,
+    character_id: str | None = None,
+    status: str | None = None,
+    page: int = 1,
+    page_size: int = 20,
+) -> tuple[list[PersonalityProposal], int]:
+    """按创建时间倒序分页返回人格修改建议；可选按人物/状态过滤，返回 (条目, 总数)。"""
+    conds = []
+    if character_id is not None:
+        conds.append(PersonalityProposal.character_id == character_id)
+    if status is not None:
+        conds.append(PersonalityProposal.status == status)
+    count_stmt = select(func.count()).select_from(PersonalityProposal)
+    list_stmt = select(PersonalityProposal)
+    if conds:
+        count_stmt = count_stmt.where(*conds)
+        list_stmt = list_stmt.where(*conds)
+    total = int((await session.execute(count_stmt)).scalar_one())
+    result = await session.execute(
+        list_stmt.order_by(PersonalityProposal.created_at.desc())
+        .offset((page - 1) * page_size)
+        .limit(page_size)
+    )
+    return list(result.scalars().all()), total