hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-06-19 03:26:39 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use larger batches for .get_dialogs(limit=None)

Browse Source
This commit is contained in:
Lonami Exo 2017-11-04 12:34:44 +01:00
parent 0bfd8ff032
commit 1741608f28
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
telethon/telegram_client.py
View File

@ -279,22 +279,21 @@ class TelegramClient(TelegramBareClient):
The peer to be used as an offset. The peer to be used as an offset.
:return: A tuple of lists ([dialogs], [entities]). :return: A tuple of lists ([dialogs], [entities]).
""" """
if limit is None: limit = float('inf') if limit is None else int(limit)
limit = float('inf') if limit == 0:
return [], []
dialogs = {} # Use peer id as identifier to avoid dupes dialogs = {} # Use peer id as identifier to avoid dupes
messages = {} # Used later for sorting TODO also return these? messages = {} # Used later for sorting TODO also return these?
entities = {} entities = {}
while len(dialogs) < limit: while len(dialogs) < limit:
need = limit - len(dialogs) real_limit = min(limit - len(dialogs), 100)
r = self(GetDialogsRequest( r = self(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=need if need < float('inf') else 0 limit=real_limit
)) ))
if not r.dialogs:
break
for d in r.dialogs: for d in r.dialogs:
dialogs[utils.get_peer_id(d.peer, True)] = d dialogs[utils.get_peer_id(d.peer, True)] = d
@ -307,8 +306,9 @@ class TelegramClient(TelegramBareClient):
for c in r.chats: for c in r.chats:
entities[c.id] = c entities[c.id] = c
if not isinstance(r, DialogsSlice): if len(r.dialogs) < real_limit or not isinstance(r, DialogsSlice):
# Don't enter next iteration if we already got all # Less than we requested means we reached the end, or
# we didn't get a DialogsSlice which means we got all.
break break
offset_date = r.messages[-1].date offset_date = r.messages[-1].date