mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2024-11-26 11:23:46 +03:00
First attempt at updates catch_up for private chats/groups
This commit is contained in:
parent
6652fe276c
commit
37436bdeee
|
@ -16,9 +16,13 @@ from mimetypes import guess_type
|
||||||
from .crypto import CdnDecrypter
|
from .crypto import CdnDecrypter
|
||||||
from .tl import TLObject
|
from .tl import TLObject
|
||||||
from .tl.custom import InputSizedFile
|
from .tl.custom import InputSizedFile
|
||||||
|
from .tl.functions.updates import GetDifferenceRequest
|
||||||
from .tl.functions.upload import (
|
from .tl.functions.upload import (
|
||||||
SaveBigFilePartRequest, SaveFilePartRequest, GetFileRequest
|
SaveBigFilePartRequest, SaveFilePartRequest, GetFileRequest
|
||||||
)
|
)
|
||||||
|
from .tl.types.updates import (
|
||||||
|
DifferenceSlice, DifferenceEmpty, Difference, DifferenceTooLong
|
||||||
|
)
|
||||||
from .tl.types.upload import FileCdnRedirect
|
from .tl.types.upload import FileCdnRedirect
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
@ -2400,6 +2404,42 @@ class TelegramClient(TelegramBareClient):
|
||||||
def list_update_handlers(self):
|
def list_update_handlers(self):
|
||||||
return [callback for callback, _ in self.list_event_handlers()]
|
return [callback for callback, _ in self.list_event_handlers()]
|
||||||
|
|
||||||
|
def catch_up(self):
|
||||||
|
state = self.session.get_update_state(0)
|
||||||
|
self.session.catching_up = True
|
||||||
|
try:
|
||||||
|
while True:
|
||||||
|
d = self(GetDifferenceRequest(state.pts, state.date, state.qts))
|
||||||
|
if isinstance(d, DifferenceEmpty):
|
||||||
|
state.date = d.date
|
||||||
|
state.seq = d.seq
|
||||||
|
break
|
||||||
|
elif isinstance(d, (DifferenceSlice, Difference)):
|
||||||
|
if isinstance(d, 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
|
||||||
|
|
||||||
|
self.updates.process(Updates(
|
||||||
|
users=d.users,
|
||||||
|
chats=d.chats,
|
||||||
|
date=state.date,
|
||||||
|
seq=state.seq,
|
||||||
|
updates=d.other_updates + [UpdateNewMessage(m, 0, 0)
|
||||||
|
for m in d.new_messages]
|
||||||
|
))
|
||||||
|
elif isinstance(d, DifferenceTooLong):
|
||||||
|
break
|
||||||
|
finally:
|
||||||
|
self.session.set_update_state(0, state)
|
||||||
|
self.session.catching_up = False
|
||||||
|
|
||||||
# endregion
|
# endregion
|
||||||
|
|
||||||
# region Small utilities to make users' life easier
|
# region Small utilities to make users' life easier
|
||||||
|
|
Loading…
Reference in New Issue
Block a user