diff --git a/backend/rooms.py b/backend/rooms.py index 18fd35a..48a3dec 100644 --- a/backend/rooms.py +++ b/backend/rooms.py @@ -23,7 +23,42 @@ def create_room( ) ) - return room_id + query2 = """ + SELECT id + FROM rooms + WHERE invite_code = ? + """ + room = fetch_one( + query2, + ( + invite_code, + ) + ) + + if room is None: + return None + + room_id = room["id"] + role = DEFAULT_ROLE + query3 = """ + INSERT INTO "main"."room_members" + ("room_id", "user_id", "role") + VALUES (?, ?, ?); + """ + + member_id = execute( + query3, + ( + room_id, + creator_id, + role + ) + ) + + return { + "room_id": room_id, + "member_id": member_id + } def search_room( invite_code @@ -43,7 +78,7 @@ def search_room( if room is None: return None - return room_name["name"] + return room["name"] def accept_room(