Add get_dialogs_count method

delivrance · delivrance · commit 4e77ead18171 · 2019-05-06T16:58:59.000+02:00
diff --git 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
@@ -39,6 +39,7 @@
 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
@@ -0,0 +1,51 @@
+# Pyrogram - Telegram MTProto API Client Library for Python
+# Copyright (C) 2017-2019 Dan Tès <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 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 <pyrogram.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
