LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-02-17 03:51:05 +03:00
Code Issues Packages Projects Releases Wiki Activity

Except ConnectionAbortedError on TcpClient

Browse Source
This commit is contained in:
Lonami Exo 2017-12-22 11:27:57 +01:00
parent 23ab70fc29
commit 992017ddf8
1 changed files with 6 additions and 1 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) self._socket.sendall(data)
except socket.timeout as e: except socket.timeout as e:
raise TimeoutError() from e raise TimeoutError() from e
except BrokenPipeError: except (BrokenPipeError, ConnectionAbortedError):
self._raise_connection_reset() self._raise_connection_reset()
except OSError as e: except OSError as e:
if e.errno == errno.EBADF: if e.errno == errno.EBADF:
@ -139,6 +139,11 @@ class TcpClient:
partial = self._socket.recv(bytes_left) partial = self._socket.recv(bytes_left)
except socket.timeout as e: except socket.timeout as e:
raise TimeoutError() from e raise TimeoutError() from e
except ConnectionAbortedError:
# ConnectionAbortedError: [WinError 10053]
# An established connection was aborted by
# the software in your host machine.
self._raise_connection_reset()
except OSError as e: except OSError as e:
if e.errno == errno.EBADF or e.errno == errno.ENOTSOCK: if e.errno == errno.EBADF or e.errno == errno.ENOTSOCK:
self._raise_connection_reset() self._raise_connection_reset()