From 94587b6d807d6fbec220f0f06acfc6554823517e Mon Sep 17 00:00:00 2001 From: wulan17 Date: Sat, 2 Sep 2023 18:05:29 +0700 Subject: [PATCH] Pyrofork: Add get_stories_history method Signed-off-by: wulan17 --- compiler/docs/compiler.py | 1 + pyrogram/methods/users/__init__.py | 2 + pyrogram/methods/users/get_stories_history.py | 65 +++++++++++++++++++ 3 files changed, 68 insertions(+) create mode 100644 pyrogram/methods/users/get_stories_history.py diff --git a/compiler/docs/compiler.py b/compiler/docs/compiler.py index 01c5d584..bd5ed5d9 100644 --- a/compiler/docs/compiler.py +++ b/compiler/docs/compiler.py @@ -274,6 +274,7 @@ def pyrogram_api(): edit_story export_story_link get_stories + get_stories_history get_user_stories send_story """, diff --git a/pyrogram/methods/users/__init__.py b/pyrogram/methods/users/__init__.py index 2f7551e4..3569968a 100644 --- a/pyrogram/methods/users/__init__.py +++ b/pyrogram/methods/users/__init__.py @@ -27,6 +27,7 @@ from .get_common_chats import GetCommonChats from .get_default_emoji_statuses import GetDefaultEmojiStatuses from .get_me import GetMe from .get_stories import GetStories +from .get_stories_history import GetUserStoriesHistory from .get_user_stories import GetUserStories from .get_users import GetUsers from .send_story import SendStory @@ -49,6 +50,7 @@ class Users( GetUsers, GetMe, GetStories, + GetUserStoriesHistory, GetUserStories, SetUsername, GetChatPhotosCount, diff --git a/pyrogram/methods/users/get_stories_history.py b/pyrogram/methods/users/get_stories_history.py new file mode 100644 index 00000000..44428829 --- /dev/null +++ b/pyrogram/methods/users/get_stories_history.py @@ -0,0 +1,65 @@ +# 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 GetUserStoriesHistory: + async def get_stories_history( + self: "pyrogram.Client", + limit: int = 0, + offset_id: int = 0 + ) -> Optional[AsyncGenerator["types.Story", None]]: + """Get stories history. + + .. include:: /_includes/usable-by/users.rst + + Parameters: + limit (``int``, *optional*): + Limits the number of stories to be retrieved. + By default, no limit is applied and all stories are returned. + + offset_id (``int``, *optional*): + Identifier of the first story to be returned. + + Returns: + ``Generator``: On success, a generator yielding :obj:`~pyrogram.types.Story` objects is returned. + + Example: + .. code-block:: python + + # Get story history + async for story in app.get_stories_history(): + print(story) + + Raises: + ValueError: In case of invalid arguments. + """ + + rpc = raw.functions.stories.GetStoriesArchive(offset_id=offset_id, limit=limit) + + r = await self.invoke(rpc, sleep_threshold=-1) + + for story in r.stories: + yield await types.Story._parse(self, story, self.me.id)