o Yg @sddlmZmZddlmZddlmZmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZd ed ed eefd dZd ed ed eefddZd ede ded efddZd d ed ededed ef ddZd edededed ef ddZd ededed efddZdS)!)ListDict)Session)or_case)datetime) Friendship)User) UserReferral)message_service) HTTPExceptiondbuser_idreturnc Csttj|ktjftjdd}|ttjtjtj | ttj|ktj tjk@tj|ktj tjk@B tj dk}i}|D]\}}}}} | |vr]|j | ||||j |j|jdd || <q@t|S)u 기존 Friendship 테이블에서 직접 추가된 친구 목록을 조회합니다. computed_friend_id를 기준으로 중복 제거합니다. else_computed_friend_idaccepted friendship) id friend_id first_name last_name profile_imagestatus created_at updated_atsource)rrrrlabelqueryr rrrjoinrfilterrallrrlistvalues) r rrfriendship_rows friend_maprrrrcomp_friend_idr('/app/app/services/friendship_service.pyget_friend_list sJ   r*c Csttj|ktjftjdd}|ttjtjtj tjtj| tt tj|ktj tjk@tj|ktj tjk@ tjgd}i}|D]%\}}}}} } } | |k} | |vrm|j | | | ||||j|j|j| d || <qHt|S)z Retrieve the friend list and include additional information: - requested_by / requested_to (who sent/received the friend request) - is_request_received flag (true if the logged in user is the recipient) rr)pendingrblocked) r requested_by requested_torrrrrrris_request_received)rrrrrrr rrrr rrr!rin_r"rrr#r$) r rrr%r&rrrrr-r.r'r/r(r(r)get_extended_friend_list7sV   r1rcurrent_user_idc Csx|j}|j}||kr |n|}||k}|ttj|k}|s'tddd|j||||j|j |j |j ||j |j d S)a= Serialize a Friendship instance so the response includes: - requested_by: original requester (friendship.user_id) - requested_to: recipient (friendship.friend_id) - is_request_received: whether the current user is the recipient. - friend_id: the 'other' user's id (convenience field) zFriend not found status_codedetail) rr-r.rrrrrr/rr)rrrr r!rfirstr rrrrrr)r rr2r-r.rr/Zfriendr(r(r)_serialize_friendshipks& r8rrrcCs|tttj|ktj|k@tj|ktj|k@}|r%t|||St|||d}||| | |t|||S)u 두 사용자 간의 친구 관계를 생성합니다. 이미 동일한 관계가 존재하면 해당 레코드를 직렬화하여 반환합니다. )rrr) rrr!rrrr7r8addcommitrefresh)r rrrexistingZnew_friendshipr(r(r)create_friendships&     r= new_statuscCsn|tttj|ktj|k@tj|ktj|k@}|s%tddd||_| | |t |||S)u4 현재 사용자와 friend_id 간의 친구 관계 상태를 변경하고, 응답으로 FriendsResponse에 정의된 필드를 포함한 딕셔너리를 반환합니다. new_status는 "accepted", "blocked", (또는 enum에 추가할 경우 "rejected") 등으로 지정할 수 있습니다. r3Friendship not foundr4 rrr!rrrr7r rr:r;r8)r r2rr>rr(r(r)update_friendship_statuss    rAcCsn|tttj|ktj|k@tj|ktj|k@}|s%tdddd|_| | |t |||S)u 현재 사용자와 friend_id 간의 친구 관계를 삭제하는 대신, 상태를 'blocked'로 업데이트하며, 응답으로 FriendshipResponse 스키마에 맞는 데이터를 포함한 딕셔너리를 반환합니다. r3r?r4r,r@)r r2rrr(r(r)delete_friendships    rBN)r)typingrrsqlalchemy.ormr sqlalchemyrrrZapp.models.friendshiprapp.models.userr app.models.user_referralr app.servicesr fastapir intr*dictr1r8strr=rArBr(r(r(r)s       ,4 "