diff --git a/compiler/docs/compiler.py b/compiler/docs/compiler.py index bd5ed5d9..de96a04b 100644 --- a/compiler/docs/compiler.py +++ b/compiler/docs/compiler.py @@ -273,6 +273,7 @@ def pyrogram_api(): delete_stories edit_story export_story_link + get_all_stories get_stories get_stories_history get_user_stories diff --git a/pyrogram/methods/users/__init__.py b/pyrogram/methods/users/__init__.py index 3569968a..0f1bb167 100644 --- a/pyrogram/methods/users/__init__.py +++ b/pyrogram/methods/users/__init__.py @@ -26,6 +26,7 @@ from .get_chat_photos_count import GetChatPhotosCount from .get_common_chats import GetCommonChats from .get_default_emoji_statuses import GetDefaultEmojiStatuses from .get_me import GetMe +from .get_all_stories import GetAllStories from .get_stories import GetStories from .get_stories_history import GetUserStoriesHistory from .get_user_stories import GetUserStories @@ -49,6 +50,7 @@ class Users( DeleteProfilePhotos, GetUsers, GetMe, + GetAllStories, GetStories, GetUserStoriesHistory, GetUserStories, diff --git a/pyrogram/methods/users/get_all_stories.py b/pyrogram/methods/users/get_all_stories.py new file mode 100644 index 00000000..55e9b777 --- /dev/null +++ b/pyrogram/methods/users/get_all_stories.py @@ -0,0 +1,57 @@ +# Pyrofork - Telegram MTProto API Client Library for Python +# Copyright (C) 2022-present Mayuri-Chan +# +# This file is part of Pyrofork. +# +# Pyrofork 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. +# +# Pyrofork 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 Pyrofork. If not, see . + +import logging +from typing import AsyncGenerator, Optional + +import pyrogram +from pyrogram import raw +from pyrogram import types + +log = logging.getLogger(__name__) + +class GetAllStories: + async def get_all_stories( + self: "pyrogram.Client" + ) -> Optional[AsyncGenerator["types.Story", None]]: + """Get all active stories. + + .. include:: /_includes/usable-by/users.rst + + Returns: + ``Generator``: On success, a generator yielding :obj:`~pyrogram.types.Story` objects is returned. + + Example: + .. code-block:: python + + # Get all active story + async for story in app.get_all_stories(): + print(story) + + Raises: + ValueError: In case of invalid arguments. + """ + + rpc = raw.functions.stories.GetAllStories() + + r = await self.invoke(rpc, sleep_threshold=-1) + + for user_story in r.user_stories: + user_id = user_story.user_id + for story in user_story.stories: + yield await types.Story._parse(self, story, user_id)