LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-09-21 11:18:50 +03:00
Code Issues Packages Projects Releases Wiki Activity

TcpClient: Catch ConnectionError instead of its particular cases

Browse Source 
That can be more reliable, especially in the case of using PySocks.
This commit is contained in:
Dmitry D. Chernov 2017-12-23 05:45:23 +10:00
parent 992017ddf8
commit 4a2a64ce2f
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
telethon/extensions/tcp_client.py
View File

@ -113,7 +113,7 @@ class TcpClient:
self._socket.sendall(data)
except socket.timeout as e:
raise TimeoutError() from e
except (BrokenPipeError, ConnectionAbortedError):
except ConnectionError:
self._raise_connection_reset()
except OSError as e:
if e.errno == errno.EBADF:
@ -139,10 +139,7 @@ class TcpClient:
partial = self._socket.recv(bytes_left)
except socket.timeout as e:
raise TimeoutError() from e
except ConnectionAbortedError:
# ConnectionAbortedError: [WinError 10053]
# An established connection was aborted by
# the software in your host machine.
except ConnectionError:
self._raise_connection_reset()
except OSError as e:
if e.errno == errno.EBADF or e.errno == errno.ENOTSOCK: