Fix get_dialogs_count breaking in case of less than 200 dialogs

delivrance · delivrance · commit 692073c85683 · 2019-05-06T17:09:44.000+02:00
diff --git a/pyrogram/client/methods/chats/get_dialogs_count.py b/pyrogram/client/methods/chats/get_dialogs_count.py
@@ -48,4 +48,7 @@ def get_dialogs_count(self, pinned_only: bool = False) -> int:
                 )
             )
 
-            return r.count
+            if isinstance(r, types.messages.Dialogs):
+                return len(r.dialogs)
+            else:
+                return r.count

Original file line number	Diff line number	Diff line change
`@@ -48,4 +48,7 @@ def get_dialogs_count(self, pinned_only: bool = False) -> int:`
`48`	`48`	`)`
`49`	`49`	`)`
`50`	`50`
`51`		`- return r.count`
	`51`	`+ if isinstance(r, types.messages.Dialogs):`
	`52`	`+ return len(r.dialogs)`
	`53`	`+ else:`
	`54`	`+ return r.count`