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

Revisit catch_up (#808)

Browse Source
This commit is contained in:
Lonami Exo 2018-08-13 12:32:12 +02:00
parent 6b57dba5e1
commit 653686996a
1 changed files with 12 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
telethon/client/updates.py
View File

@ -129,29 +129,20 @@ class UpdateMethods(UserMethods):
async def catch_up(self):
state = self.session.get_update_state(0)
if not state or not state.pts:
return
state = await self(functions.updates.GetStateRequest())
self.session.catching_up = True
try:
while True:
d = await self(functions.updates.GetDifferenceRequest(
state.pts, state.date, state.qts))
if isinstance(d, types.updates.DifferenceEmpty):
state.date = d.date
state.seq = d.seq
break
elif isinstance(d, (types.updates.DifferenceSlice,
types.updates.Difference)):
state.pts, state.date, state.qts
))
if isinstance(d, (types.updates.DifferenceSlice,
types.updates.Difference)):
if isinstance(d, types.updates.Difference):
state = d.state
elif d.intermediate_state.pts > state.pts:
state = d.intermediate_state
else:
# TODO Figure out why other applications can rely on
# using always the intermediate_state to eventually
# reach a DifferenceEmpty, but that leads to an
# infinite loop here (so check against old pts to stop)
break
state = d.intermediate_state
self._handle_update(types.Updates(
users=d.users,
@ -163,7 +154,12 @@ class UpdateMethods(UserMethods):
for m in d.new_messages
]
))
elif isinstance(d, types.updates.DifferenceTooLong):
else:
if isinstance(d, types.updates.DifferenceEmpty):
state.date = d.date
state.seq = d.seq
elif isinstance(d, types.updates.DifferenceTooLong):
state.pts = d.pts
break
finally:
self.session.set_update_state(0, state)