Fix sleep method calls in asyncio: time.sleep -> asyncio.sleep

delivrance · delivrance · commit 4eb26c5b9276 · 2019-02-04T18:34:58.000+01:00
diff --git a/pyrogram/client/client.py b/pyrogram/client/client.py
@@ -48,7 +48,6 @@
     VolumeLocNotFound, UserMigrate, FileIdInvalid, ChannelPrivate, PhoneNumberOccupied,
     PasswordRecoveryNa, PasswordEmpty
 )
-from pyrogram.client.handlers import DisconnectHandler
 from pyrogram.client.handlers.handler import Handler
 from pyrogram.client.methods.password.utils import compute_check
 from pyrogram.crypto import AES
@@ -571,7 +570,7 @@ async def default_phone_number_callback():
                     raise
                 else:
                     print(e.MESSAGE.format(x=e.x))
-                    time.sleep(e.x)
+                    await asyncio.sleep(e.x)
             except Exception as e:
                 log.error(e, exc_info=True)
                 raise
@@ -707,7 +706,7 @@ async def default_recovery_callback(email_pattern: str) -> str:
                             raise
                         else:
                             print(e.MESSAGE.format(x=e.x))
-                            time.sleep(e.x)
+                            await asyncio.sleep(e.x)
                             self.password = None
                             self.recovery_code = None
                     except Exception as e:
@@ -721,7 +720,7 @@ async def default_recovery_callback(email_pattern: str) -> str:
                     raise
                 else:
                     print(e.MESSAGE.format(x=e.x))
-                    time.sleep(e.x)
+                    await asyncio.sleep(e.x)
             except Exception as e:
                 log.error(e, exc_info=True)
                 raise
diff --git a/pyrogram/client/methods/chats/get_dialogs.py b/pyrogram/client/methods/chats/get_dialogs.py
@@ -16,8 +16,8 @@
 # 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 asyncio
 import logging
-import time
 
 import pyrogram
 from pyrogram.api import functions, types
@@ -73,8 +73,8 @@ async def get_dialogs(self,
                         )
                     )
             except FloodWait as e:
-                log.warning("Sleeping {}s".format(e.x))
-                time.sleep(e.x)
+                log.warning("Sleeping for {}s".format(e.x))
+                await asyncio.sleep(e.x)
             else:
                 break
 
diff --git a/pyrogram/client/methods/messages/get_history.py b/pyrogram/client/methods/messages/get_history.py
@@ -16,8 +16,8 @@
 # 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 asyncio
 import logging
-import time
 from typing import Union
 
 import pyrogram
diff --git a/pyrogram/client/methods/messages/get_messages.py b/pyrogram/client/methods/messages/get_messages.py
@@ -16,8 +16,8 @@
 # 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 asyncio
 import logging
-import time
 from typing import Union, Iterable
 
 import pyrogram
@@ -88,7 +88,7 @@ async def get_messages(self,
                 r = await self.send(rpc)
             except FloodWait as e:
                 log.warning("Sleeping for {}s".format(e.x))
-                time.sleep(e.x)
+                await asyncio.sleep(e.x)
             else:
                 break
 
diff --git a/pyrogram/session/auth.py b/pyrogram/session/auth.py
@@ -16,6 +16,7 @@
 # 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 asyncio
 import logging
 import time
 from hashlib import sha1
@@ -254,7 +255,7 @@ async def create(self):
                 else:
                     raise e
 
-                time.sleep(1)
+                await asyncio.sleep(1)
                 continue
             else:
                 return auth_key
