70 lines
1.8 KiB
Python
70 lines
1.8 KiB
Python
|
"""
|
||
|
|
章节相关 API 路由
|
||
|
|
"""
|
||
|
|
from typing import List
|
||
|
|
|
||
|
|
from fastapi import APIRouter, Depends, HTTPException
|
||
|
|
from sqlalchemy import select
|
||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
|
|
||
|
|
from database import get_async_db
|
||
|
|
from database.models import Chapter as ChapterModel
|
||
|
|
|
||
|
|
router = APIRouter(prefix="/api/chapters", tags=["chapters"])
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("", response_model=List[dict])
|
||
|
|
async def get_chapters(
|
||
|
|
user_id: str,
|
||
|
|
db: AsyncSession = Depends(get_async_db)
|
||
|
|
):
|
||
|
|
"""获取用户所有章节"""
|
||
|
|
stmt = select(ChapterModel).where(ChapterModel.user_id == user_id).order_by(ChapterModel.order_index)
|
||
|
|
result = await db.execute(stmt)
|
||
|
|
chapters = result.scalars().all()
|
||
|
|
|
||
|
|
return [
|
||
|
|
{
|
||
|
|
"id": ch.id,
|
||
|
|
"title": ch.title,
|
||
|
|
"content": ch.content,
|
||
|
|
"order_index": ch.order_index,
|
||
|
|
"status": ch.status,
|
||
|
|
"category": ch.category,
|
||
|
|
"images": ch.images or []
|
||
|
|
}
|
||
|
|
for ch in chapters
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/{chapter_id}", response_model=dict)
|
||
|
|
async def get_chapter(
|
||
|
|
chapter_id: str,
|
||
|
|
db: AsyncSession = Depends(get_async_db)
|
||
|
|
):
|
||
|
|
"""获取章节详情"""
|
||
|
|
chapter = await db.get(ChapterModel, chapter_id)
|
||
|
|
if not chapter:
|
||
|
|
raise HTTPException(status_code=404, detail="Chapter not found")
|
||
|
|
|
||
|
|
return {
|
||
|
|
"id": chapter.id,
|
||
|
|
"title": chapter.title,
|
||
|
|
"content": chapter.content,
|
||
|
|
"order_index": chapter.order_index,
|
||
|
|
"status": chapter.status,
|
||
|
|
"category": chapter.category,
|
||
|
|
"images": chapter.images or []
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/{chapter_id}/regenerate")
|
||
|
|
async def regenerate_chapter(
|
||
|
|
chapter_id: str,
|
||
|
|
db: AsyncSession = Depends(get_async_db)
|
||
|
|
):
|
||
|
|
"""重新整理章节"""
|
||
|
|
# TODO: 实现重新整理逻辑
|
||
|
|
return {"status": "ok", "message": "Chapter regeneration triggered"}
|
||
|
|
|