Create a future result before sending its request

delivrance · delivrance · commit 5ca422b31425 · 2022-12-28T00:19:28.000+01:00
diff --git a/pyrogram/session/session.py b/pyrogram/session/session.py
@@ -282,6 +282,9 @@ async def send(self, data: TLObject, wait_response: bool = True, timeout: float
         message = self.msg_factory(data)
         msg_id = message.msg_id
 
+        if wait_response:
+            self.results[msg_id] = Result()
+
         log.debug("Sent: %s", message)
 
         payload = await self.loop.run_in_executor(
@@ -297,8 +300,6 @@ async def send(self, data: TLObject, wait_response: bool = True, timeout: float
         await self.connection.send(payload)
 
         if wait_response:
-            self.results[msg_id] = Result()
-
             try:
                 await asyncio.wait_for(self.results[msg_id].event.wait(), timeout)
             except asyncio.TimeoutError:
