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

Add support for scheduled messages in iter_messages (#3127)

Browse Source
This commit is contained in:
Devesh Pal 2021-08-29 15:23:06 +05:30 committed by GitHub
parent 828cf2dcad
commit befba11657
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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/client/messages.py
View File

@ -19,7 +19,8 @@ class _MessagesIter(RequestIter):
""" """
async def _init( async def _init(
self, entity, offset_id, min_id, max_id, self, entity, offset_id, min_id, max_id,
from_user, offset_date, add_offset, filter, search, reply_to from_user, offset_date, add_offset, filter, search, reply_to,
scheduled
): ):
# Note that entity being `None` will perform a global search. # Note that entity being `None` will perform a global search.
if entity: if entity:
@ -84,6 +85,11 @@ class _MessagesIter(RequestIter):
offset_id=offset_id, offset_id=offset_id,
limit=1 limit=1
) )
elif scheduled:
self.request = functions.messages.GetScheduledHistoryRequest(
peer=entity,
hash=0
)
elif reply_to is not None: elif reply_to is not None:
self.request = functions.messages.GetRepliesRequest( self.request = functions.messages.GetRepliesRequest(
peer=self.entity, peer=self.entity,
@ -336,7 +342,8 @@ class MessageMethods:
wait_time: float = None, wait_time: float = None,
ids: 'typing.Union[int, typing.Sequence[int]]' = None, ids: 'typing.Union[int, typing.Sequence[int]]' = None,
reverse: bool = False, reverse: bool = False,
reply_to: int = None reply_to: int = None,
scheduled: bool = False
) -> 'typing.Union[_MessagesIter, _IDsIter]': ) -> 'typing.Union[_MessagesIter, _IDsIter]':
""" """
Iterator over the messages for the given chat. Iterator over the messages for the given chat.
@ -463,6 +470,10 @@ class MessageMethods:
a message and replies to it itself, that reply will not a message and replies to it itself, that reply will not
be included in the results. be included in the results.
scheduled (`bool`, optional):
If set to `True`, messages which are scheduled will be returned.
All other parameter will be ignored for this, except `entity`.
Yields Yields
Instances of `Message <telethon.tl.custom.message.Message>`. Instances of `Message <telethon.tl.custom.message.Message>`.
@ -521,7 +532,8 @@ class MessageMethods:
add_offset=add_offset, add_offset=add_offset,
filter=filter, filter=filter,
search=search, search=search,
reply_to=reply_to reply_to=reply_to,
scheduled=scheduled
) )
async def get_messages(self: 'TelegramClient', *args, **kwargs) -> 'hints.TotalList': async def get_messages(self: 'TelegramClient', *args, **kwargs) -> 'hints.TotalList':