yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2026-01-01 05:04:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Raise OSError in case "send" fails

Browse source
This commit is contained in:
Dan 2018-06-17 18:45:08 +02:00
parent 52354b93d0
commit 5d58ff2d94
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
pyrogram/connection/connection.py
View file

@ -64,7 +64,10 @@ class Connection:
log.info("Disconnected")
async def send(self, data: bytes):
await self.protocol.send(data)
try:
await self.protocol.send(data)
except Exception:
raise OSError
async def recv(self) -> bytes or None:
return await self.protocol.recv()