logs-fixes-debug

resolvers/messages.py

@@ -21,59 +21,61 @@ async def create_message(_, info, chat_id: str, body: str, reply_to=None):
     # Если данных чата нет, возвращаем ошибку
     if not chat_data:
         return {"error": "chat is not exist"}
-    else:
+    elif isinstance(chat_data, str):
         # Преобразование данных чата из строки JSON в словарь
         chat_dict = json.loads(chat_data)
-        print(chat_dict)
+        chat_id = chat_dict["id"]
 
         # Получение ID следующего сообщения
         message_id = await redis.execute("GET", f"chats/{chat_dict['id']}/next_message_id")
-        message_id = int(message_id) if message_id else 0
-        chat_id = chat_dict["id"]
-        # Создание нового сообщения
-        new_message: Message = {
-            "chat_id": chat_id,
-            "id": message_id,
-            "created_by": author_id,
-            "body": body,
-            "created_at": int(time.time()),
-            "updated_at": None,
-        }
+        if isinstance(message_id, str) or isinstance(message_id, int):
+            message_id = int(message_id) if message_id else 0
+            # Создание нового сообщения
+            new_message: Message = {
+                "chat_id": chat_id,
+                "id": message_id,
+                "created_by": author_id,
+                "body": body,
+                "created_at": int(time.time()),
+                "updated_at": None,
+                "reply_to": None
+            }
 
-        # Если есть ответ, добавляем его в сообщение
-        if reply_to:
-            new_message["reply_to"] = reply_to
+            # Если есть ответ, добавляем его в сообщение
+            if reply_to:
+                new_message["reply_to"] = reply_to
 
-        # Обновление времени последнего обновления чата
-        chat_dict["updated_at"] = new_message["created_at"]
+            # Обновление времени последнего обновления чата
+            chat_dict["updated_at"] = new_message["created_at"]
 
-        # Запись обновленных данных чата обратно в Redis
-        await redis.execute("SET", f"chats/{chat_id}", json.dumps(chat_dict))
-        print(f"[inbox] creating message {new_message}")
+            # Запись обновленных данных чата обратно в Redis
+            await redis.execute("SET", f"chats/{chat_id}", json.dumps(chat_dict))
+            print(f"[inbox] creating message {new_message}")
 
-        # Запись нового сообщения в Redis
-        await redis.execute(
-            "SET",
-            f"chats/{chat_id}/messages/{message_id}",
-            json.dumps(new_message),
-        )
+            # Запись нового сообщения в Redis
+            await redis.execute(
+                "SET",
+                f"chats/{chat_id}/messages/{message_id}",
+                json.dumps(new_message),
+            )
 
-        # Добавление ID нового сообщения в список ID сообщений чата
-        await redis.execute("LPUSH", f"chats/{chat_id}/message_ids", str(message_id))
+            # Добавление ID нового сообщения в список ID сообщений чата
+            await redis.execute("LPUSH", f"chats/{chat_id}/message_ids", str(message_id))
 
-        # Обновление ID следующего сообщения
-        await redis.execute("SET", f"chats/{chat_id}/next_message_id", str(message_id + 1))
+            # Обновление ID следующего сообщения
+            await redis.execute("SET", f"chats/{chat_id}/next_message_id", str(message_id + 1))
 
-        # Добавление нового сообщения в список непрочитанных сообщений для каждого участника чата
-        members = chat_dict["members"]
-        for member_id in members:
-            await redis.execute("LPUSH", f"chats/{chat_dict['id']}/unread/{member_id}", str(message_id))
+            # Добавление нового сообщения в список непрочитанных сообщений для каждого участника чата
+            members = chat_dict["members"]
+            for member_id in members:
+                await redis.execute("LPUSH", f"chats/{chat_dict['id']}/unread/{member_id}", str(message_id))
 
-        # Отправка уведомления о новом сообщении
-        new_message["chat_id"] = chat_id
-        await notify_message(new_message, "create")
+            # Отправка уведомления о новом сообщении
+            new_message["chat_id"] = chat_id
+            await notify_message(new_message, "create")
 
-        return {"message": new_message, "error": None}
+            return {"message": new_message, "error": None}
+    return {"error": "cannot create message"}
 
 
 @mutation.field("update_message")
@@ -94,24 +96,24 @@ async def update_message(_, info, message):
         message = await redis.execute("GET", f"chats/{chat_id}/messages/{message_id}")
         if not message:
             return {"error": "message  not exist"}
+        elif isinstance(message, str):
+            message = json.loads(message)
+            if message["created_by"] != author_id:
+                return {"error": "access denied"}
 
-        message = json.loads(message)
-        if message["created_by"] != author_id:
-            return {"error": "access denied"}
+            if body:
+                message["body"] = body
+                message["updated_at"] = int(time.time())
 
-        if body:
-            message["body"] = body
-            message["updated_at"] = int(time.time())
+            await redis.execute("SET", f"chats/{chat_id}/messages/{message_id}", json.dumps(message))
 
-        await redis.execute("SET", f"chats/{chat_id}/messages/{message_id}", json.dumps(message))
+            # Отправка уведомления
+            message["chat_id"] = chat_id
+            await notify_message(message, "update")
 
-        # Отправка уведомления
-        message["chat_id"] = chat_id
-        await notify_message(message, "update")
+            return {"message": message, "error": None}
 
-        return {"message": message, "error": None}
-    else:
-        return {"message": message, "error": "cannot update, no message_id"}
+    return {"message": message, "error": "cannot update"}
 
 
 @mutation.field("delete_message")
@@ -122,24 +124,26 @@ async def delete_message(_, info, chat_id: str, message_id: int):
     chat_str = await redis.execute("GET", f"chats/{chat_id}")
     if not chat_str:
         return {"error": "chat not exist"}
-    chat = json.loads(chat_str)
+    elif isinstance(chat_str, str):
+        chat = json.loads(chat_str)
 
-    message_data = await redis.execute("GET", f"chats/{chat_id}/messages/{str(message_id)}")
-    if not message_data:
-        return {"error": "message  not exist"}
-    message: Message = json.loads(message_data)
-    if message["author"] != author_id:
-        return {"error": "access denied"}
+        message_data = await redis.execute("GET", f"chats/{chat_id}/messages/{str(message_id)}")
+        if not message_data:
+            return {"error": "message  not exist"}
+        elif isinstance(message_data, str):
+            message: Message = json.loads(message_data)
+            if message["created_by"] != author_id:
+                return {"error": "access denied"}
 
-    await redis.execute("LREM", f"chats/{chat_id}/message_ids", 0, str(message_id))
-    await redis.execute("DEL", f"chats/{chat_id}/messages/{str(message_id)}")
+            await redis.execute("LREM", f"chats/{chat_id}/message_ids", 0, str(message_id))
+            await redis.execute("DEL", f"chats/{chat_id}/messages/{str(message_id)}")
 
-    members = chat["members"]
-    for member_id in members:
-        await redis.execute("LREM", f"chats/{chat_id}/unread/{member_id}", 0, str(message_id))
+            members = chat["members"]
+            for member_id in members:
+                await redis.execute("LREM", f"chats/{chat_id}/unread/{member_id}", 0, str(message_id))
 
-    message["chat_id"] = chat_id
-    await notify_message(message, "delete")
+            message["chat_id"] = chat_id
+            await notify_message(message, "delete")
 
     return {}
 
@@ -152,19 +156,20 @@ async def mark_as_read(_, info, chat_id: str, message_id: int):
     chat_str = await redis.execute("GET", f"chats/{chat_id}")
     if not chat_str:
         return {"error": "chat not exist"}
+    if isinstance(chat_str, str):
+        chat = json.loads(chat_str)
+        members = set(chat["members"])
+        if author_id not in members:
+            return {"error": "access denied"}
 
-    chat = json.loads(chat_str)
-    members = set(chat["members"])
-    if author_id not in members:
-        return {"error": "access denied"}
+        await redis.execute("LREM", f"chats/{chat_id}/unread/{author_id}", 0, str(message_id))
 
-    await redis.execute("LREM", f"chats/{chat_id}/unread/{author_id}", 0, str(message_id))
+        message_data = await redis.execute("GET", f"chats/{chat_id}/messages/{str(message_id)}")
+        if not message_data:
+            return {"error": "message  not exist"}
+        elif isinstance(message_data, str):
+            message: Message = json.loads(message_data)
 
-    message_data = await redis.execute("GET", f"chats/{chat_id}/messages/{str(message_id)}")
-    if not message_data:
-        return {"error": "message  not exist"}
-    message: Message = json.loads(message_data)
-
-    await notify_message(message, "seen")
+            await notify_message(message, "seen")
 
     return {"error": None}