Add get_stories to get story information from identifiers (#47)

Author: SpEcHiDe

compiler/docs/compiler.py

@@ -359,6 +359,10 @@ def get_title_list(s: str) -> list:
             get_message_effects
             get_stickers
         """,
+        stories="""
+        Stories
+            get_stories
+        """,
         users="""
         Users
             get_me

compiler/docs/template/methods.rst

@@ -151,6 +151,19 @@ Stickers
 
     {stickers}
 
+Stories
+--------
+
+.. autosummary::
+    :nosignatures:
+
+    {stories}
+
+.. toctree::
+    :hidden:
+
+    {stories}
+
 Password
 --------
 

docs/source/releases/changes-in-this-fork.rst

@@ -14,6 +14,8 @@ If you found any issue or have any suggestions, feel free to make `an issue <htt
 | Scheme layer used: 183 |
 +------------------------+
 
+- Added :meth:`~pyrogram.Client.get_stories`.
+- Added :obj:`~pyrogram.filters.thread` and :obj:`~pyrogram.filters.self_destruct`.
 - Added the field ``can_send_paid_media`` to the class :obj:`~pyrogram.types.Chat`.
 - Added support for launching Web Apps via ``t.me`` link in the class :obj:`~pyrogram.types.MenuButtonWebApp`.
 - `View new and changed raw API methods <https://telegramplayground.github.io/TG-APIs/TL/diff/tdesktop.html?from=181&to=183>`__.

pyrogram/dispatcher.py

@@ -200,7 +200,8 @@ async def story_parser(update, users, chats):
                     users,
                     chats,
                     None, None,
-                    update
+                    update,
+                    None, None
                 ),
                 StoryHandler
             )

pyrogram/methods/__init__.py

@@ -28,6 +28,7 @@
 from .password import Password
 from .phone import Phone
 from .stickers import Stickers
+from .stories import Stories
 from .users import Users
 from .utilities import Utilities
 from .business import TelegramBusiness
@@ -46,6 +47,7 @@ class Methods(
     Password,
     Phone,
     Stickers,
+    Stories,
     TelegramBusiness,
     Users,
     Utilities,

pyrogram/methods/stories/__init__.py

@@ -0,0 +1,25 @@
+#  Pyrogram - Telegram MTProto API Client Library for Python
+#  Copyright (C) 2017-present Dan <https://github.com/delivrance>
+#
+#  This file is part of Pyrogram.
+#
+#  Pyrogram is free software: you can redistribute it and/or modify
+#  it under the terms of the GNU Lesser General Public License as published
+#  by the Free Software Foundation, either version 3 of the License, or
+#  (at your option) any later version.
+#
+#  Pyrogram is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU Lesser General Public License for more details.
+#
+#  You should have received a copy of the GNU Lesser General Public License
+#  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
+
+
+from .get_stories import GetStories
+
+class Stories(
+    GetStories,
+):
+    pass

pyrogram/methods/stories/get_stories.py

@@ -0,0 +1,89 @@
+#  Pyrogram - Telegram MTProto API Client Library for Python
+#  Copyright (C) 2017-present Dan <https://github.com/delivrance>
+#
+#  This file is part of Pyrogram.
+#
+#  Pyrogram is free software: you can redistribute it and/or modify
+#  it under the terms of the GNU Lesser General Public License as published
+#  by the Free Software Foundation, either version 3 of the License, or
+#  (at your option) any later version.
+#
+#  Pyrogram is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU Lesser General Public License for more details.
+#
+#  You should have received a copy of the GNU Lesser General Public License
+#  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
+
+from typing import Union, Iterable
+
+import pyrogram
+from pyrogram import raw, types
+
+
+class GetStories:
+    async def get_stories(
+        self: "pyrogram.Client",
+        story_sender_chat_id: Union[int, str],
+        story_ids: Union[int, Iterable[int]],
+    ) -> "types.Story":
+        """Get one or more stories from a chat by using stories identifiers.
+
+        .. include:: /_includes/usable-by/users.rst
+
+        Parameters:
+            story_sender_chat_id (``int`` | ``str``):
+                Identifier of the chat that posted the story.
+
+            story_ids (``int`` | Iterable of ``int``, *optional*):
+                Pass a single story identifier or an iterable of story ids (as integers) to get the content of the
+                story themselves.
+
+        Returns:
+            :obj:`~pyrogram.types.Story` | List of :obj:`~pyrogram.types.Story`: In case *story_ids* was not
+            a list, a single story is returned, otherwise a list of stories is returned.
+
+        Example:
+            .. code-block:: python
+
+                # Get stories by id
+                stories = await app.get_stories(
+                    story_sender_chat_id,
+                    [1, 2, 3]
+                )
+
+                for story in stories:
+                    print(story)
+        """
+        is_iterable = not isinstance(story_ids, int)
+        ids = list(story_ids) if is_iterable else [story_ids]
+
+        peer = await self.resolve_peer(story_sender_chat_id)
+        r = await self.invoke(
+            raw.functions.stories.GetStoriesByID(
+                peer=peer,
+                id=ids
+            )
+        )
+
+        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,
+                    users,
+                    chats,
+                    None, None, None,
+                    # TODO
+                    story,
+                    None, #
+                    # TODO
+                )
+            )
+
+        return types.List(stories) if is_iterable else stories[0] if stories else None

pyrogram/types/input_message_content/external_reply_info.py

@@ -291,7 +291,7 @@ async def _parse(
                     dice = types.Dice._parse(client, media)
                     media_type = enums.MessageMediaType.DICE
                 elif isinstance(media, raw.types.MessageMediaStory):
-                    story = await types.Story._parse(client, users, chats, media, None, None)
+                    story = await types.Story._parse(client, users, chats, media, None, None, None, None)
                     media_type = enums.MessageMediaType.STORY
                 elif isinstance(media, raw.types.MessageMediaGiveaway):
                     giveaway = types.Giveaway._parse(client, chats, media)

pyrogram/types/messages_and_media/message.py

@@ -1121,7 +1121,7 @@ async def _parse(
                     dice = types.Dice._parse(client, media)
                     media_type = enums.MessageMediaType.DICE
                 elif isinstance(media, raw.types.MessageMediaStory):
-                    story = await types.Story._parse(client, users, chats, media, None, None)
+                    story = await types.Story._parse(client, users, chats, media, None, None, None, None)
                     media_type = enums.MessageMediaType.STORY
                 elif isinstance(media, raw.types.MessageMediaGiveaway):
                     giveaway = types.Giveaway._parse(client, chats, media)
@@ -1267,7 +1267,7 @@ async def _parse(
                     )
 
             if isinstance(message.reply_to, raw.types.MessageReplyStoryHeader):
-                parsed_message.reply_to_story = await types.Story._parse(client, users, chats, None, message.reply_to, None)
+                parsed_message.reply_to_story = await types.Story._parse(client, users, chats, None, message.reply_to, None, None, None)
 
             if replies:
                 try:

pyrogram/types/messages_and_media/story.py

@@ -206,7 +206,9 @@ async def _parse(
         chats: dict,
         story_media: "raw.types.MessageMediaStory",
         reply_story: "raw.types.MessageReplyStoryHeader",
-        story_update: "raw.types.UpdateStory"
+        story_update: "raw.types.UpdateStory",
+        story_item: "raw.types.StoryItem",
+        peer: "raw.base.peer"
     ) -> "Story":
         story_id = None
         chat = None
@@ -310,6 +312,33 @@ async def _parse(
                 deleted
             ) = Story._parse_story_item(client, story_update.story)
 
+        if peer:
+            raw_peer_id = utils.get_raw_peer_id(peer)
+            if isinstance(peer, raw.types.PeerUser):
+                chat = types.Chat._parse_chat(client, users.get(raw_peer_id))
+            else:
+                chat = types.Chat._parse_chat(client, chats.get(raw_peer_id))
+            
+        if story_item:
+            story_id = getattr(story_item, "id", None)
+            (
+                date,
+                expire_date,
+                media,
+                has_protected_content,
+                photo,
+                video,
+                edited,
+                pinned,
+                caption,
+                caption_entities,
+                views,
+                forwards,
+                reactions,
+                skipped,
+                deleted
+            ) = Story._parse_story_item(client, story_item)
+
         return Story(
             client=client,
             _raw=rawupdate,