初始化提交

backend/app/repositories/candidates.py

@@ -0,0 +1,70 @@
+"""候选数据访问（api-spec §8；档案 / 记忆候选）。"""
+
+from sqlalchemy import select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.models.candidates import MemoryCandidate, ProfileCandidate
+
+
+async def add_profile_candidates(session: AsyncSession, candidates: list[ProfileCandidate]) -> None:
+    session.add_all(candidates)
+    await session.flush()
+
+
+async def add_memory_candidates(session: AsyncSession, candidates: list[MemoryCandidate]) -> None:
+    session.add_all(candidates)
+    await session.flush()
+
+
+async def list_profile_candidates(
+    session: AsyncSession,
+    character_id: str,
+    *,
+    import_job_id: str | None = None,
+    status: str | None = None,
+) -> list[ProfileCandidate]:
+    conditions = [ProfileCandidate.character_id == character_id]
+    if import_job_id is not None:
+        conditions.append(ProfileCandidate.import_job_id == import_job_id)
+    if status is not None:
+        conditions.append(ProfileCandidate.status == status)
+    result = await session.execute(
+        select(ProfileCandidate).where(*conditions).order_by(ProfileCandidate.created_at.desc())
+    )
+    return list(result.scalars().all())
+
+
+async def list_memory_candidates(
+    session: AsyncSession,
+    character_id: str,
+    *,
+    import_job_id: str | None = None,
+    cycle_id: str | None = None,
+    status: str | None = None,
+    source: str | None = None,
+) -> list[MemoryCandidate]:
+    conditions = [MemoryCandidate.character_id == character_id]
+    if import_job_id is not None:
+        conditions.append(MemoryCandidate.import_job_id == import_job_id)
+    if cycle_id is not None:
+        conditions.append(MemoryCandidate.cycle_id == cycle_id)
+    if status is not None:
+        conditions.append(MemoryCandidate.status == status)
+    if source == "cycle":
+        conditions.append(MemoryCandidate.import_job_id.is_(None))
+    elif source == "import":
+        conditions.append(MemoryCandidate.import_job_id.is_not(None))
+    result = await session.execute(
+        select(MemoryCandidate).where(*conditions).order_by(MemoryCandidate.created_at.desc())
+    )
+    return list(result.scalars().all())
+
+
+async def get_profile_candidate(
+    session: AsyncSession, candidate_id: str
+) -> ProfileCandidate | None:
+    return await session.get(ProfileCandidate, candidate_id)
+
+
+async def get_memory_candidate(session: AsyncSession, candidate_id: str) -> MemoryCandidate | None:
+    return await session.get(MemoryCandidate, candidate_id)