LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-22 01:16:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add more debug traces to the messagebox

Browse Source
This commit is contained in:
Lonami Exo 2022-12-18 12:57:51 +01:00
parent 59ffad0090
commit e750eb7ab5
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
telethon/_updates/messagebox.py
View File

@ -283,6 +283,9 @@ class MessageBox:
self.getting_diff_for.update(entry for entry, gap in self.possible_gaps.items() if now > gap.deadline) self.getting_diff_for.update(entry for entry, gap in self.possible_gaps.items() if now > gap.deadline)
self.getting_diff_for.update(entry for entry, state in self.map.items() if now > state.deadline) self.getting_diff_for.update(entry for entry, state in self.map.items() if now > state.deadline)
if __debug__:
self._trace('Deadlines met, now getting diff for %r', self.getting_diff_for)
# When extending `getting_diff_for`, it's important to have the moral equivalent of # When extending `getting_diff_for`, it's important to have the moral equivalent of
# `begin_get_diff` (that is, clear possible gaps if we're now getting difference). # `begin_get_diff` (that is, clear possible gaps if we're now getting difference).
for entry in self.getting_diff_for: for entry in self.getting_diff_for:
@ -587,12 +590,15 @@ class MessageBox:
if entry not in self.map: if entry not in self.map:
raise RuntimeError('Should not try to get difference for an entry without known state') raise RuntimeError('Should not try to get difference for an entry without known state')
return fn.updates.GetDifferenceRequest( gd = fn.updates.GetDifferenceRequest(
pts=self.map[ENTRY_ACCOUNT].pts, pts=self.map[ENTRY_ACCOUNT].pts,
pts_total_limit=None, pts_total_limit=None,
date=self.date, date=self.date,
qts=self.map[ENTRY_SECRET].pts if ENTRY_SECRET in self.map else NO_SEQ, qts=self.map[ENTRY_SECRET].pts if ENTRY_SECRET in self.map else NO_SEQ,
) )
if __debug__:
self._trace('Requesting account difference %s', gd)
return gd
return None return None
@ -673,6 +679,9 @@ class MessageBox:
return updates, diff.users, diff.chats return updates, diff.users, diff.chats
def end_difference(self): def end_difference(self):
if __debug__:
self._trace('Ending account difference')
account = ENTRY_ACCOUNT in self.getting_diff_for account = ENTRY_ACCOUNT in self.getting_diff_for
secret = ENTRY_SECRET in self.getting_diff_for secret = ENTRY_SECRET in self.getting_diff_for
@ -712,13 +721,16 @@ class MessageBox:
if not state: if not state:
raise RuntimeError('Should not try to get difference for an entry without known state') raise RuntimeError('Should not try to get difference for an entry without known state')
return fn.updates.GetChannelDifferenceRequest( gd = fn.updates.GetChannelDifferenceRequest(
force=False, force=False,
channel=tl.InputChannel(packed.id, packed.hash), channel=tl.InputChannel(packed.id, packed.hash),
filter=tl.ChannelMessagesFilterEmpty(), filter=tl.ChannelMessagesFilterEmpty(),
pts=state.pts, pts=state.pts,
limit=BOT_CHANNEL_DIFF_LIMIT if chat_hashes.self_bot else USER_CHANNEL_DIFF_LIMIT limit=BOT_CHANNEL_DIFF_LIMIT if chat_hashes.self_bot else USER_CHANNEL_DIFF_LIMIT
) )
if __debug__:
self._trace('Requesting channel difference %s', gd)
return gd
# Similar to [`MessageBox::process_updates`], but using the result from getting difference. # Similar to [`MessageBox::process_updates`], but using the result from getting difference.
def apply_channel_difference( def apply_channel_difference(
@ -775,7 +787,7 @@ class MessageBox:
def end_channel_difference(self, request, reason: PrematureEndReason, chat_hashes): def end_channel_difference(self, request, reason: PrematureEndReason, chat_hashes):
entry = request.channel.channel_id entry = request.channel.channel_id
if __debug__: if __debug__:
self._trace('Ending channel difference for %r', entry) self._trace('Ending channel difference for %r because %s', entry, reason)
if reason == PrematureEndReason.TEMPORARY_SERVER_ISSUES: if reason == PrematureEndReason.TEMPORARY_SERVER_ISSUES:
# Temporary issues. End getting difference without updating the pts so we can retry later. # Temporary issues. End getting difference without updating the pts so we can retry later.