171 lines
6.3 KiB
Python
171 lines
6.3 KiB
Python
from fastapi import APIRouter, HTTPException, status
|
|
from pathlib import Path
|
|
from typing import Dict, List
|
|
from backend.core.models.WorldBook import WorldBook
|
|
from backend.core.models.WorldItem import WorldInfoEntry
|
|
|
|
router = APIRouter(prefix="/worldbooks", tags=["worldbooks"])
|
|
|
|
|
|
# ========== 世界书基础路由 ==========
|
|
|
|
@router.get("", response_model=Dict[str, List[Dict]])
|
|
async def list_worldbooks():
|
|
"""获取所有世界书列表"""
|
|
worldbook_dir = Path("data/worldbooks")
|
|
if not worldbook_dir.exists():
|
|
return {"worldbooks": []}
|
|
|
|
worldbooks = []
|
|
for wb_file in worldbook_dir.glob("*.json"):
|
|
try:
|
|
worldbook = WorldBook.from_sillytavern_json(str(wb_file))
|
|
worldbooks.append(worldbook.to_summary_dict())
|
|
except Exception as e:
|
|
print(f"警告: 跳过文件 {wb_file.name}，加载失败: {e}")
|
|
continue
|
|
|
|
return {"worldbooks": worldbooks}
|
|
|
|
|
|
@router.get("/{worldbook_uid}")
|
|
async def get_worldbook(worldbook_uid: str):
|
|
"""获取指定世界书完整内容"""
|
|
worldbook_path = Path("data/worldbooks") / f"{worldbook_uid}.json"
|
|
if not worldbook_path.exists():
|
|
raise HTTPException(status_code=404, detail="WorldBook not found")
|
|
|
|
try:
|
|
worldbook = WorldBook.from_sillytavern_json(str(worldbook_path))
|
|
return worldbook.to_dict()
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"Failed to load worldbook: {str(e)}")
|
|
|
|
|
|
@router.post("", status_code=status.HTTP_201_CREATED)
|
|
async def create_worldbook(worldbook_data: Dict):
|
|
"""创建新世界书"""
|
|
worldbook_dir = Path("data/worldbooks")
|
|
worldbook_dir.mkdir(parents=True, exist_ok=True)
|
|
|
|
worldbook = WorldBook.from_dict(worldbook_data)
|
|
worldbook_path = worldbook_dir / f"{worldbook.uid}.json"
|
|
|
|
if worldbook_path.exists():
|
|
raise HTTPException(status_code=400, detail="WorldBook already exists")
|
|
|
|
worldbook.to_sillytavern_json(str(worldbook_path))
|
|
return {"message": "WorldBook created successfully", "uid": worldbook.uid}
|
|
|
|
|
|
@router.put("/{worldbook_uid}")
|
|
async def update_worldbook(worldbook_uid: str, update_data: Dict):
|
|
"""更新世界书基本信息"""
|
|
worldbook_path = Path("data/worldbooks") / f"{worldbook_uid}.json"
|
|
if not worldbook_path.exists():
|
|
raise HTTPException(status_code=404, detail="WorldBook not found")
|
|
|
|
try:
|
|
worldbook = WorldBook.from_sillytavern_json(str(worldbook_path))
|
|
for key, value in update_data.items():
|
|
if hasattr(worldbook, key):
|
|
setattr(worldbook, key, value)
|
|
worldbook.to_sillytavern_json(str(worldbook_path))
|
|
return {"message": "WorldBook updated successfully"}
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"Failed to update worldbook: {str(e)}")
|
|
|
|
|
|
@router.delete("/{worldbook_uid}")
|
|
async def delete_worldbook(worldbook_uid: str):
|
|
"""删除世界书"""
|
|
worldbook_path = Path("data/worldbooks") / f"{worldbook_uid}.json"
|
|
if not worldbook_path.exists():
|
|
raise HTTPException(status_code=404, detail="WorldBook not found")
|
|
worldbook_path.unlink()
|
|
return {"message": "WorldBook deleted successfully"}
|
|
|
|
|
|
# ========== 世界书条目路由 ==========
|
|
|
|
@router.get("/{worldbook_uid}/entries")
|
|
async def list_worldbook_entries(worldbook_uid: str):
|
|
"""获取世界书所有条目列表"""
|
|
worldbook_path = Path("data/worldbooks") / f"{worldbook_uid}.json"
|
|
if not worldbook_path.exists():
|
|
raise HTTPException(status_code=404, detail="WorldBook not found")
|
|
|
|
try:
|
|
worldbook = WorldBook.from_sillytavern_json(str(worldbook_path))
|
|
return {"entries": [entry.dict() for entry in worldbook.entries]}
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"Failed to load entries: {str(e)}")
|
|
|
|
|
|
@router.get("/{worldbook_uid}/entries/{entry_uid}")
|
|
async def get_worldbook_entry(worldbook_uid: str, entry_uid: str):
|
|
"""获取指定条目详情"""
|
|
worldbook_path = Path("data/worldbooks") / f"{worldbook_uid}.json"
|
|
if not worldbook_path.exists():
|
|
raise HTTPException(status_code=404, detail="WorldBook not found")
|
|
|
|
try:
|
|
worldbook = WorldBook.from_sillytavern_json(str(worldbook_path))
|
|
entry = worldbook.get_entry(entry_uid)
|
|
if not entry:
|
|
raise HTTPException(status_code=404, detail="Entry not found")
|
|
return entry.dict()
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"Failed to load entry: {str(e)}")
|
|
|
|
|
|
@router.post("/{worldbook_uid}/entries", status_code=status.HTTP_201_CREATED)
|
|
async def add_worldbook_entry(worldbook_uid: str, entry_data: Dict):
|
|
"""向世界书添加新条目"""
|
|
worldbook_path = Path("data/worldbooks") / f"{worldbook_uid}.json"
|
|
if not worldbook_path.exists():
|
|
raise HTTPException(status_code=404, detail="WorldBook not found")
|
|
|
|
try:
|
|
worldbook = WorldBook.from_sillytavern_json(str(worldbook_path))
|
|
entry = WorldInfoEntry(**entry_data)
|
|
worldbook.add_entry(entry)
|
|
worldbook.to_sillytavern_json(str(worldbook_path))
|
|
return {"message": "Entry added successfully", "entry_uid": entry.uid}
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"Failed to add entry: {str(e)}")
|
|
|
|
|
|
@router.put("/{worldbook_uid}/entries/{entry_uid}")
|
|
async def update_worldbook_entry(worldbook_uid: str, entry_uid: str, update_data: Dict):
|
|
"""更新指定条目"""
|
|
worldbook_path = Path("data/worldbooks") / f"{worldbook_uid}.json"
|
|
if not worldbook_path.exists():
|
|
raise HTTPException(status_code=404, detail="WorldBook not found")
|
|
|
|
try:
|
|
worldbook = WorldBook.from_sillytavern_json(str(worldbook_path))
|
|
if not worldbook.update_entry(entry_uid, **update_data):
|
|
raise HTTPException(status_code=404, detail="Entry not found")
|
|
worldbook.to_sillytavern_json(str(worldbook_path))
|
|
return {"message": "Entry updated successfully"}
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"Failed to update entry: {str(e)}")
|
|
|
|
|
|
@router.delete("/{worldbook_uid}/entries/{entry_uid}")
|
|
async def delete_worldbook_entry(worldbook_uid: str, entry_uid: str):
|
|
"""从世界书删除指定条目"""
|
|
worldbook_path = Path("data/worldbooks") / f"{worldbook_uid}.json"
|
|
if not worldbook_path.exists():
|
|
raise HTTPException(status_code=404, detail="WorldBook not found")
|
|
|
|
try:
|
|
worldbook = WorldBook.from_sillytavern_json(str(worldbook_path))
|
|
if not worldbook.remove_entry(entry_uid):
|
|
raise HTTPException(status_code=404, detail="Entry not found")
|
|
worldbook.to_sillytavern_json(str(worldbook_path))
|
|
return {"message": "Entry deleted successfully"}
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"Failed to delete entry: {str(e)}")
|