Use create_task instead of ensure_future

delivrance · delivrance · commit 5f087e5f824c · 2020-08-22T14:05:54.000+02:00
diff --git a/pyrogram/dispatcher.py b/pyrogram/dispatcher.py
@@ -115,7 +115,7 @@ async def start(self):
                 self.locks_list.append(asyncio.Lock())
 
                 self.handler_worker_tasks.append(
-                    asyncio.ensure_future(self.handler_worker(self.locks_list[-1]))
+                    self.loop.create_task(self.handler_worker(self.locks_list[-1]))
                 )
 
             logging.info(f"Started {self.client.workers} HandlerTasks")
@@ -148,7 +148,7 @@ async def fn():
                 for lock in self.locks_list:
                     lock.release()
 
-        asyncio.ensure_future(fn())
+        self.loop.create_task(fn())
 
     def remove_handler(self, handler, group: int):
         async def fn():
@@ -164,7 +164,7 @@ async def fn():
                 for lock in self.locks_list:
                     lock.release()
 
-        asyncio.ensure_future(fn())
+        self.loop.create_task(fn())
 
     async def handler_worker(self, lock):
         while True:
diff --git a/pyrogram/methods/advanced/save_file.py b/pyrogram/methods/advanced/save_file.py
@@ -101,7 +101,7 @@ async def worker(session):
                     return
 
                 try:
-                    await asyncio.ensure_future(session.send(data))
+                    await self.loop.create_task(session.send(data))
                 except Exception as e:
                     log.error(e)
 
@@ -139,7 +139,7 @@ async def worker(session):
                 await self.storage.test_mode(), is_media=True
             ) for _ in range(pool_size)
         ]
-        workers = [asyncio.ensure_future(worker(session)) for session in pool for _ in range(workers_count)]
+        workers = [self.loop.create_task(worker(session)) for session in pool for _ in range(workers_count)]
         queue = asyncio.Queue(16)
 
         try:
diff --git a/pyrogram/methods/utilities/restart.py b/pyrogram/methods/utilities/restart.py
@@ -65,6 +65,6 @@ async def do_it():
         if block:
             await do_it()
         else:
-            asyncio.ensure_future(do_it())
+            self.loop.create_task(do_it())
 
         return self
diff --git a/pyrogram/methods/utilities/stop.py b/pyrogram/methods/utilities/stop.py
@@ -60,6 +60,6 @@ async def do_it():
         if block:
             await do_it()
         else:
-            asyncio.ensure_future(do_it())
+            self.loop.create_task(do_it())
 
         return self
diff --git a/pyrogram/syncer.py b/pyrogram/syncer.py
@@ -59,7 +59,7 @@ async def remove(cls, client):
     @classmethod
     def start(cls):
         cls.event.clear()
-        asyncio.ensure_future(cls.worker())
+        asyncio.get_event_loop().create_task(cls.worker())
 
     @classmethod
     def stop(cls):
