load-recipients-fix-2

2023-12-18 21:32:49 +03:00
parent 93c1727be3
commit 9d1a4e90c9
3 changed files with 41 additions and 32 deletions
--- a/resolvers/load.py
+++ b/resolvers/load.py
@@ -52,36 +52,37 @@ async def load_messages(
@login_required
 async def load_chats(_, info, limit: int = 50, offset: int = 0) -> Dict[str, Union[List[Dict[str, Any]], None]]:
    """load :limit chats of current user with :offset"""
-    author_id = info.context["author_id"]
-    cids = await redis.execute("SMEMBERS", f"chats_by_author/{author_id}")
-    chats = []
-    if cids:
-        members_online = (await redis.execute("SMEMBERS", "authors-online")) or []
-        cids = list(cids)[offset : (offset + limit)]
-        lock = asyncio.Lock()
-        if len(cids) == 0:
-            print(f"[resolvers.load] no chats for user with id={author_id}")
-            r = await create_chat(None, info, members=[1])  # member with id = 1 is discours
-            print(f"[resolvers.load] created chat: {r['chat_id']}")
-            cids.append(r["chat"]["id"])
-        all_authors: List[ChatMember] = await get_all_authors()
-        authors = {a["id"]: a for a in all_authors}
-        for cid in cids:
-            async with lock:
-                chat_str: str = await redis.execute("GET", f"chats/{cid}")
-                print(f"[resolvers.load] redis GET by {cid}: {chat_str}")
-                if chat_str:
-                    c: ChatPayload = json.loads(chat_str)
-                    c["messages"] = await load_messages(cid, 5, 0)
-                    c["unread"] = await get_unread_counter(cid, author_id)
-                    member_ids = c["members"].copy()
-                    c["members"] = []
-                    for member_id in member_ids:
-                        a = authors.get(member_id)
-                        if a:
-                            a["online"] = a.get("id") in members_online
-                            c["members"].append(a)
-                    chats.append(c)
+    author_id = info.context.get("author_id")
+    if author_id:
+        cids = await redis.execute("SMEMBERS", f"chats_by_author/{author_id}")
+        chats = []
+        if cids:
+            members_online = (await redis.execute("SMEMBERS", "authors-online")) or []
+            cids = list(cids)[offset : (offset + limit)]
+            lock = asyncio.Lock()
+            if len(cids) == 0:
+                print(f"[resolvers.load] no chats for user with id={author_id}")
+                r = await create_chat(None, info, members=[1])  # member with id = 1 is discours
+                print(f"[resolvers.load] created chat: {r['chat_id']}")
+                cids.append(r["chat"]["id"])
+            all_authors: List[ChatMember] = await get_all_authors()
+            authors = {a["id"]: a for a in all_authors}
+            for cid in cids:
+                async with lock:
+                    chat_str: str = await redis.execute("GET", f"chats/{cid}")
+                    print(f"[resolvers.load] redis GET by {cid}: {chat_str}")
+                    if chat_str:
+                        c: ChatPayload = json.loads(chat_str)
+                        c["messages"] = await load_messages(cid, 5, 0)
+                        c["unread"] = await get_unread_counter(cid, author_id)
+                        member_ids = c["members"].copy()
+                        c["members"] = []
+                        for member_id in member_ids:
+                            a = authors.get(member_id)
+                            if a:
+                                a["online"] = a.get("id") in members_online
+                                c["members"].append(a)
+                        chats.append(c)
    return {"chats": chats, "error": None}