Fix connect not saving different authkeys

This commit is contained in:
Lonami Exo 2018-06-26 16:20:30 +02:00
parent 9159e2a720
commit 2b090f8888
2 changed files with 5 additions and 2 deletions

View File

@ -271,7 +271,7 @@ class TelegramBaseClient(abc.ABC):
"""
Connects to Telegram.
"""
had_auth = self.session.auth_key is not None
old_auth = self.session.auth_key
await self._sender.connect(
self.session.server_address, self.session.port)
@ -280,7 +280,7 @@ class TelegramBaseClient(abc.ABC):
self._updates_handle = self._loop.create_task(self._update_loop())
if not had_auth:
if old_auth != self._sender.state.auth_key:
self.session.auth_key = self._sender.state.auth_key
self.session.save()

View File

@ -38,3 +38,6 @@ class AuthKey:
# Calculates the message key from the given data
return int.from_bytes(sha1(data).digest()[4:20], 'little', signed=True)
def __eq__(self, other):
return isinstance(other, type(self)) and other.key == self.key