首个可运行的版本
This commit is contained in:
@@ -0,0 +1,88 @@
|
||||
"""好友路由"""
|
||||
|
||||
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.friend import FriendRequestCreate, FriendRead, FriendRequestRead
|
||||
from app.services.friend_service import FriendService
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
|
||||
@router.get("/", response_model=list[dict])
|
||||
async def list_friends(
|
||||
user: User = Depends(get_current_user),
|
||||
db: AsyncSession = Depends(get_db),
|
||||
):
|
||||
"""获取好友列表"""
|
||||
service = FriendService(db)
|
||||
return await service.get_friends(user.id)
|
||||
|
||||
|
||||
@router.get("/requests", response_model=list[dict])
|
||||
async def list_requests(
|
||||
user: User = Depends(get_current_user),
|
||||
db: AsyncSession = Depends(get_db),
|
||||
):
|
||||
"""获取待处理的好友请求"""
|
||||
service = FriendService(db)
|
||||
return await service.get_pending_requests(user.id)
|
||||
|
||||
|
||||
@router.post("/request")
|
||||
async def send_friend_request(
|
||||
req: FriendRequestCreate,
|
||||
user: User = Depends(get_current_user),
|
||||
db: AsyncSession = Depends(get_db),
|
||||
):
|
||||
"""发送好友请求"""
|
||||
service = FriendService(db)
|
||||
try:
|
||||
await service.send_request(user.id, req.to_user_id, req.message)
|
||||
return {"success": True, "message": "好友请求已发送"}
|
||||
except ValueError as e:
|
||||
raise HTTPException(status_code=400, detail=str(e))
|
||||
|
||||
|
||||
@router.put("/request/{request_id}/accept")
|
||||
async def accept_friend_request(
|
||||
request_id: str,
|
||||
user: User = Depends(get_current_user),
|
||||
db: AsyncSession = Depends(get_db),
|
||||
):
|
||||
"""接受好友请求"""
|
||||
service = FriendService(db)
|
||||
try:
|
||||
await service.accept_request(request_id, user.id)
|
||||
return {"success": True, "message": "已添加好友"}
|
||||
except ValueError as e:
|
||||
raise HTTPException(status_code=400, detail=str(e))
|
||||
|
||||
|
||||
@router.put("/request/{request_id}/reject")
|
||||
async def reject_friend_request(
|
||||
request_id: str,
|
||||
user: User = Depends(get_current_user),
|
||||
db: AsyncSession = Depends(get_db),
|
||||
):
|
||||
"""拒绝好友请求"""
|
||||
service = FriendService(db)
|
||||
try:
|
||||
await service.reject_request(request_id, user.id)
|
||||
return {"success": True, "message": "已拒绝"}
|
||||
except ValueError as e:
|
||||
raise HTTPException(status_code=400, detail=str(e))
|
||||
|
||||
|
||||
@router.delete("/{friend_id}")
|
||||
async def remove_friend(
|
||||
friend_id: str,
|
||||
user: User = Depends(get_current_user),
|
||||
db: AsyncSession = Depends(get_db),
|
||||
):
|
||||
"""删除好友"""
|
||||
service = FriendService(db)
|
||||
await service.remove_friend(user.id, friend_id)
|
||||
return {"success": True, "message": "已删除好友"}
|
||||
Reference in New Issue
Block a user