Automatically handle flood waits when using get_chat_members

pyrogram/client/methods/chats/get_chat_members.py

@@ -16,12 +16,17 @@
 # You should have received a copy of the GNU Lesser General Public License
 # along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
+import logging
+import time
 from typing import Union
 
 import pyrogram
 from pyrogram.api import functions, types
+from pyrogram.errors import FloodWait
 from ...ext import BaseClient
 
+log = logging.getLogger(__name__)
+
 
 class Filters:
     ALL = "all"
@@ -116,17 +121,22 @@ class GetChatMembers(BaseClient):
             else:
                 raise ValueError("Invalid filter \"{}\"".format(filter))
 
-            return pyrogram.ChatMembers._parse(
+            while True:
-                self,
+                try:
-                self.send(
+                    return pyrogram.ChatMembers._parse(
-                    functions.channels.GetParticipants(
+                        self,
-                        channel=peer,
+                        self.send(
-                        filter=filter,
+                            functions.channels.GetParticipants(
-                        offset=offset,
+                                channel=peer,
-                        limit=limit,
+                                filter=filter,
-                        hash=0
+                                offset=offset,
+                                limit=limit,
+                                hash=0
+                            )
+                        )
                     )
-                )
+                except FloodWait as e:
-            )
+                    log.warning("Sleeping for {}s".format(e.x))
+                    time.sleep(e.x)
         else:
             raise ValueError("The chat_id \"{}\" belongs to a user".format(chat_id))