hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-05 03:22:29 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Support connection timeout

Browse Source
This commit is contained in:
Lonami Exo
2018-10-04 16:39:57 +02:00
parent 0cc8bca098
commit db83709c6b
9 changed files with 30 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
telethon/network/connection/connection.py
View File

@@ -16,7 +16,7 @@ class Connection(abc.ABC):
under any conditions for as long as the user doesn't disconnect or
the input parameters to auto-reconnect dictate otherwise.
"""
# TODO Support proxy. Support timeout?
# TODO Support proxy
def __init__(self, ip, port, *, loop):
self._ip = ip
self._port = port
@@ -31,12 +31,14 @@ class Connection(abc.ABC):
self._send_queue = asyncio.Queue(1)
self._recv_queue = asyncio.Queue(1)
async def connect(self):
async def connect(self, timeout=None):
"""
Establishes a connection with the server.
"""
self._reader, self._writer = await asyncio.open_connection(
self._ip, self._port, loop=self._loop)
self._reader, self._writer = await asyncio.wait_for(
asyncio.open_connection(self._ip, self._port, loop=self._loop),
loop=self._loop, timeout=timeout
)
self._disconnected.clear()
self._disconnected_future = None