Remove Dialogs type

delivrance · delivrance · commit c8fd446cb618 · 2019-06-08T14:00:00.000+02:00
diff --git a/docs/source/api/types.rst b/docs/source/api/types.rst
@@ -32,7 +32,6 @@ Users & Chats
     - :class:`ChatMember`
     - :class:`ChatPermissions`
     - :class:`Dialog`
-    - :class:`Dialogs`
 
 Messages & Media
 ^^^^^^^^^^^^^^^^
@@ -124,7 +123,6 @@ Details
 .. autoclass:: ChatMember()
 .. autoclass:: ChatPermissions()
 .. autoclass:: Dialog()
-.. autoclass:: Dialogs()
 
 ..  Messages & Media
 .. autoclass:: Message()
diff --git a/pyrogram/client/methods/chats/get_dialogs.py b/pyrogram/client/methods/chats/get_dialogs.py
@@ -18,6 +18,7 @@
 
 import logging
 import time
+from typing import List
 
 import pyrogram
 from pyrogram.api import functions, types
@@ -33,7 +34,7 @@ def get_dialogs(
         offset_date: int = 0,
         limit: int = 100,
         pinned_only: bool = False
-    ) -> "pyrogram.Dialogs":
+    ) -> List["pyrogram.Dialog"]:
         """Get a chunk of the user's dialogs.
 
         You can get up to 100 dialogs at once.
@@ -53,7 +54,7 @@ def get_dialogs(
                 Defaults to False.
 
         Returns:
-            :obj:`Dialogs`: On success, an object containing a list of dialogs is returned.
+            List of :obj:`Dialog`: On success, a list of dialogs is returned.
 
         Raises:
             RPCError: In case of a Telegram RPC error.
@@ -80,4 +81,32 @@ def get_dialogs(
             else:
                 break
 
-        return pyrogram.Dialogs._parse(self, r)
+        users = {i.id: i for i in r.users}
+        chats = {i.id: i for i in r.chats}
+
+        messages = {}
+
+        for message in r.messages:
+            to_id = message.to_id
+
+            if isinstance(to_id, types.PeerUser):
+                if message.out:
+                    chat_id = to_id.user_id
+                else:
+                    chat_id = message.from_id
+            elif isinstance(to_id, types.PeerChat):
+                chat_id = -to_id.chat_id
+            else:
+                chat_id = int("-100" + str(to_id.channel_id))
+
+            messages[chat_id] = pyrogram.Message._parse(self, message, users, chats)
+
+        parsed_dialogs = []
+
+        for dialog in r.dialogs:
+            if not isinstance(dialog, types.Dialog):
+                continue
+
+            parsed_dialogs.append(pyrogram.Dialog._parse(self, dialog, messages, users, chats))
+
+        return pyrogram.List(parsed_dialogs)
diff --git a/pyrogram/client/methods/chats/iter_dialogs.py b/pyrogram/client/methods/chats/iter_dialogs.py
@@ -55,7 +55,7 @@ def iter_dialogs(
 
         pinned_dialogs = self.get_dialogs(
             pinned_only=True
-        ).dialogs
+        )
 
         for dialog in pinned_dialogs:
             yield dialog
@@ -69,7 +69,7 @@ def iter_dialogs(
             dialogs = self.get_dialogs(
                 offset_date=offset_date,
                 limit=limit
-            ).dialogs
+            )
 
             if not dialogs:
                 return
diff --git a/pyrogram/client/types/user_and_chats/__init__.py b/pyrogram/client/types/user_and_chats/__init__.py
@@ -22,10 +22,9 @@
 from .chat_photo import ChatPhoto
 from .chat_preview import ChatPreview
 from .dialog import Dialog
-from .dialogs import Dialogs
 from .user import User
 from .user_status import UserStatus
 
 __all__ = [
-    "Chat", "ChatMember", "ChatPermissions", "ChatPhoto", "ChatPreview", "Dialog", "Dialogs", "User", "UserStatus"
+    "Chat", "ChatMember", "ChatPermissions", "ChatPhoto", "ChatPreview", "Dialog", "User", "UserStatus"
 ]
diff --git a/pyrogram/client/types/user_and_chats/dialogs.py b/pyrogram/client/types/user_and_chats/dialogs.py
