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

Handle MessagesNotModified on iter_messages

Browse Source
This commit is contained in:
Lonami Exo 2018-06-03 12:03:02 +02:00
parent 8d7c7a19c0
commit 81c61a0a2e
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
telethon/telegram_client.py
View File

@ -91,7 +91,7 @@ from .tl.types import (
ChannelParticipantsBanned, ChannelParticipantsKicked, ChannelParticipantsBanned, ChannelParticipantsKicked,
InputMessagesFilterEmpty, UpdatesCombined InputMessagesFilterEmpty, UpdatesCombined
) )
from .tl.types.messages import DialogsSlice from .tl.types.messages import DialogsSlice, MessagesNotModified
from .tl.types.account import PasswordInputSettings, NoPassword from .tl.types.account import PasswordInputSettings, NoPassword
from .tl import custom from .tl import custom
from .utils import Default from .utils import Default
@ -1220,6 +1220,9 @@ class TelegramClient(TelegramBareClient):
return return
# No messages, but we still need to know the total message count # No messages, but we still need to know the total message count
result = self(request) result = self(request)
if isinstance(result, MessagesNotModified):
_total[0] = result.count
else:
_total[0] = getattr(result, 'count', len(result.messages)) _total[0] = getattr(result, 'count', len(result.messages))
return return
@ -1279,6 +1282,10 @@ class TelegramClient(TelegramBareClient):
else: else:
r = self(messages.GetMessagesRequest(ids)) r = self(messages.GetMessagesRequest(ids))
if isinstance(r, MessagesNotModified):
for _ in ids:
yield None
entities = {utils.get_peer_id(x): x entities = {utils.get_peer_id(x): x
for x in itertools.chain(r.users, r.chats)} for x in itertools.chain(r.users, r.chats)}