52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
"""时光胶囊路由"""
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.dependencies import get_db, get_current_user
|
|
from app.models.user import User
|
|
from app.schemas.capsule import CapsuleCreate
|
|
from app.services.capsule_service import CapsuleService
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/")
|
|
async def list_capsules(
|
|
user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""获取我的时光胶囊"""
|
|
service = CapsuleService(db)
|
|
return await service.get_capsules(user.id)
|
|
|
|
|
|
@router.get("/{capsule_id}")
|
|
async def get_capsule(
|
|
capsule_id: str,
|
|
user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""获取单个胶囊"""
|
|
service = CapsuleService(db)
|
|
try:
|
|
return await service.get_capsule(user.id, capsule_id)
|
|
except ValueError as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|
|
|
|
|
|
@router.post("/")
|
|
async def create_capsule(
|
|
req: CapsuleCreate,
|
|
user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""创建时光胶囊"""
|
|
service = CapsuleService(db)
|
|
try:
|
|
return await service.create_capsule(
|
|
user.id, req.recipient_id, req.title, req.content, req.unlock_at, req.mood
|
|
)
|
|
except ValueError as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|