Raise OSError in case "send" fails

2025-09-05 08:45:13 +00:00 · 2018-06-17 18:45:08 +02:00
parent 52354b93d0
commit 5d58ff2d94
1 changed files with 4 additions and 1 deletions
--- a/pyrogram/connection/connection.py
+++ b/pyrogram/connection/connection.py
@@ -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()