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

Handle negative limits gracefully in async generators

Browse Source 
We rely on >= 0 for setting the batch size to use (which must
be valid), so it makes sense to make negative limits equal 0.

This is similar to how asyncio.sleep(negative) sleeps 0 seconds,
despite the fact that time.sleep(negative) fails.
This commit is contained in:
Lonami Exo 2019-02-27 13:01:04 +01:00
parent d508e58d49
commit d02d0e2d5e
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
telethon/client/chats.py
View File

@ -37,7 +37,7 @@ class _ParticipantsIter(RequestIter):
functions.channels.GetFullChannelRequest(entity) functions.channels.GetFullChannelRequest(entity)
)).full_chat.participants_count )).full_chat.participants_count
if self.limit == 0: if self.limit <= 0:
raise StopAsyncIteration raise StopAsyncIteration
self.seen = set() self.seen = set()

2
telethon/client/dialogs.py
View File

@ -18,7 +18,7 @@ class _DialogsIter(RequestIter):
hash=0 hash=0
) )
if self.limit == 0: if self.limit <= 0:
# Special case, get a single dialog and determine count # Special case, get a single dialog and determine count
dialogs = await self.client(self.request) dialogs = await self.client(self.request)
self.total = getattr(dialogs, 'count', len(dialogs.dialogs)) self.total = getattr(dialogs, 'count', len(dialogs.dialogs))

2
telethon/client/messages.py
View File

@ -111,7 +111,7 @@ class _MessagesIter(RequestIter):
hash=0 hash=0
) )
if self.limit == 0: if self.limit <= 0:
# 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 = await self.client(self.request) result = await self.client(self.request)
if isinstance(result, types.messages.MessagesNotModified): if isinstance(result, types.messages.MessagesNotModified):