初始化提交

backend/app/repositories/lorebook.py

@@ -0,0 +1,81 @@
+"""世谱数据访问。"""
+
+from __future__ import annotations
+
+from sqlalchemy import or_, select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.models.lorebook import Lorebook, LorebookEntry
+
+
+async def list_active_books(
+    session: AsyncSession,
+    character_id: str,
+    owner_scope: str,
+) -> list[Lorebook]:
+    """查角色的 active 世谱：character 作用域（按人物隔离）+ global 作用域（跨人物共享）。"""
+    result = await session.execute(
+        select(Lorebook).where(
+            Lorebook.status == "active",
+            Lorebook.owner_scope == owner_scope,
+            or_(Lorebook.character_id == character_id, Lorebook.scope == "global"),
+        )
+    )
+    return list(result.scalars().all())
+
+
+async def list_entries_for_books(
+    session: AsyncSession,
+    lorebook_ids: list[str],
+) -> list[LorebookEntry]:
+    """批量查多本书的启用条目，按 insertion_order 排序。"""
+    if not lorebook_ids:
+        return []
+    result = await session.execute(
+        select(LorebookEntry)
+        .where(
+            LorebookEntry.lorebook_id.in_(lorebook_ids),
+            LorebookEntry.enabled.is_(True),
+        )
+        .order_by(LorebookEntry.insertion_order)
+    )
+    return list(result.scalars().all())
+
+
+async def get_book(session: AsyncSession, lorebook_id: str) -> Lorebook | None:
+    return await session.get(Lorebook, lorebook_id)
+
+
+async def list_books(
+    session: AsyncSession,
+    character_id: str | None,
+    owner_scope: str,
+    *,
+    scope: str | None = None,
+) -> list[Lorebook]:
+    """列出 active 世谱（archived 不返回）。"""
+    conditions = [
+        Lorebook.status == "active",
+        Lorebook.owner_scope == owner_scope,
+    ]
+    if character_id is not None:
+        conditions.append(or_(Lorebook.character_id == character_id, Lorebook.scope == "global"))
+    if scope is not None:
+        conditions.append(Lorebook.scope == scope)
+    result = await session.execute(
+        select(Lorebook).where(*conditions).order_by(Lorebook.created_at.desc())
+    )
+    return list(result.scalars().all())
+
+
+async def list_book_entries(session: AsyncSession, lorebook_id: str) -> list[LorebookEntry]:
+    result = await session.execute(
+        select(LorebookEntry)
+        .where(LorebookEntry.lorebook_id == lorebook_id)
+        .order_by(LorebookEntry.insertion_order)
+    )
+    return list(result.scalars().all())
+
+
+async def get_entry(session: AsyncSession, entry_id: str) -> LorebookEntry | None:
+    return await session.get(LorebookEntry, entry_id)