222 lines
6.8 KiB
Python
222 lines
6.8 KiB
Python
"""
|
||
故事相关API路由
|
||
"""
|
||
import random
|
||
from fastapi import APIRouter, Depends, Query, HTTPException
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
from sqlalchemy import select, update, func, distinct
|
||
from typing import Optional
|
||
from pydantic import BaseModel
|
||
|
||
from app.database import get_db
|
||
from app.models.story import Story, StoryNode, StoryChoice
|
||
|
||
router = APIRouter()
|
||
|
||
|
||
# ========== 请求/响应模型 ==========
|
||
class LikeRequest(BaseModel):
|
||
like: bool
|
||
|
||
|
||
class RewriteRequest(BaseModel):
|
||
ending_name: str
|
||
ending_content: str
|
||
prompt: str
|
||
|
||
|
||
# ========== API接口 ==========
|
||
|
||
@router.get("")
|
||
async def get_stories(
|
||
category: Optional[str] = Query(None),
|
||
featured: bool = Query(False),
|
||
limit: int = Query(20, ge=1, le=100),
|
||
offset: int = Query(0, ge=0),
|
||
db: AsyncSession = Depends(get_db)
|
||
):
|
||
"""获取故事列表"""
|
||
query = select(Story).where(Story.status == 1)
|
||
|
||
if category:
|
||
query = query.where(Story.category == category)
|
||
if featured:
|
||
query = query.where(Story.is_featured == True)
|
||
|
||
query = query.order_by(Story.is_featured.desc(), Story.play_count.desc())
|
||
query = query.limit(limit).offset(offset)
|
||
|
||
result = await db.execute(query)
|
||
stories = result.scalars().all()
|
||
|
||
data = [{
|
||
"id": s.id,
|
||
"title": s.title,
|
||
"cover_url": s.cover_url,
|
||
"description": s.description,
|
||
"category": s.category,
|
||
"play_count": s.play_count,
|
||
"like_count": s.like_count,
|
||
"is_featured": s.is_featured
|
||
} for s in stories]
|
||
|
||
return {"code": 0, "data": data}
|
||
|
||
|
||
@router.get("/hot")
|
||
async def get_hot_stories(
|
||
limit: int = Query(10, ge=1, le=50),
|
||
db: AsyncSession = Depends(get_db)
|
||
):
|
||
"""获取热门故事"""
|
||
query = select(Story).where(Story.status == 1).order_by(Story.play_count.desc()).limit(limit)
|
||
result = await db.execute(query)
|
||
stories = result.scalars().all()
|
||
|
||
data = [{
|
||
"id": s.id,
|
||
"title": s.title,
|
||
"cover_url": s.cover_url,
|
||
"description": s.description,
|
||
"category": s.category,
|
||
"play_count": s.play_count,
|
||
"like_count": s.like_count
|
||
} for s in stories]
|
||
|
||
return {"code": 0, "data": data}
|
||
|
||
|
||
@router.get("/categories")
|
||
async def get_categories(db: AsyncSession = Depends(get_db)):
|
||
"""获取分类列表"""
|
||
query = select(distinct(Story.category)).where(Story.status == 1)
|
||
result = await db.execute(query)
|
||
categories = [row[0] for row in result.all()]
|
||
|
||
return {"code": 0, "data": categories}
|
||
|
||
|
||
@router.get("/{story_id}")
|
||
async def get_story_detail(story_id: int, db: AsyncSession = Depends(get_db)):
|
||
"""获取故事详情（含节点和选项）"""
|
||
# 获取故事
|
||
result = await db.execute(select(Story).where(Story.id == story_id, Story.status == 1))
|
||
story = result.scalar_one_or_none()
|
||
|
||
if not story:
|
||
raise HTTPException(status_code=404, detail="故事不存在")
|
||
|
||
# 获取节点
|
||
nodes_result = await db.execute(
|
||
select(StoryNode).where(StoryNode.story_id == story_id).order_by(StoryNode.sort_order)
|
||
)
|
||
nodes = nodes_result.scalars().all()
|
||
|
||
# 获取选项
|
||
choices_result = await db.execute(
|
||
select(StoryChoice).where(StoryChoice.story_id == story_id).order_by(StoryChoice.sort_order)
|
||
)
|
||
choices = choices_result.scalars().all()
|
||
|
||
# 组装节点和选项
|
||
nodes_map = {}
|
||
for node in nodes:
|
||
nodes_map[node.node_key] = {
|
||
"id": node.id,
|
||
"node_key": node.node_key,
|
||
"content": node.content,
|
||
"speaker": node.speaker,
|
||
"background_image": node.background_image,
|
||
"character_image": node.character_image,
|
||
"bgm": node.bgm,
|
||
"is_ending": node.is_ending,
|
||
"ending_name": node.ending_name,
|
||
"ending_score": node.ending_score,
|
||
"ending_type": node.ending_type,
|
||
"choices": []
|
||
}
|
||
|
||
for choice in choices:
|
||
# 找到对应的节点
|
||
for node in nodes:
|
||
if node.id == choice.node_id and node.node_key in nodes_map:
|
||
nodes_map[node.node_key]["choices"].append({
|
||
"text": choice.text,
|
||
"nextNodeKey": choice.next_node_key,
|
||
"isLocked": choice.is_locked
|
||
})
|
||
break
|
||
|
||
data = {
|
||
"id": story.id,
|
||
"title": story.title,
|
||
"cover_url": story.cover_url,
|
||
"description": story.description,
|
||
"category": story.category,
|
||
"author_id": story.author_id,
|
||
"play_count": story.play_count,
|
||
"like_count": story.like_count,
|
||
"is_featured": story.is_featured,
|
||
"nodes": nodes_map
|
||
}
|
||
|
||
return {"code": 0, "data": data}
|
||
|
||
|
||
@router.post("/{story_id}/play")
|
||
async def record_play(story_id: int, db: AsyncSession = Depends(get_db)):
|
||
"""记录游玩"""
|
||
await db.execute(
|
||
update(Story).where(Story.id == story_id).values(play_count=Story.play_count + 1)
|
||
)
|
||
await db.commit()
|
||
return {"code": 0, "message": "记录成功"}
|
||
|
||
|
||
@router.post("/{story_id}/like")
|
||
async def toggle_like(story_id: int, request: LikeRequest, db: AsyncSession = Depends(get_db)):
|
||
"""点赞/取消点赞"""
|
||
delta = 1 if request.like else -1
|
||
await db.execute(
|
||
update(Story).where(Story.id == story_id).values(like_count=Story.like_count + delta)
|
||
)
|
||
await db.commit()
|
||
return {"code": 0, "message": "点赞成功" if request.like else "取消点赞成功"}
|
||
|
||
|
||
@router.post("/{story_id}/rewrite")
|
||
async def ai_rewrite_ending(story_id: int, request: RewriteRequest, db: AsyncSession = Depends(get_db)):
|
||
"""AI改写结局"""
|
||
if not request.prompt:
|
||
raise HTTPException(status_code=400, detail="请输入改写指令")
|
||
|
||
# 获取故事信息
|
||
result = await db.execute(select(Story).where(Story.id == story_id))
|
||
story = result.scalar_one_or_none()
|
||
|
||
# 模拟AI生成（后续替换为真实API调用）
|
||
templates = [
|
||
f"根据你的愿望「{request.prompt}」，故事有了新的发展...\n\n",
|
||
f"命运的齿轮开始转动，{request.prompt}...\n\n",
|
||
f"在另一个平行世界里，{request.prompt}成为了现实...\n\n"
|
||
]
|
||
|
||
template = random.choice(templates)
|
||
new_content = (
|
||
template +
|
||
"原本的结局被改写，新的故事在这里展开。\n\n" +
|
||
f"【AI改写提示】这是基于「{request.prompt}」生成的新结局。\n" +
|
||
"实际部署时，这里将由AI大模型根据上下文生成更精彩的内容。"
|
||
)
|
||
|
||
return {
|
||
"code": 0,
|
||
"data": {
|
||
"content": new_content,
|
||
"speaker": "旁白",
|
||
"is_ending": True,
|
||
"ending_name": f"{request.ending_name}（改写版）",
|
||
"ending_type": "rewrite"
|
||
}
|
||
}
|