LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-07-30 17:59:55 +03:00
Code Issues Packages Projects Releases Wiki Activity

Save update states every minute instead of only saving on disconnection

Browse Source
This commit is contained in:
vanutp 2021-05-27 13:27:32 +03:00
parent 34bb2b8fc3
commit 37fed6dd23
No known key found for this signature in database
GPG Key ID: 1E36ED1DFB6FDE78
2 changed files with 35 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
telethon/client/telegrambaseclient.py
View File

@ -605,6 +605,27 @@ class TelegramBaseClient(abc.ABC):
else:
connection._proxy = proxy
def _update_state_for(self, channel_id: 'typing.Optional[int]'):
if channel_id is None:
pts, qts, date = self._state_cache[None]
if pts and date:
self.session.set_update_state(0, types.updates.State(
pts=pts,
qts=qts,
date=date,
seq=0,
unread_count=0
))
else:
pts = self._state_cache[channel_id]
self.session.set_update_state(channel_id, types.updates.State(
pts=pts,
qts=0,
date=datetime.fromtimestamp(0),
seq=0,
unread_count=0
))
async def _disconnect_coro(self: 'TelegramClient'):
await self._disconnect()
@ -632,23 +653,9 @@ class TelegramBaseClient(abc.ABC):
await asyncio.wait(self._updates_queue)
self._updates_queue.clear()
pts, qts, date = self._state_cache[None]
if pts and date:
self.session.set_update_state(0, types.updates.State(
pts=pts,
qts=qts,
date=date,
seq=0,
unread_count=0
))
for channel_id, pts in self._state_cache.get_channel_pts().items():
self.session.set_update_state(channel_id, types.updates.State(
pts=pts,
qts=0,
date=datetime.fromtimestamp(0),
seq=0,
unread_count=0
))
self._update_state_for(None)
for channel_id in self._state_cache.get_channel_pts():
self._update_state_for(channel_id)
self.session.close()

19
telethon/client/updates.py
View File

@ -220,8 +220,8 @@ class UpdateMethods:
if not pts:
# First-time, can't get difference. Get pts instead.
result = await self(functions.channels.GetFullChannelRequest(channel_id))
pts = self._state_cache[channel_id] = result.full_chat.pts
self.session.set_update_state(channel_id, types.updates.State(pts, 0, datetime.fromtimestamp(0), 0, 0))
self._state_cache[channel_id] = result.full_chat.pts
self._update_state_for(channel_id)
return
try:
while True:
@ -249,7 +249,8 @@ class UpdateMethods:
# there is no way to get them without raising limit or GetHistoryRequest, so just break
break
finally:
self.session.set_update_state(channel_id, types.updates.State(pts, 0, datetime.fromtimestamp(0), 0, 0))
self._state_cache[channel_id] = pts
self._update_state_for(channel_id)
async def catch_up(self: 'TelegramClient', pts_total_limit=None, limit=None):
"""
@ -320,7 +321,7 @@ class UpdateMethods:
pass
finally:
self._state_cache[None] = (pts, qts, date)
self.session.set_update_state(0, types.updates.State(pts, qts, date, seq=0, unread_count=0))
self._update_state_for(None)
self.session.catching_up = False
# endregion
@ -404,10 +405,10 @@ class UpdateMethods:
except (ConnectionError, asyncio.CancelledError):
return
# Entities and cached files are not saved when they are
# inserted because this is a rather expensive operation
# (default's sqlite3 takes ~0.1s to commit changes). Do
# it every minute instead. No-op if there's nothing new.
# Entities, cached files and update states are not saved
# when they are inserted because this is a rather expensive
# operation (default's sqlite3 takes ~0.1s to commit changes).
# Do it every minute instead. No-op if there's nothing new.
self.session.save()
# We need to send some content-related request at least hourly
@ -589,6 +590,7 @@ class UpdateMethods:
utils.get_input_channel(where)
))
self._state_cache[channel_id] = result.full_chat.pts
self._update_state_for(channel_id)
return
result = await self(functions.updates.GetChannelDifferenceRequest(
@ -603,6 +605,7 @@ class UpdateMethods:
# First-time, can't get difference. Get pts instead.
result = await self(functions.updates.GetStateRequest())
self._state_cache[None] = result.pts, result.qts, result.date
self._update_state_for(None)
return
result = await self(functions.updates.GetDifferenceRequest(