mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2024-11-10 19:46:36 +03:00
Save self user ID in session file
Should result in one less request after connecting, as there is no longer a need to fetch the self user.
This commit is contained in:
parent
88bc6a46a6
commit
a657ae0134
|
@ -9,7 +9,7 @@ class EntityCache:
|
||||||
self,
|
self,
|
||||||
hash_map: dict = _sentinel,
|
hash_map: dict = _sentinel,
|
||||||
self_id: int = None,
|
self_id: int = None,
|
||||||
self_bot: bool = False
|
self_bot: bool = None
|
||||||
):
|
):
|
||||||
self.hash_map = {} if hash_map is _sentinel else hash_map
|
self.hash_map = {} if hash_map is _sentinel else hash_map
|
||||||
self.self_id = self_id
|
self.self_id = self_id
|
||||||
|
|
|
@ -530,7 +530,7 @@ class AuthMethods:
|
||||||
except errors.RPCError:
|
except errors.RPCError:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
self._mb_entity_cache.set_self_user(None, False, None)
|
self._mb_entity_cache.set_self_user(None, None, None)
|
||||||
self._authorized = False
|
self._authorized = False
|
||||||
|
|
||||||
await self.disconnect()
|
await self.disconnect()
|
||||||
|
|
|
@ -549,6 +549,14 @@ class TelegramBaseClient(abc.ABC):
|
||||||
self.session.auth_key = self._sender.auth_key
|
self.session.auth_key = self._sender.auth_key
|
||||||
self.session.save()
|
self.session.save()
|
||||||
|
|
||||||
|
try:
|
||||||
|
# See comment when saving entities to understand this hack
|
||||||
|
self_id = self.session.get_input_entity(0).access_hash
|
||||||
|
self_user = self.session.get_input_entity(self_id)
|
||||||
|
self._mb_entity_cache.set_self_user(self_id, None, self_user.access_hash)
|
||||||
|
except ValueError:
|
||||||
|
pass
|
||||||
|
|
||||||
if self._catch_up:
|
if self._catch_up:
|
||||||
ss = SessionState(0, 0, False, 0, 0, 0, 0, None)
|
ss = SessionState(0, 0, False, 0, 0, 0, 0, None)
|
||||||
cs = []
|
cs = []
|
||||||
|
@ -670,6 +678,11 @@ class TelegramBaseClient(abc.ABC):
|
||||||
# It doesn't matter if we put users in the list of chats.
|
# It doesn't matter if we put users in the list of chats.
|
||||||
self.session.process_entities(types.contacts.ResolvedPeer(None, [e._as_input_peer() for e in entities], []))
|
self.session.process_entities(types.contacts.ResolvedPeer(None, [e._as_input_peer() for e in entities], []))
|
||||||
|
|
||||||
|
# As a hack to not need to change the session files, save ourselves with ``id=0`` and ``access_hash`` of our ``id``.
|
||||||
|
# This way it is possible to determine our own ID by querying for 0. However, whether we're a bot is not saved.
|
||||||
|
if self._mb_entity_cache.self_id:
|
||||||
|
self.session.process_entities(types.contacts.ResolvedPeer(None, [types.InputPeerUser(0, self._mb_entity_cache.self_id)], []))
|
||||||
|
|
||||||
ss, cs = self._message_box.session_state()
|
ss, cs = self._message_box.session_state()
|
||||||
self.session.set_update_state(0, types.updates.State(**ss, unread_count=0))
|
self.session.set_update_state(0, types.updates.State(**ss, unread_count=0))
|
||||||
now = datetime.datetime.now() # any datetime works; channels don't need it
|
now = datetime.datetime.now() # any datetime works; channels don't need it
|
||||||
|
|
|
@ -188,7 +188,7 @@ class UserMethods:
|
||||||
else:
|
else:
|
||||||
print('Hello')
|
print('Hello')
|
||||||
"""
|
"""
|
||||||
if self._mb_entity_cache.self_id is None:
|
if self._mb_entity_cache.self_bot is None:
|
||||||
await self.get_me(input_peer=True)
|
await self.get_me(input_peer=True)
|
||||||
|
|
||||||
return self._mb_entity_cache.self_bot
|
return self._mb_entity_cache.self_bot
|
||||||
|
|
Loading…
Reference in New Issue
Block a user