pyrofork: Add get_chat_forum_topics_count method

Signed-off-by: wulan17 <[email protected]>

compiler/docs/compiler.py

@@ -253,6 +253,7 @@ def get_title_list(s: str) -> list:
             get_folders
             get_forum_topics
             get_forum_topics_by_id
+            get_forum_topics_count
             set_chat_username
             archive_chats
             unarchive_chats

pyrogram/methods/chats/__init__.py

@@ -50,6 +50,7 @@
 from .get_folders import GetFolders
 from .get_forum_topics import GetForumTopics
 from .get_forum_topics_by_id import GetForumTopicsByID
+from .get_forum_topics_count import GetForumTopicsCount
 from .get_send_as_chats import GetSendAsChats
 from .join_chat import JoinChat
 from .leave_chat import LeaveChat
@@ -98,6 +99,7 @@ class Chats(
     GetFolders,
     GetForumTopics,
     GetForumTopicsByID,
+    GetForumTopicsCount,
     ArchiveChats,
     UnarchiveChats,
     CreateGroup,

pyrogram/methods/chats/get_forum_topics_count.py

@@ -0,0 +1,63 @@
+#  Pyrofork - Telegram MTProto API Client Library for Python
+#  Copyright (C) 2017-present Dan <https://github.com/delivrance>
+#  Copyright (C) 2022-present Mayuri-Chan <https://github.com/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 <http://www.gnu.org/licenses/>.
+
+import logging
+from typing import Union, Optional, AsyncGenerator
+
+import pyrogram
+from pyrogram import raw
+from pyrogram import types
+
+log = logging.getLogger(__name__)
+
+
+class GetForumTopicsCount:
+    async def get_forum_topics_count(
+        self: "pyrogram.Client",
+        chat_id: Union[int, str]
+    ) -> Optional[AsyncGenerator["types.ForumTopic", None]]:
+        """Get forum topics count from a chat.
+
+        .. include:: /_includes/usable-by/users.rst
+
+        Parameters:
+            chat_id (``int`` | ``str``):
+                Unique identifier (int) or username (str) of the target chat.
+                You can also use chat public link in form of *t.me/<username>* (str).
+
+        Returns:
+            ``int``: On success, the count of forum topics is returned.
+
+        Example:
+            .. code-block:: python
+
+                # get all forum topics count
+                app.get_forum_topics_count(chat_id)
+
+        Raises:
+            ValueError: In case of invalid arguments.
+        """
+
+        peer = await self.resolve_peer(chat_id)
+
+        rpc = raw.functions.channels.GetForumTopics(channel=peer, offset_date=0, offset_id=0, offset_topic=0, limit=limit)
+
+        r = await self.invoke(rpc, sleep_threshold=-1)
+
+        return r.count