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

Add timeout to connect()

Browse Source
This commit is contained in:
Lonami Exo 2017-06-22 10:39:00 +02:00
parent e7fac8e254
commit 52a42661ee
2 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
telethon/telegram_bare_client.py
View File

@ -70,12 +70,14 @@ class TelegramBareClient:
# region Connecting
def connect(self, exported_auth=None):
def connect(self, timeout=timedelta(seconds=5), exported_auth=None):
"""Connects to the Telegram servers, executing authentication if
required. Note that authenticating to the Telegram servers is
not the same as authenticating the desired user itself, which
may require a call (or several) to 'sign_in' for the first time.
The specified timeout will be used on internal .invoke()'s.
If 'exported_auth' is not None, it will be used instead to
determine the authorization key for the current session.
"""
@ -115,13 +117,14 @@ class TelegramBareClient:
query=query)
result = self.invoke(
InvokeWithLayerRequest(
layer=layer, query=request))
InvokeWithLayerRequest(layer=layer, query=request),
timeout=timeout
)
if exported_auth is not None:
# TODO Don't actually need this for exported authorizations,
# they're only valid on such data center.
result = self.invoke(GetConfigRequest())
result = self.invoke(GetConfigRequest(), timeout=timeout)
# We're only interested in the DC options,
# although many other options are available!

4
telethon/telegram_client.py
View File

@ -129,12 +129,14 @@ class TelegramClient(TelegramBareClient):
# region Connecting
def connect(self, *args):
def connect(self, timeout=timedelta(seconds=5), *args):
"""Connects to the Telegram servers, executing authentication if
required. Note that authenticating to the Telegram servers is
not the same as authenticating the desired user itself, which
may require a call (or several) to 'sign_in' for the first time.
The specified timeout will be used on internal .invoke()'s.
*args will be ignored.
"""
return super(TelegramClient, self).connect()