TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-30 17:43:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Raise and handle send errors in order to immediately act upon

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

1
pyrogram/connection/transport/tcp/tcp.py
View File

@ -101,6 +101,7 @@ class TCP:
await self.writer.drain() await self.writer.drain()
except Exception as e: except Exception as e:
log.warning("Send exception: %s %s", type(e).__name__, e) log.warning("Send exception: %s %s", type(e).__name__, e)
raise OSError(e)
async def recv(self, length: int = 0): async def recv(self, length: int = 0):
data = b"" data = b""

4
pyrogram/session/session.py
View File

@ -297,7 +297,11 @@ class Session:
self.auth_key_id self.auth_key_id
) )
try:
await self.connection.send(payload) await self.connection.send(payload)
except OSError as e:
self.results.pop(msg_id, None)
raise e
if wait_response: if wait_response:
try: try: