fix: added missing coroutines (#26)

null-nick · HitaloM · commit b51720443571 · 2024-05-26T18:34:47.000-03:00
diff --git a/hydrogram/methods/messages/download_media.py b/hydrogram/methods/messages/download_media.py
@@ -196,5 +196,5 @@ async def progress(current, total):
 
         if block:
             return await downloader
-        asyncio.get_event_loop().create_task(downloader)
+        await asyncio.get_event_loop().create_task(downloader)
         return None
diff --git a/hydrogram/session/session.py b/hydrogram/session/session.py
@@ -263,7 +263,7 @@ async def handle_packet(self, packet):
             elif isinstance(msg.body, raw.types.Pong):
                 msg_id = msg.body.msg_id
             elif self.client is not None:
-                self.loop.create_task(self.client.handle_updates(msg.body))
+                await self.loop.create_task(self.client.handle_updates(msg.body))
 
             if msg_id in self.results:
                 self.results[msg_id].value = getattr(msg.body, "result", msg.body)
@@ -317,11 +317,11 @@ async def recv_worker(self):
                     )
 
                 if self.is_started.is_set():
-                    self.loop.create_task(self.restart())
+                    await self.loop.create_task(self.restart())
 
                 break
 
-            self.loop.create_task(self.handle_packet(packet))
+            await self.loop.create_task(self.handle_packet(packet))
 
         log.info("NetworkTask stopped")
 
