62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
|
"""
|
|||
|
|
回忆录状态 API
|
|||
|
|
"""
|
|||
|
|
from fastapi import APIRouter, Depends
|
|||
|
|
from sqlalchemy import select
|
|||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|||
|
|
|
|||
|
|
from database import get_async_db
|
|||
|
|
from database.models import Book as BookModel
|
|||
|
|
from database.models import Chapter as ChapterModel
|
|||
|
|
from database.models import User as UserModel
|
|||
|
|
from middleware.auth import get_current_user
|
|||
|
|
from services.memoir_state_service import get_or_create_state
|
|||
|
|
|
|||
|
|
router = APIRouter(prefix="/api/memoir-state", tags=["memoir-state"])
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.get("")
|
|||
|
|
async def get_memoir_state(
|
|||
|
|
current_user: UserModel = Depends(get_current_user),
|
|||
|
|
db: AsyncSession = Depends(get_async_db),
|
|||
|
|
):
|
|||
|
|
"""获取当前用户回忆录状态"""
|
|||
|
|
state = await get_or_create_state(current_user.id, db)
|
|||
|
|
return state.model_dump()
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.get("/next-question")
|
|||
|
|
async def get_next_question_context(
|
|||
|
|
current_user: UserModel = Depends(get_current_user),
|
|||
|
|
db: AsyncSession = Depends(get_async_db),
|
|||
|
|
):
|
|||
|
|
"""获取下一步问题的上下文（当前阶段与空 slot）"""
|
|||
|
|
state = await get_or_create_state(current_user.id, db)
|
|||
|
|
return {
|
|||
|
|
"current_stage": state.current_stage,
|
|||
|
|
"empty_slots": state.empty_slots_for_current_stage(),
|
|||
|
|
"covered_stages": state.covered_stages,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.post("/mark-read")
|
|||
|
|
async def mark_memoir_read(
|
|||
|
|
current_user: UserModel = Depends(get_current_user),
|
|||
|
|
db: AsyncSession = Depends(get_async_db),
|
|||
|
|
):
|
|||
|
|
"""标记回忆录更新已读"""
|
|||
|
|
stmt = select(ChapterModel).where(ChapterModel.user_id == current_user.id, ChapterModel.is_new == True)
|
|||
|
|
result = await db.execute(stmt)
|
|||
|
|
chapters = result.scalars().all()
|
|||
|
|
for chapter in chapters:
|
|||
|
|
chapter.is_new = False
|
|||
|
|
|
|||
|
|
stmt_book = select(BookModel).where(BookModel.user_id == current_user.id).order_by(BookModel.updated_at.desc())
|
|||
|
|
result_book = await db.execute(stmt_book)
|
|||
|
|
book = result_book.scalar_one_or_none()
|
|||
|
|
if book:
|
|||
|
|
book.has_update = False
|
|||
|
|
|
|||
|
|
await db.commit()
|
|||
|
|
return {"status": "ok"}
|