LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-02-24 15:30:48 +03:00
Code Issues Packages Projects Releases Wiki Activity

Support filtering by folder on iter_dialogs

Browse Source
This commit is contained in:
Lonami Exo 2019-05-10 15:45:24 +02:00
parent e408550553
commit 1828dca0b9
1 changed files with 38 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
telethon/client/dialogs.py
View File

@ -14,14 +14,15 @@ if typing.TYPE_CHECKING:
class _DialogsIter(RequestIter): class _DialogsIter(RequestIter):
async def _init( async def _init(
self, offset_date, offset_id, offset_peer, ignore_migrated self, offset_date, offset_id, offset_peer, ignore_migrated, folder
): ):
self.request = functions.messages.GetDialogsRequest( self.request = functions.messages.GetDialogsRequest(
offset_date=offset_date, offset_date=offset_date,
offset_id=offset_id, offset_id=offset_id,
offset_peer=offset_peer, offset_peer=offset_peer,
limit=1, limit=1,
hash=0 hash=0,
folder_id=folder
) )
if self.limit <= 0: if self.limit <= 0:
@ -108,7 +109,9 @@ class DialogMethods(UserMethods):
offset_date: 'hints.DateLike' = None, offset_date: 'hints.DateLike' = None,
offset_id: int = 0, offset_id: int = 0,
offset_peer: 'hints.EntityLike' = types.InputPeerEmpty(), offset_peer: 'hints.EntityLike' = types.InputPeerEmpty(),
ignore_migrated: bool = False ignore_migrated: bool = False,
folder: int = None,
archived: bool = None
) -> _DialogsIter: ) -> _DialogsIter:
""" """
Iterator over the dialogs (open conversations/subscribed channels). Iterator over the dialogs (open conversations/subscribed channels).
@ -136,6 +139,25 @@ class DialogMethods(UserMethods):
dialogs are returned, but setting this to ``True`` will hide dialogs are returned, but setting this to ``True`` will hide
them in the same way official applications do. them in the same way official applications do.
folder (`int`, optional):
The folder from which the dialogs should be retrieved.
If left unspecified, all dialogs (including those from
folders) will be returned.
If set to ``0``, all dialogs that don't belong to any
folder will be returned.
If set to a folder number like ``1``, only those from
said folder will be returned.
By default Telegram assigns the folder ID ``1`` to
archived chats, so you should use that if you need
to fetch the archived dialogs.
archived (`bool`, optional):
Alias for `folder`. If unspecified, all will be returned,
``False`` implies ``folder=0`` and ``True`` implies ``folder=1``.
Yields: Yields:
Instances of `telethon.tl.custom.dialog.Dialog`. Instances of `telethon.tl.custom.dialog.Dialog`.
@ -153,14 +175,26 @@ class DialogMethods(UserMethods):
# Get drafts # Get drafts
drafts = client.get_drafts() drafts = client.get_drafts()
# Getting only non-archived dialogs (both equivalent)
non_archived = client.get_dialogs(folder=0)
non_archived = client.get_dialogs(archived=False)
# Getting only archived dialogs (both equivalent)
archived = client.get_dialogs(folder=1)
non_archived = client.get_dialogs(archived=True)
""" """
if archived is not None:
folder = 1 if archived else 0
return _DialogsIter( return _DialogsIter(
self, self,
limit, limit,
offset_date=offset_date, offset_date=offset_date,
offset_id=offset_id, offset_id=offset_id,
offset_peer=offset_peer, offset_peer=offset_peer,
ignore_migrated=ignore_migrated ignore_migrated=ignore_migrated,
folder=folder
) )
async def get_dialogs(self: 'TelegramClient', *args, **kwargs) -> 'hints.TotalList': async def get_dialogs(self: 'TelegramClient', *args, **kwargs) -> 'hints.TotalList':