Add search_posts method

compiler/docs/compiler.py

@@ -191,6 +191,8 @@ def pyrogram_api():
             send_dice
             search_messages
             search_messages_count
+            search_posts
+            search_posts_count
             search_global
             search_global_count
             download_media

pyrogram/methods/messages/__init__.py

@@ -48,6 +48,8 @@ from .search_global import SearchGlobal
 from .search_global_count import SearchGlobalCount
 from .search_messages import SearchMessages
 from .search_messages_count import SearchMessagesCount
+from .search_posts import SearchPosts
+from .search_posts_count import SearchPostsCount
 from .send_animation import SendAnimation
 from .send_audio import SendAudio
 from .send_cached_media import SendCachedMedia
@@ -124,6 +126,8 @@ class Messages(
     CopyMessage,
     CopyMediaGroup,
     SearchMessagesCount,
+    SearchPosts,
+    SearchPostsCount,
     SearchGlobalCount,
     GetDiscussionMessage,
     SendReaction,

pyrogram/methods/messages/search_posts.py

@@ -0,0 +1,96 @@
+#  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 AsyncGenerator
+
+import pyrogram
+from pyrogram import raw
+from pyrogram import types
+from pyrogram import utils
+
+
+class SearchPosts:
+    async def search_posts(
+        self: "pyrogram.Client",
+        hashtag: str,
+        limit: int = 0,
+    ) -> AsyncGenerator["types.Message", None]:
+        """Search posts globally by hashtag.
+
+        If you want to get the posts count only, see :meth:`~pyrogram.Client.search_posts_count`.
+
+        .. include:: /_includes/usable-by/users.rst
+
+        Parameters:
+            hashtag (``str``):
+                Text query string.
+
+            limit (``int``, *optional*):
+                Limits the number of posts to be retrieved.
+                By default, no limit is applied and all posts are returned.
+
+        Returns:
+            ``Generator``: A generator yielding :obj:`~pyrogram.types.Message` objects.
+
+        Example:
+            .. code-block:: python
+
+                # Search for "#pyrogram". Get the first 50 results
+                async for message in app.search_posts("#pyrogram", limit=50):
+                    print(message.text)
+        """
+        current = 0
+        total = abs(limit) or (1 << 31)
+        limit = min(100, total)
+
+        offset_date = 0
+        offset_peer = raw.types.InputPeerEmpty()
+        offset_id = 0
+
+        while True:
+            messages = await utils.parse_messages(
+                self,
+                await self.invoke(
+                    raw.functions.channels.SearchPosts(
+                        hashtag=hashtag,
+                        offset_rate=offset_date,
+                        offset_peer=offset_peer,
+                        offset_id=offset_id,
+                        limit=limit
+                    ),
+                    sleep_threshold=60
+                ),
+                replies=0
+            )
+
+            if not messages:
+                return
+
+            last = messages[-1]
+
+            offset_date = utils.datetime_to_timestamp(last.date)
+            offset_peer = await self.resolve_peer(last.chat.id)
+            offset_id = last.id
+
+            for message in messages:
+                yield message
+
+                current += 1
+
+                if current >= total:
+                    return

pyrogram/methods/messages/search_posts_count.py

@@ -0,0 +1,54 @@
+#  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/>.
+
+import pyrogram
+from pyrogram import raw
+
+
+class SearchPostsCount:
+    async def search_posts_count(
+        self: "pyrogram.Client",
+        hashtag: str,
+    ) -> int:
+        """Get the count of posts with hashtag resulting from a search.
+
+        If you want to get the actual posts, see :meth:`~pyrogram.Client.search_posts`.
+
+        .. include:: /_includes/usable-by/users.rst
+
+        Parameters:
+            hashtag (``str``):
+                Text query string.
+
+        Returns:
+            ``int``: On success, the posts count is returned.
+        """
+        r = await self.invoke(
+            raw.functions.channels.SearchPosts(
+                hashtag=hashtag,
+                offset_rate=0,
+                offset_peer=raw.types.InputPeerEmpty(),
+                offset_id=0,
+                limit=1
+            )
+        )
+
+        if hasattr(r, "count"):
+            return r.count
+        else:
+            return len(r.messages)