Renaming to something clearer

This commit is contained in:
Lonami Exo 2018-06-17 19:27:00 +02:00
parent 7faf808ad7
commit 049483a199
3 changed files with 19 additions and 17 deletions

View File

@ -219,11 +219,12 @@ class TelegramBaseClient(abc.ABC):
return self._loop return self._loop
@property @property
def connection_dropped(self): def disconnected(self):
""" """
Future that resolves when the connection to Telegram ends. Future that resolves when the connection to Telegram
ends, either by user action or in the background.
""" """
return self._sender.connection_dropped return self._sender.disconnected
# endregion # endregion

View File

@ -14,7 +14,7 @@ class UpdateMethods(UserMethods):
# region Public methods # region Public methods
def run_loop(self): def run_until_disconnected(self):
""" """
Runs the event loop until `disconnect` is called or if an error Runs the event loop until `disconnect` is called or if an error
while connecting/sending/receiving occurs in the background. In while connecting/sending/receiving occurs in the background. In
@ -22,10 +22,10 @@ class UpdateMethods(UserMethods):
to ``except`` it on your own code. to ``except`` it on your own code.
This method shouldn't be called from ``async def`` as the loop This method shouldn't be called from ``async def`` as the loop
will be running already. Use ``await client.connection_dropped`` will be running already. Use ``await client.disconnected`` in
in this situation instead. this situation instead.
""" """
self.loop.run_until_complete(self.connection_dropped) self.loop.run_until_complete(self.disconnected)
def on(self, event): def on(self, event):
""" """

View File

@ -63,7 +63,7 @@ class MTProtoSender:
# pending futures should be cancelled. # pending futures should be cancelled.
self._user_connected = False self._user_connected = False
self._reconnecting = False self._reconnecting = False
self._connection_dropped = None self._disconnected = None
# We need to join the loops upon disconnection # We need to join the loops upon disconnection
self._send_loop_handle = None self._send_loop_handle = None
@ -159,9 +159,9 @@ class MTProtoSender:
__log__.info('Disconnection from {} complete!'.format(self._ip)) __log__.info('Disconnection from {} complete!'.format(self._ip))
if error: if error:
self._connection_dropped.set_exception(error) self._disconnected.set_exception(error)
else: else:
self._connection_dropped.set_result(None) self._disconnected.set_result(None)
def send(self, request, ordered=False): def send(self, request, ordered=False):
""" """
@ -205,14 +205,15 @@ class MTProtoSender:
return message.future return message.future
@property @property
def connection_dropped(self): def disconnected(self):
""" """
Future that resolves when the connection to Telegram ends. Future that resolves when the connection to Telegram
ends, either by user action or in the background.
""" """
if self._connection_dropped is not None: if self._disconnected is not None:
return self._connection_dropped return self._disconnected
else: else:
raise ConnectionError('No connection yet') raise ConnectionError('Sender was never connected')
# Private methods # Private methods
@ -262,8 +263,8 @@ class MTProtoSender:
self._recv_loop_handle = self._loop.create_task(self._recv_loop()) self._recv_loop_handle = self._loop.create_task(self._recv_loop())
# First connection or manual reconnection after a failure # First connection or manual reconnection after a failure
if self._connection_dropped is None or self._connection_dropped.done(): if self._disconnected is None or self._disconnected.done():
self._connection_dropped = asyncio.Future() self._disconnected = asyncio.Future()
__log__.info('Connection to {} complete!'.format(self._ip)) __log__.info('Connection to {} complete!'.format(self._ip))
async def _reconnect(self): async def _reconnect(self):