Skip to content

Commit cc74e84

Browse files
committed
feat: 프로필 이미지 url 추가
1 parent 6a03e5d commit cc74e84

1 file changed

Lines changed: 20 additions & 1 deletion

File tree

app/crud/chat.py

Lines changed: 20 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,12 +4,31 @@
44
from app.models.chat_room import ChatRoom
55
from app.models.create_chat_message_request import CreateChatMessageRequest
66
from app.models.create_chat_room_request import CreateChatRoomRequest
7+
from app.crud.s3_bucket import get_class_image
78

89
def find_chat_messages(chat_room_id: int):
910
with connect_database() as db:
1011
messages = db.query(ChatMessage).filter(ChatMessage.chat_room_id == chat_room_id).all()
12+
room = db.query(ChatRoom).filter(ChatRoom.id == chat_room_id).first()
13+
14+
# 채팅방 이름을 기반으로 이미지 URL 생성
15+
image_url = get_class_image(room.name) if room else None
16+
17+
# 각 메시지에 이미지 URL 추가
18+
result = []
19+
for message in messages:
20+
message_dict = {
21+
"id": message.id,
22+
"chat_room_id": message.chat_room_id,
23+
"user_id": message.user_id,
24+
"contents": message.contents,
25+
"created_at": message.created_at,
26+
"updated_at": message.updated_at,
27+
"image": image_url if message.user_id == None else None
28+
}
29+
result.append(message_dict)
1130

12-
return messages
31+
return result
1332

1433
def insert_chat_message(chat_room_id: int, request: CreateChatMessageRequest):
1534
message = ChatMessage(

0 commit comments

Comments
 (0)