Merge pull request #12 from nightbug-xx/x0l35n-codex/add-from_user_email-to-response
Add friend emails to listing
This commit is contained in:
commit
314e95d00f
@ -1,7 +1,12 @@
|
|||||||
from fastapi import APIRouter, Depends, HTTPException
|
from fastapi import APIRouter, Depends, HTTPException
|
||||||
from sqlalchemy.orm import Session
|
from sqlalchemy.orm import Session
|
||||||
from app.core.deps import get_db, get_current_user
|
from app.core.deps import get_db, get_current_user
|
||||||
from app.schemas.friend import FriendRequestCreate, FriendRequestResponse, FriendResponse
|
from app.schemas.friend import (
|
||||||
|
FriendRequestCreate,
|
||||||
|
FriendRequestResponse,
|
||||||
|
FriendResponse,
|
||||||
|
FriendListItem,
|
||||||
|
)
|
||||||
from app.schemas.character import CharacterResponse
|
from app.schemas.character import CharacterResponse
|
||||||
from app.services import friend_service
|
from app.services import friend_service
|
||||||
from app.models.user import User
|
from app.models.user import User
|
||||||
@ -55,7 +60,7 @@ def respond_to_request(
|
|||||||
return {"detail": "요청을 처리했습니다."}
|
return {"detail": "요청을 처리했습니다."}
|
||||||
|
|
||||||
|
|
||||||
@router.get("/list", response_model=list[int])
|
@router.get("/list", response_model=list[FriendListItem])
|
||||||
def get_friend_list(
|
def get_friend_list(
|
||||||
db: Session = Depends(get_db),
|
db: Session = Depends(get_db),
|
||||||
current_user: User = Depends(get_current_user)
|
current_user: User = Depends(get_current_user)
|
||||||
|
|||||||
@ -36,3 +36,11 @@ class FriendResponse(BaseModel):
|
|||||||
|
|
||||||
class Config:
|
class Config:
|
||||||
orm_mode = True
|
orm_mode = True
|
||||||
|
|
||||||
|
|
||||||
|
class FriendListItem(BaseModel):
|
||||||
|
id: int
|
||||||
|
email: str
|
||||||
|
|
||||||
|
class Config:
|
||||||
|
orm_mode = True
|
||||||
|
|||||||
@ -139,7 +139,9 @@ def get_friend_list(db: Session, user_id: int):
|
|||||||
result = []
|
result = []
|
||||||
for f in friends:
|
for f in friends:
|
||||||
friend_id = f.user_id_2 if f.user_id_1 == user_id else f.user_id_1
|
friend_id = f.user_id_2 if f.user_id_1 == user_id else f.user_id_1
|
||||||
result.append(friend_id)
|
friend = db.query(User).filter(User.id == friend_id).first()
|
||||||
|
friend_email = friend.email if friend else None
|
||||||
|
result.append({"id": friend_id, "email": friend_email})
|
||||||
|
|
||||||
return result
|
return result
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user