信息架构/菜单框架大重构-20260109

backend/app/crud/knowledge_note.py

@@ -0,0 +1,67 @@
+import uuid
+from typing import Sequence
+
+from sqlalchemy import select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.models.knowledge_note import KnowledgeNote
+from app.schemas.knowledge_note import KnowledgeNoteCreate, KnowledgeNoteUpdate
+
+
+async def create_note(
+    db: AsyncSession,
+    study_id: uuid.UUID,
+    note_in: KnowledgeNoteCreate,
+    created_by: uuid.UUID | None,
+) -> KnowledgeNote:
+    note = KnowledgeNote(
+        study_id=study_id,
+        site_name=note_in.site_name,
+        title=note_in.title,
+        content=note_in.content,
+        level=note_in.level,
+        created_by=created_by,
+    )
+    db.add(note)
+    await db.commit()
+    await db.refresh(note)
+    return note
+
+
+async def get_note(db: AsyncSession, note_id: uuid.UUID) -> KnowledgeNote | None:
+    result = await db.execute(select(KnowledgeNote).where(KnowledgeNote.id == note_id))
+    return result.scalar_one_or_none()
+
+
+async def list_notes(
+    db: AsyncSession,
+    study_id: uuid.UUID,
+    site_name: str | None = None,
+    keyword: str | None = None,
+    skip: int = 0,
+    limit: int = 100,
+) -> Sequence[KnowledgeNote]:
+    stmt = select(KnowledgeNote).where(KnowledgeNote.study_id == study_id)
+    if site_name:
+        stmt = stmt.where(KnowledgeNote.site_name.ilike(f"%{site_name}%"))
+    if keyword:
+        stmt = stmt.where(KnowledgeNote.title.ilike(f"%{keyword}%"))
+    stmt = stmt.order_by(KnowledgeNote.updated_at.desc()).offset(skip).limit(limit)
+    result = await db.execute(stmt)
+    return result.scalars().all()
+
+
+async def update_note(
+    db: AsyncSession, note: KnowledgeNote, note_in: KnowledgeNoteUpdate
+) -> KnowledgeNote:
+    update_data = note_in.model_dump(exclude_unset=True)
+    for key, value in update_data.items():
+        setattr(note, key, value)
+    await db.commit()
+    await db.refresh(note)
+    return note
+
+
+async def delete_note(db: AsyncSession, note: KnowledgeNote) -> None:
+    await db.delete(note)
+    await db.commit()