TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-16 12:51:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Create a future result before sending its request

Browse Source
This commit is contained in:
Dan 2022-12-28 00:19:28 +01:00
parent 2dca5aeac2
commit 5ca422b314
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pyrogram/session/session.py
View File

@ -282,6 +282,9 @@ class Session:
message = self.msg_factory(data) message = self.msg_factory(data)
msg_id = message.msg_id msg_id = message.msg_id
if wait_response:
self.results[msg_id] = Result()
log.debug("Sent: %s", message) log.debug("Sent: %s", message)
payload = await self.loop.run_in_executor( payload = await self.loop.run_in_executor(
@ -297,8 +300,6 @@ class Session:
await self.connection.send(payload) await self.connection.send(payload)
if wait_response: if wait_response:
self.results[msg_id] = Result()
try: try:
await asyncio.wait_for(self.results[msg_id].event.wait(), timeout) await asyncio.wait_for(self.results[msg_id].event.wait(), timeout)
except asyncio.TimeoutError: except asyncio.TimeoutError: