Add chat_list to the methods get_dialogs and get_dialogs_count

Author: SpEcHiDe

docs/source/releases/changes-in-this-fork.rst

@@ -14,6 +14,7 @@ If you found any issue or have any suggestions, feel free to make `an issue <htt
 | Scheme layer used: 185 |
 +------------------------+
 
+- Added the parameter ``chat_list`` to the methods :meth:`~pyrogram.Client.get_dialogs` and :meth:`~pyrogram.Client.get_dialogs_count`.
 - Added ``gifted_stars`` service message to the class :obj:`~pyrogram.types.Message`.
 - Added the fields ``have_access``, ``has_main_web_app``, ``active_user_count`` to the class :obj:`~pyrogram.types.User`, which is returned in response to  :meth:`~pyrogram.Client.get_me`.
 - Added the parameter ``business_connection_id`` to the methods :meth:`~pyrogram.Client.pin_chat_message` and :meth:`~pyrogram.Client.unpin_chat_message`, allowing bots to manage pinned messages on behalf of a business account.

pyrogram/methods/chats/get_dialogs.py

@@ -25,7 +25,8 @@
 class GetDialogs:
     async def get_dialogs(
         self: "pyrogram.Client",
-        limit: int = 0
+        limit: int = 0,
+        chat_list: int = 0
     ) -> Optional[AsyncGenerator["types.Dialog", None]]:
         """Get a user's dialogs sequentially.
 
@@ -35,6 +36,9 @@ async def get_dialogs(
             limit (``int``, *optional*):
                 Limits the number of dialogs to be retrieved.
                 By default, no limit is applied and all dialogs are returned.
+            
+            chat_list (``int``, *optional*):
+                Chat list in which to search messages; Only Main (0) and Archive (1) chat lists are supported. Defaults to (0) Main chat list.
 
         Returns:
             ``Generator``: A generator yielding :obj:`~pyrogram.types.Dialog` objects.
@@ -61,7 +65,8 @@ async def get_dialogs(
                     offset_id=offset_id,
                     offset_peer=offset_peer,
                     limit=limit,
-                    hash=0
+                    hash=0,
+                    folder_id=chat_list
                 ),
                 sleep_threshold=60
             )

pyrogram/methods/chats/get_dialogs_count.py

@@ -23,7 +23,8 @@
 class GetDialogsCount:
     async def get_dialogs_count(
         self: "pyrogram.Client",
-        pinned_only: bool = False
+        pinned_only: bool = False,
+        chat_list: int = 0
     ) -> int:
         """Get the total count of your dialogs.
 
@@ -33,6 +34,9 @@ async def get_dialogs_count(
             pinned_only (``bool``, *optional*):
                 Pass True if you want to count only pinned dialogs.
                 Defaults to False.
+            
+            chat_list (``int``, *optional*):
+                Chat list in which to search messages; Only Main (0) and Archive (1) chat lists are supported. Defaults to (0) Main chat list.
 
         Returns:
             ``int``: On success, the dialogs count is returned.
@@ -45,15 +49,16 @@ async def get_dialogs_count(
         """
 
         if pinned_only:
-            return len((await self.invoke(raw.functions.messages.GetPinnedDialogs(folder_id=0))).dialogs)
+            return len((await self.invoke(raw.functions.messages.GetPinnedDialogs(folder_id=chat_list))).dialogs)
         else:
             r = await self.invoke(
                 raw.functions.messages.GetDialogs(
                     offset_date=0,
                     offset_id=0,
                     offset_peer=raw.types.InputPeerEmpty(),
                     limit=1,
-                    hash=0
+                    hash=0,
+                    folder_id=chat_list
                 )
             )