File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 44from app .models .chat_room import ChatRoom
55from app .models .create_chat_message_request import CreateChatMessageRequest
66from app .models .create_chat_room_request import CreateChatRoomRequest
7+ from app .crud .s3_bucket import get_class_image
78
89def 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
1433def insert_chat_message (chat_room_id : int , request : CreateChatMessageRequest ):
1534 message = ChatMessage (
You can’t perform that action at this time.
0 commit comments