hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2026-09-13 04:05:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Workaround #1134 by early checking if proxy closes connection

Browse Source
This commit is contained in:
Lonami Exo
2019-03-21 11:22:09 +01:00
parent f6c4ab6f41
commit 2e4476a754
2 changed files with 31 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
telethon/network/connection/connection.py
+8 -5
View File
@@ -39,10 +39,7 @@ class Connection(abc.ABC):
self._send_queue = asyncio.Queue(1)
self._recv_queue = asyncio.Queue(1)
async def connect(self, timeout=None, ssl=None):
"""
Establishes a connection with the server.
"""
async def _connect(self, timeout=None, ssl=None):
if not self._proxy:
self._reader, self._writer = await asyncio.wait_for(
asyncio.open_connection(
@@ -81,11 +78,17 @@ class Connection(abc.ABC):
self._reader, self._writer = \
await asyncio.open_connection(sock=s, loop=self._loop)
self._connected = True
self._codec = self.packet_codec(self)
self._init_conn()
await self._writer.drain()
async def connect(self, timeout=None, ssl=None):
"""
Establishes a connection with the server.
"""
await self._connect(timeout=timeout, ssl=ssl)
self._connected = True
self._send_task = self._loop.create_task(self._send_loop())
self._recv_task = self._loop.create_task(self._recv_loop())