diff --git a/docs/source/pyrogram/Client.rst b/docs/source/pyrogram/Client.rst index 59293273..e704cc1e 100644 --- a/docs/source/pyrogram/Client.rst +++ b/docs/source/pyrogram/Client.rst @@ -101,6 +101,7 @@ Chats iter_chat_members get_dialogs iter_dialogs + get_dialogs_count restrict_chat update_chat_username diff --git a/pyrogram/client/methods/chats/__init__.py b/pyrogram/client/methods/chats/__init__.py index c708453f..8db44abe 100644 --- a/pyrogram/client/methods/chats/__init__.py +++ b/pyrogram/client/methods/chats/__init__.py @@ -39,6 +39,7 @@ from .set_chat_title import SetChatTitle from .unban_chat_member import UnbanChatMember from .unpin_chat_message import UnpinChatMessage from .update_chat_username import UpdateChatUsername +from .get_dialogs_count import GetDialogsCount class Chats( @@ -64,6 +65,7 @@ class Chats( IterDialogs, IterChatMembers, UpdateChatUsername, - RestrictChat + RestrictChat, + GetDialogsCount ): pass diff --git a/pyrogram/client/methods/chats/get_dialogs_count.py b/pyrogram/client/methods/chats/get_dialogs_count.py new file mode 100644 index 00000000..1ba6895f --- /dev/null +++ b/pyrogram/client/methods/chats/get_dialogs_count.py @@ -0,0 +1,51 @@ +# Pyrogram - Telegram MTProto API Client Library for Python +# Copyright (C) 2017-2019 Dan Tès +# +# 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 . + +from pyrogram.api import functions, types +from ...ext import BaseClient + + +class GetDialogsCount(BaseClient): + def get_dialogs_count(self, pinned_only: bool = False) -> int: + """Use this method to get the total count of your dialogs. + + pinned_only (``bool``, *optional*): + Pass True if you want to count only pinned dialogs. + Defaults to False. + + Returns: + On success, an integer is returned. + + Raises: + :class:`RPCError ` in case of a Telegram RPC error. + """ + + if pinned_only: + return len(self.send(functions.messages.GetPinnedDialogs()).dialogs) + else: + r = self.send( + functions.messages.GetDialogs( + offset_date=0, + offset_id=0, + offset_peer=types.InputPeerEmpty(), + limit=1, + hash=0 + ) + ) + + return r.count