From 7c58bb34af1a525923610908224ed3f6b6f99d85 Mon Sep 17 00:00:00 2001 From: rvck Date: Wed, 6 Aug 2025 22:41:37 +0700 Subject: [PATCH] Update tcp.py --- pyrogram/connection/transport/tcp/tcp.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pyrogram/connection/transport/tcp/tcp.py b/pyrogram/connection/transport/tcp/tcp.py index d1ce181b..86fa6ae5 100644 --- a/pyrogram/connection/transport/tcp/tcp.py +++ b/pyrogram/connection/transport/tcp/tcp.py @@ -456,7 +456,7 @@ class TCP: try: self.writer.close() await asyncio.wait_for(self.writer.wait_closed(), self.TIMEOUT) - except Exception: + except (OSError, asyncio.TimeoutError): pass async def send(self, data: bytes) -> None: @@ -468,7 +468,7 @@ class TCP: try: self.writer.write(data) await self.writer.drain() - except Exception as e: + except (OSError, asyncio.TimeoutError) as e: raise OSError(e) async def recv(self, length: int = 0) -> Optional[bytes]: