Browse Source

Add reason to Thread.delete_messages and Thread.purge

pull/7831/head
Rapptz 3 years ago
parent
commit
b049cf77f2
  1. 21
      discord/threads.py

21
discord/threads.py

@ -367,7 +367,7 @@ class Thread(Messageable, Hashable):
raise ClientException('Parent channel not found')
return parent.permissions_for(obj)
async def delete_messages(self, messages: Iterable[Snowflake], /) -> None:
async def delete_messages(self, messages: Iterable[Snowflake], /, *, reason: Optional[str] = None) -> None:
"""|coro|
Deletes a list of messages. This is similar to :meth:`Message.delete`
@ -383,12 +383,12 @@ class Thread(Messageable, Hashable):
You must have the :attr:`~Permissions.manage_messages` permission to
use this.
Usable only by bot accounts.
Parameters
-----------
messages: Iterable[:class:`abc.Snowflake`]
An iterable of messages denoting which ones to bulk delete.
reason: Optional[:class:`str`]
The reason for deleting the messages. Shows up on the audit log.
Raises
------
@ -410,14 +410,14 @@ class Thread(Messageable, Hashable):
if len(messages) == 1:
message_id = messages[0].id
await self._state.http.delete_message(self.id, message_id)
await self._state.http.delete_message(self.id, message_id, reason=reason)
return
if len(messages) > 100:
raise ClientException('Can only bulk delete messages up to 100 messages')
message_ids: SnowflakeList = [m.id for m in messages]
await self._state.http.delete_messages(self.id, message_ids)
await self._state.http.delete_messages(self.id, message_ids, reason=reason)
async def purge(
self,
@ -429,6 +429,7 @@ class Thread(Messageable, Hashable):
around: Optional[SnowflakeTime] = None,
oldest_first: Optional[bool] = False,
bulk: bool = True,
reason: Optional[str] = None,
) -> List[Message]:
"""|coro|
@ -472,6 +473,8 @@ class Thread(Messageable, Hashable):
If ``True``, use bulk delete. Setting this to ``False`` is useful for mass-deleting
a bot's own messages without :attr:`Permissions.manage_messages`. When ``True``, will
fall back to single delete if messages are older than two weeks.
reason: Optional[:class:`str`]
The reason for purging the messages. Shows up on the audit log.
Raises
-------
@ -495,7 +498,7 @@ class Thread(Messageable, Hashable):
minimum_time = int((time.time() - 14 * 24 * 60 * 60) * 1000.0 - 1420070400000) << 22
async def _single_delete_strategy(messages: Iterable[Message]):
async def _single_delete_strategy(messages: Iterable[Message], *, reason: Optional[str] = None):
for m in messages:
await m.delete()
@ -504,7 +507,7 @@ class Thread(Messageable, Hashable):
async for message in iterator:
if count == 100:
to_delete = ret[-100:]
await strategy(to_delete)
await strategy(to_delete, reason=reason)
count = 0
await asyncio.sleep(1)
@ -517,7 +520,7 @@ class Thread(Messageable, Hashable):
await ret[-1].delete()
elif count >= 2:
to_delete = ret[-count:]
await strategy(to_delete)
await strategy(to_delete, reason=reason)
count = 0
strategy = _single_delete_strategy
@ -529,7 +532,7 @@ class Thread(Messageable, Hashable):
if count >= 2:
# more than 2 messages -> bulk delete
to_delete = ret[-count:]
await strategy(to_delete)
await strategy(to_delete, reason=reason)
elif count == 1:
# delete a single message
await ret[-1].delete()

Loading…
Cancel
Save