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

Support ignore_migrated in iter_dialogs

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

14
telethon/client/dialogs.py
View File

@ -15,7 +15,8 @@ class DialogMethods(UserMethods):
@async_generator @async_generator
async def iter_dialogs( async def iter_dialogs(
self, limit=None, offset_date=None, offset_id=0, self, limit=None, offset_date=None, offset_id=0,
offset_peer=types.InputPeerEmpty(), _total=None): offset_peer=types.InputPeerEmpty(), ignore_migrated=False,
_total=None):
""" """
Returns an iterator over the dialogs, yielding 'limit' at most. Returns an iterator over the dialogs, yielding 'limit' at most.
Dialogs are the open "chats" or conversations with other people, Dialogs are the open "chats" or conversations with other people,
@ -38,6 +39,12 @@ class DialogMethods(UserMethods):
offset_peer (:tl:`InputPeer`, optional): offset_peer (:tl:`InputPeer`, optional):
The peer to be used as an offset. The peer to be used as an offset.
ignore_migrated (`bool`, optional):
Whether :tl:`Chat` that have ``migrated_to`` a :tl:`Channel`
should be included or not. By default all the chats in your
dialogs are returned, but setting this to ``True`` will hide
them in the same way official applications do.
_total (`list`, optional): _total (`list`, optional):
A single-item list to pass the total parameter by reference. A single-item list to pass the total parameter by reference.
@ -84,7 +91,10 @@ class DialogMethods(UserMethods):
peer_id = utils.get_peer_id(d.peer) peer_id = utils.get_peer_id(d.peer)
if peer_id not in seen: if peer_id not in seen:
seen.add(peer_id) seen.add(peer_id)
await yield_(custom.Dialog(self, d, entities, messages)) cd = custom.Dialog(self, d, entities, messages)
if not ignore_migrated or getattr(
cd.entity, 'migrated_to', None) is None:
await yield_(cd)
if len(r.dialogs) < req.limit\ if len(r.dialogs) < req.limit\
or not isinstance(r, types.messages.DialogsSlice): or not isinstance(r, types.messages.DialogsSlice):