Fix check to avoid (dis)connecting if already (dis)connected

This commit is contained in:
Lonami Exo 2017-09-03 09:59:37 +02:00
parent 4de4026bb3
commit 494b82ea9d

View File

@ -127,7 +127,7 @@ class TelegramClient(TelegramBareClient):
*args will be ignored. *args will be ignored.
""" """
if self._sender.is_connected(): if self._sender and self._sender.is_connected():
return return
ok = super().connect() ok = super().connect()
@ -152,7 +152,7 @@ class TelegramClient(TelegramBareClient):
def disconnect(self): def disconnect(self):
"""Disconnects from the Telegram server """Disconnects from the Telegram server
and stops all the spawned threads""" and stops all the spawned threads"""
if not self._sender.is_connected(): if not self._sender or not self._sender.is_connected():
return return
super().disconnect() super().disconnect()
@ -914,7 +914,6 @@ class TelegramClient(TelegramBareClient):
while self._sender.is_connected(): while self._sender.is_connected():
try: try:
self._sender.receive() self._sender.receive()
print('got one')
except TimeoutError: except TimeoutError:
# No problem. # No problem.
pass pass