Fix flood waits generated when iterating over basic chat members

delivrance · delivrance · commit 90b959555da6 · 2019-03-12T16:46:54.000+01:00
diff --git a/pyrogram/client/methods/chats/iter_chat_members.py b/pyrogram/client/methods/chats/iter_chat_members.py
@@ -20,6 +20,7 @@
 from typing import Union, Generator
 
 import pyrogram
+from pyrogram.api import types
 from ...ext import BaseClient
 
 
@@ -82,6 +83,7 @@ def iter_chat_members(self,
         queries = [query] if query else QUERIES
         total = limit or (1 << 31) - 1
         limit = min(200, total)
+        resolved_chat_id = self.resolve_peer(chat_id)
 
         filter = (
             Filters.RECENT
@@ -107,6 +109,9 @@ def iter_chat_members(self,
                 if not chat_members:
                     break
 
+                if isinstance(resolved_chat_id, types.InputPeerChat):
+                    total = len(chat_members)
+
                 offset += len(chat_members)
 
                 for chat_member in chat_members:
