将知识库笔记迁移为注意事项

2026-05-28 10:52:05 +08:00
parent 8da7fc715c
commit d2b41ae454
13 changed files with 479 additions and 281 deletions
@@ -1,70 +0,0 @@
-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()
@@ -0,0 +1,70 @@
+import uuid
+from typing import Sequence
+
+from sqlalchemy import select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.models.precaution import Precaution
+from app.schemas.precaution import PrecautionCreate, PrecautionUpdate
+
+
+async def create_precaution(
+    db: AsyncSession,
+    study_id: uuid.UUID,
+    precaution_in: PrecautionCreate,
+    created_by: uuid.UUID | None,
+) -> Precaution:
+    precaution = Precaution(
+        study_id=study_id,
+        site_name=precaution_in.site_name,
+        title=precaution_in.title,
+        content=precaution_in.content,
+        level=precaution_in.level,
+        created_by=created_by,
+    )
+    db.add(precaution)
+    await db.commit()
+    await db.refresh(precaution)
+    return precaution
+
+
+async def get_precaution(db: AsyncSession, precaution_id: uuid.UUID) -> Precaution | None:
+    result = await db.execute(select(Precaution).where(Precaution.id == precaution_id))
+    return result.scalar_one_or_none()
+
+
+async def list_precautions(
+    db: AsyncSession,
+    study_id: uuid.UUID,
+    site_name: str | None = None,
+    keyword: str | None = None,
+    skip: int = 0,
+    limit: int = 100,
+) -> Sequence[Precaution]:
+    stmt = (
+        select(Precaution)
+        .where(Precaution.study_id == study_id)
+    )
+    if site_name:
+        stmt = stmt.where(Precaution.site_name.ilike(f"%{site_name}%"))
+    if keyword:
+        stmt = stmt.where(Precaution.title.ilike(f"%{keyword}%"))
+    stmt = stmt.order_by(Precaution.updated_at.desc()).offset(skip).limit(limit)
+    result = await db.execute(stmt)
+    return result.scalars().all()
+
+
+async def update_precaution(
+    db: AsyncSession, precaution: Precaution, precaution_in: PrecautionUpdate
+) -> Precaution:
+    update_data = precaution_in.model_dump(exclude_unset=True)
+    for key, value in update_data.items():
+        setattr(precaution, key, value)
+    await db.commit()
+    await db.refresh(precaution)
+    return precaution
+
+
+async def delete_precaution(db: AsyncSession, precaution: Precaution) -> None:
+    await db.delete(precaution)
+    await db.commit()