Fix dialogs after L100 update

delivrance · delivrance · commit c2424029b57e · 2019-05-29T09:37:50.000+02:00
diff --git a/pyrogram/client/client.py b/pyrogram/client/client.py
@@ -313,7 +313,7 @@ def start(self):
                     self.get_initial_dialogs()
                     self.get_contacts()
                 else:
-                    self.send(functions.messages.GetPinnedDialogs())
+                    self.send(functions.messages.GetPinnedDialogs(folder_id=0))
                     self.get_initial_dialogs_chunk()
             else:
                 self.send(functions.updates.GetState())
@@ -1325,7 +1325,7 @@ def get_initial_dialogs_chunk(self, offset_date: int = 0):
                 return r
 
     def get_initial_dialogs(self):
-        self.send(functions.messages.GetPinnedDialogs())
+        self.send(functions.messages.GetPinnedDialogs(folder_id=0))
 
         dialogs = self.get_initial_dialogs_chunk()
         offset_date = utils.get_offset_date(dialogs)
diff --git a/pyrogram/client/methods/chats/get_dialogs.py b/pyrogram/client/methods/chats/get_dialogs.py
@@ -62,7 +62,7 @@ def get_dialogs(
         while True:
             try:
                 if pinned_only:
-                    r = self.send(functions.messages.GetPinnedDialogs())
+                    r = self.send(functions.messages.GetPinnedDialogs(folder_id=0))
                 else:
                     r = self.send(
                         functions.messages.GetDialogs(
diff --git a/pyrogram/client/methods/chats/get_dialogs_count.py b/pyrogram/client/methods/chats/get_dialogs_count.py
@@ -36,7 +36,7 @@ def get_dialogs_count(self, pinned_only: bool = False) -> int:
         """
 
         if pinned_only:
-            return len(self.send(functions.messages.GetPinnedDialogs()).dialogs)
+            return len(self.send(functions.messages.GetPinnedDialogs(folder_id=0)).dialogs)
         else:
             r = self.send(
                 functions.messages.GetDialogs(
diff --git a/pyrogram/client/methods/chats/iter_dialogs.py b/pyrogram/client/methods/chats/iter_dialogs.py
@@ -26,7 +26,7 @@ class IterDialogs(BaseClient):
     def iter_dialogs(
         self,
         offset_date: int = 0,
-        limit: int = 0
+        limit: int = None
     ) -> Generator["pyrogram.Dialog", None, None]:
         """Iterate through a user's dialogs sequentially.
 
diff --git a/pyrogram/client/types/user_and_chats/dialog.py b/pyrogram/client/types/user_and_chats/dialog.py
@@ -69,7 +69,7 @@ def __init__(
         self.is_pinned = is_pinned
 
     @staticmethod
-    def _parse(client, dialog, messages, users, chats) -> "Dialog":
+    def _parse(client, dialog: types.Dialog, messages, users, chats) -> "Dialog":
         chat_id = dialog.peer
 
         if isinstance(chat_id, types.PeerUser):
diff --git a/pyrogram/client/types/user_and_chats/dialogs.py b/pyrogram/client/types/user_and_chats/dialogs.py
@@ -51,7 +51,7 @@ def __init__(
         self.dialogs = dialogs
 
     @staticmethod
-    def _parse(client, dialogs) -> "Dialogs":
+    def _parse(client, dialogs: types.messages.Dialogs) -> "Dialogs":
         users = {i.id: i for i in dialogs.users}
         chats = {i.id: i for i in dialogs.chats}
 
@@ -72,8 +72,16 @@ def _parse(client, dialogs) -> "Dialogs":
 
             messages[chat_id] = Message._parse(client, message, users, chats)
 
+        parsed_dialogs = []
+
+        for dialog in dialogs.dialogs:
+            if not isinstance(dialog, types.Dialog):
+                continue
+
+            parsed_dialogs.append(Dialog._parse(client, dialog, messages, users, chats))
+
         return Dialogs(
             total_count=getattr(dialogs, "count", len(dialogs.dialogs)),
-            dialogs=[Dialog._parse(client, dialog, messages, users, chats) for dialog in dialogs.dialogs],
+            dialogs=parsed_dialogs,
             client=client
         )
