LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-23 01:46:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

Support deleting more than 100 messages at once

Browse Source
This commit is contained in:
Lonami Exo 2018-06-05 21:27:49 +02:00
parent 30ad06f29a
commit ed5c44a260
2 changed files with 21 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
telethon/telegram_client.py
View File

@ -1092,25 +1092,24 @@ class TelegramClient(TelegramBareClient):
This has no effect on channels or megagroups. This has no effect on channels or megagroups.
Returns: Returns:
The :tl:`AffectedMessages`. A list of :tl:`AffectedMessages`, each item being the result
for the delete calls of the messages in chunks of 100 each.
""" """
if not utils.is_list_like(message_ids): if not utils.is_list_like(message_ids):
message_ids = (message_ids,) message_ids = (message_ids,)
message_ids = [ message_ids = (
m.id if isinstance(m, (Message, MessageService, MessageEmpty)) m.id if isinstance(m, (Message, MessageService, MessageEmpty))
else int(m) for m in message_ids else int(m) for m in message_ids
] )
if entity is None:
return self(messages.DeleteMessagesRequest(message_ids, revoke=revoke))
entity = self.get_input_entity(entity)
entity = self.get_input_entity(entity) if entity else None
if isinstance(entity, InputPeerChannel): if isinstance(entity, InputPeerChannel):
return self(channels.DeleteMessagesRequest(entity, message_ids)) return self([channels.DeleteMessagesRequest(entity, list(c))
for c in utils.chunks(message_ids)])
else: else:
return self(messages.DeleteMessagesRequest(message_ids, revoke=revoke)) return self([messages.DeleteMessagesRequest(list(c), revoke)
for c in utils.chunks(message_ids)])
def iter_messages(self, entity, limit=None, offset_date=None, def iter_messages(self, entity, limit=None, offset_date=None,
offset_id=0, max_id=0, min_id=0, add_offset=0, offset_id=0, max_id=0, min_id=0, add_offset=0,

12
telethon/utils.py
View File

@ -2,6 +2,7 @@
Utilities for working with the Telegram API itself (such as handy methods Utilities for working with the Telegram API itself (such as handy methods
to convert between an entity like an User, Chat, etc. into its Input version) to convert between an entity like an User, Chat, etc. into its Input version)
""" """
import itertools
import math import math
import mimetypes import mimetypes
import os import os
@ -56,6 +57,17 @@ class Default:
""" """
def chunks(iterable, size=100):
"""
Turns the given iterable into chunks of the specified size,
which is 100 by default since that's what Telegram uses the most.
"""
it = iter(iterable)
size -= 1
for head in it:
yield itertools.chain([head], itertools.islice(it, size))
def get_display_name(entity): def get_display_name(entity):
""" """
Gets the display name for the given entity, if it's an :tl:`User`, Gets the display name for the given entity, if it's an :tl:`User`,