feat(mvp): complete M13 clean-up milestone (real usernames, avatar studio redesign, pause menu, audio manager)
checkpoint / checkpoint (push) Successful in 6m22s
checkpoint / checkpoint (push) Successful in 6m22s
This commit is contained in:
@@ -46,7 +46,8 @@ DbDep = Annotated[AsyncSession, Depends(get_db)]
|
||||
|
||||
|
||||
class FriendRequestCreate(BaseModel):
|
||||
user_id: int
|
||||
user_id: int | None = None
|
||||
username: str | None = None
|
||||
|
||||
|
||||
class FriendRequestResponse(BaseModel):
|
||||
@@ -55,6 +56,7 @@ class FriendRequestResponse(BaseModel):
|
||||
addressee_id: int
|
||||
status: FriendshipStatus
|
||||
created_at: datetime
|
||||
requester_username: str | None = None
|
||||
|
||||
model_config = {"from_attributes": True}
|
||||
|
||||
@@ -132,7 +134,18 @@ async def _get_user_or_404(db: AsyncSession, user_id: int) -> User:
|
||||
@router.post("/requests", response_model=FriendRequestResponse, status_code=status.HTTP_201_CREATED)
|
||||
async def send_friend_request(body: FriendRequestCreate, current_user: CurrentUser, db: DbDep):
|
||||
"""Send a friend request to another user."""
|
||||
target_id = body.user_id
|
||||
target_id: int
|
||||
if body.user_id is not None and body.user_id > 0:
|
||||
target_id = body.user_id
|
||||
elif body.username is not None and body.username.strip() != "":
|
||||
u_result = await db.execute(select(User).where(User.username == body.username.strip()))
|
||||
user = u_result.scalar_one_or_none()
|
||||
if user is None:
|
||||
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="User not found")
|
||||
target_id = user.id
|
||||
else:
|
||||
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail="user_id or username required")
|
||||
|
||||
if target_id == current_user.id:
|
||||
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail="Cannot send a friend request to yourself")
|
||||
|
||||
@@ -153,19 +166,39 @@ async def send_friend_request(body: FriendRequestCreate, current_user: CurrentUs
|
||||
await db.rollback()
|
||||
raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="Friend request or friendship already exists")
|
||||
await db.refresh(fr)
|
||||
return fr
|
||||
return FriendRequestResponse(
|
||||
id=fr.id,
|
||||
requester_id=fr.requester_id,
|
||||
addressee_id=fr.addressee_id,
|
||||
status=fr.status,
|
||||
created_at=fr.created_at,
|
||||
requester_username=current_user.username,
|
||||
)
|
||||
|
||||
|
||||
@router.get("/requests", response_model=FriendRequestListResponse)
|
||||
async def list_incoming_requests(current_user: CurrentUser, db: DbDep):
|
||||
"""List pending friend requests sent *to* the authenticated user."""
|
||||
result = await db.execute(
|
||||
select(Friendship).where(
|
||||
select(Friendship, User.username)
|
||||
.join(User, Friendship.requester_id == User.id)
|
||||
.where(
|
||||
Friendship.addressee_id == current_user.id,
|
||||
Friendship.status == FriendshipStatus.PENDING,
|
||||
)
|
||||
)
|
||||
return FriendRequestListResponse(requests=list(result.scalars().all()))
|
||||
requests = [
|
||||
FriendRequestResponse(
|
||||
id=fr.id,
|
||||
requester_id=fr.requester_id,
|
||||
addressee_id=fr.addressee_id,
|
||||
status=fr.status,
|
||||
created_at=fr.created_at,
|
||||
requester_username=uname,
|
||||
)
|
||||
for fr, uname in result.all()
|
||||
]
|
||||
return FriendRequestListResponse(requests=requests)
|
||||
|
||||
|
||||
@router.post("/requests/{request_id}/accept", response_model=FriendRequestResponse)
|
||||
|
||||
Reference in New Issue
Block a user