diff --git a/compiler/docs/compiler.py b/compiler/docs/compiler.py index e108f0577..46363fbf8 100644 --- a/compiler/docs/compiler.py +++ b/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 diff --git a/pyrogram/methods/chats/__init__.py b/pyrogram/methods/chats/__init__.py index 1b96de013..267a05a9d 100644 --- a/pyrogram/methods/chats/__init__.py +++ b/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, diff --git a/pyrogram/methods/chats/get_forum_topics_count.py b/pyrogram/methods/chats/get_forum_topics_count.py new file mode 100644 index 000000000..22aec1694 --- /dev/null +++ b/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 +# 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 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/* (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