Improve speed of stories parsing

KurimuzonAkuma · KurimuzonAkuma · commit 2bbe5af50ba2 · 2023-11-09T11:43:57.000+03:00
diff --git a/pyrogram/methods/stories/get_all_stories.py b/pyrogram/methods/stories/get_all_stories.py
@@ -56,12 +56,15 @@ async def get_all_stories(
             )
         )
 
+        users = {i.id: i for i in r.users}
+        chats = {i.id: i for i in r.chats}
+
         for peer_story in r.peer_stories:
             for story in peer_story.stories:
                 yield await types.Story._parse(
                     self,
                     story,
-                    {i.id: i for i in r.users},
-                    {i.id: i for i in r.chats},
+                    users,
+                    chats,
                     peer_story.peer
                 )
diff --git a/pyrogram/methods/stories/get_peer_stories.py b/pyrogram/methods/stories/get_peer_stories.py
@@ -59,11 +59,14 @@ async def get_peer_stories(
             )
         )
 
+        users = {i.id: i for i in r.users}
+        chats = {i.id: i for i in r.chats}
+
         for story in r.stories.stories:
             yield await types.Story._parse(
                 self,
                 story,
-                {i.id: i for i in r.users},
-                {i.id: i for i in r.chats},
+                users,
+                chats,
                 r.stories.peer
             )
diff --git a/pyrogram/methods/stories/get_stories.py b/pyrogram/methods/stories/get_stories.py
@@ -69,13 +69,16 @@ async def get_stories(
 
         stories = []
 
+        users = {i.id: i for i in r.users}
+        chats = {i.id: i for i in r.chats}
+
         for story in r.stories:
             stories.append(
                 await types.Story._parse(
                     self,
                     story,
-                    {i.id: i for i in r.users},
-                    {i.id: i for i in r.chats},
+                    users,
+                    chats,
                     peer
                 )
             )
diff --git a/pyrogram/methods/stories/get_stories_archive.py b/pyrogram/methods/stories/get_stories_archive.py
@@ -77,12 +77,15 @@ async def get_stories_archive(
             last = r.stories[-1]
             offset_id = last.id
 
+            users = {i.id: i for i in r.users}
+            chats = {i.id: i for i in r.chats}
+
             for story in r.stories:
                 yield await types.Story._parse(
                     self,
                     story,
-                    {i.id: i for i in r.users},
-                    {i.id: i for i in r.chats},
+                    users,
+                    chats,
                     peer
                 )
 

Original file line number	Diff line number	Diff line change
`@@ -56,12 +56,15 @@ async def get_all_stories(`
`56`	`56`	`)`
`57`	`57`	`)`
`58`	`58`
	`59`	`+ users = {i.id: i for i in r.users}`
	`60`	`+ chats = {i.id: i for i in r.chats}`
	`61`	`+`
`59`	`62`	`for peer_story in r.peer_stories:`
`60`	`63`	`for story in peer_story.stories:`
`61`	`64`	`yield await types.Story._parse(`
`62`	`65`	`self,`
`63`	`66`	`story,`
`64`		`- {i.id: i for i in r.users},`
`65`		`- {i.id: i for i in r.chats},`
	`67`	`+ users,`
	`68`	`+ chats,`
`66`	`69`	`peer_story.peer`
`67`	`70`	`)`
Original file line number	Diff line number	Diff line change
`@@ -59,11 +59,14 @@ async def get_peer_stories(`
`59`	`59`	`)`
`60`	`60`	`)`
`61`	`61`
	`62`	`+ users = {i.id: i for i in r.users}`
	`63`	`+ chats = {i.id: i for i in r.chats}`
	`64`	`+`
`62`	`65`	`for story in r.stories.stories:`
`63`	`66`	`yield await types.Story._parse(`
`64`	`67`	`self,`
`65`	`68`	`story,`
`66`		`- {i.id: i for i in r.users},`
`67`		`- {i.id: i for i in r.chats},`
	`69`	`+ users,`
	`70`	`+ chats,`
`68`	`71`	`r.stories.peer`
`69`	`72`	`)`