LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-22 17:36:34 +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 # 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 """Connects to the Telegram servers, executing authentication if
required. Note that authenticating to the Telegram servers is required. Note that authenticating to the Telegram servers is
not the same as authenticating the desired user itself, which not the same as authenticating the desired user itself, which
may require a call (or several) to 'sign_in' for the first time. 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 If 'exported_auth' is not None, it will be used instead to
determine the authorization key for the current session. determine the authorization key for the current session.
""" """
@ -115,13 +117,14 @@ class TelegramBareClient:
query=query) query=query)
result = self.invoke( result = self.invoke(
InvokeWithLayerRequest( InvokeWithLayerRequest(layer=layer, query=request),
layer=layer, query=request)) timeout=timeout
)
if exported_auth is not None: if exported_auth is not None:
# TODO Don't actually need this for exported authorizations, # TODO Don't actually need this for exported authorizations,
# they're only valid on such data center. # 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, # We're only interested in the DC options,
# although many other options are available! # although many other options are available!

4
telethon/telegram_client.py
View File

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