mrkrk/melisa
1
0
Fork 0
mirror of https://github.com/MelisaDev/melisa.git synced 2024-11-11 19:07:28 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

channel delete message, bulk delete, purge

Browse source
This commit is contained in:
grey-cat-1908 2022-03-27 10:22:08 +03:00
parent 95c6713586
commit 6eeea6e977
3 changed files with 189 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
melisa/__init__.py
View file

@ -2,7 +2,7 @@
# Full MIT License can be found in `LICENSE.txt` at the project root.
from .client import *
from .models import Intents
from .models import Intents, Snowflake
from .exceptions import *
__package__ = "melisa"

27
melisa/core/http.py
View file

@ -4,7 +4,7 @@
from __future__ import annotations
import asyncio
from typing import Dict, Optional
from typing import Dict, Optional, Any
from aiohttp import ClientSession, ClientResponse
@ -65,6 +65,7 @@ class HTTPClient:
endpoint: str,
*,
_ttl: int = None,
headers: Optional[Dict[str, Any]] = None,
params: Optional[Dict] = None,
**kwargs,
) -> Optional[Dict]:
@ -80,7 +81,14 @@ class HTTPClient:
url = f"{self.url}/{endpoint}"
async with self.__aiohttp_session.request(
method, url, params=remove_none(params), **kwargs
method,
url,
params=remove_none(params),
headers={
"Content-Type": "application/json",
**(remove_none(headers) or {}),
},
**kwargs,
) as response:
return await self.__handle_response(
response, method, endpoint, _ttl=ttl, **kwargs
@ -100,6 +108,9 @@ class HTTPClient:
self.__rate_limiter.save_response_bucket(endpoint, method, res.headers)
if res.ok:
if res.status == 204:
return
return await res.json()
exception = self.__http_exceptions.get(res.status)
@ -138,7 +149,9 @@ class HTTPClient:
"""
return await self.__send("GET", route, params=params)
async def post(self, route: str, data: Optional[Dict] = None) -> Optional[Dict]:
async def post(
self, route: str, *, headers: dict = None, data: Optional[Dict] = None
) -> Optional[Dict]:
"""|coro|
Sends a POST request to a Discord REST API endpoint.
@ -154,13 +167,9 @@ class HTTPClient:
Optional[:class:`Dict`]
JSON response from the Discord API.
"""
return await self.__send(
"POST",
route,
json=data,
)
return await self.__send("POST", route, json=data, headers=headers)
async def delete(self, route: str, headers: dict = None) -> Optional[Dict]:
async def delete(self, route: str, *, headers: dict = None) -> Optional[Dict]:
"""|coro|
Sends a DELETE request to a Discord REST API endpoint.

178
melisa/models/guild/channel.py
View file

@ -3,6 +3,7 @@
from __future__ import annotations
import asyncio
from dataclasses import dataclass
from enum import IntEnum
from typing import List, Any, Optional, AsyncIterator, Union, Dict
@ -138,14 +139,21 @@ class TextChannel(Channel):
Parameters
----------
around : Optional[Union[:class:`int`, :class:`str`, :class:`Snowflake`]]
Get messages around this message ID.
before : Optional[Union[:class:`int`, :class:`str`, :class:`Snowflake`]]
Get messages before this message ID.
after : Optional[Union[:class:`int`, :class:`str`, :class:`Snowflake`]]
Get messages after this message ID.
limit : Optional[Union[:class:`int`, :class:`str`, :class:`Snowflake`]]
limit : Optional[Union[:class:`int`, :class:`str`, :class:`~.melisa.Snowflake`]]
Max number of messages to return (1-100).
around : Optional[Union[:class:`int`, :class:`str`, :class:`~.melisa.Snowflake`]]
Get messages around this message ID.
before : Optional[Union[:class:`int`, :class:`str`, :class:`~.melisa.Snowflake`]]
Get messages before this message ID.
after : Optional[Union[:class:`int`, :class:`str`, :class:`~.melisa.Snowflake`]]
Get messages after this message ID.
Raises
-------
NotFoundError
If channel is not found.
ForbiddenError
You do not have proper permissions to do the actions required.
Returns
-------
@ -153,6 +161,8 @@ class TextChannel(Channel):
An iterator of messages.
"""
# ToDo: Add check parameter
if limit is None:
limit = 100
@ -178,8 +188,160 @@ class TextChannel(Channel):
before = raw_messages[-1]["id"]
limit -= search_limit
async def fetch_message(
self,
message_id: Optional[Snowflake, int, str],
) -> Dict[str, Any]:
"""|coro|
Returns a specific message in the channel.
Parameters
----------
message_id : Optional[Union[:class:`int`, :class:`str`, :class:`~.melisa.Snowflake`]]
Id of message to fetch.
Raises
-------
NotFoundError
If message is not found.
ForbiddenError
You do not have proper permissions to do the actions required.
Returns
-------
Dict[:class:`str`, Any]
Message object.
"""
message = await self._http.get(
f"/channels/{self.id}/messages/{message_id}",
)
return message
async def bulk_delete_messages(
self, messages: List[Snowflake], *, reason: Optional[str] = None
):
"""|coro|
Delete multiple messages in a single request.
This method will not delete messages older than 2 weeks.
Parameters
----------
messages: List[:class:`~.melisa.Snowflake`]
The list of message IDs to delete (2-100).
reason: Optional[:class:`str`]
The reason of the bulk delete messages operation.
Raises
-------
BadRequestError
if any message provided is older than that or if any duplicate message IDs are provided.
ForbiddenError
You do not have proper permissions to do the actions required.
(You must have `MANAGE_MESSAGES` permission)
"""
await self._http.post(
f"channels/{self.id}/messages/bulk-delete",
headers={"X-Audit-Log-Reason": reason},
data={"messages": messages},
)
async def delete_message(
self, message_id: Optional[Snowflake, str, int], *, reason: Optional[str] = None
):
"""|coro|
Deletes only one specified message.
Parameters
----------
message_id: Union[:class:`int`, :class:`str`, :class:`~.melisa.Snowflake`]
Id of message to delete.
reason: Optional[:class:`str`]
The reason of the message delete operation.
Raises
-------
BadRequestError
Something is wrong with request, maybe specified parameters.
NotFoundError
If message is not found.
ForbiddenError
You do not have proper permissions to do the actions required.
(You must have `MANAGE_MESSAGES` permission)
"""
await self._http.delete(
f"channels/{self.id}/messages/{message_id}",
headers={"X-Audit-Log-Reason": reason},
)
async def purge(
self,
limit: int = 50,
*,
before: Optional[Union[int, str, Snowflake]] = None,
after: Optional[Union[int, str, Snowflake]] = None,
around: Optional[Union[int, str, Snowflake]] = None,
reason: Optional[str] = None,
):
"""|coro|
Purges a list of messages that meet the criteria specified in parameters.
This method will not delete messages older than 2 weeks.
Parameters
----------
limit : Optional[Union[:class:`int`, :class:`str`, :class:`~.melisa.Snowflake`]]
Max number of messages to purge.
around : Optional[Union[:class:`int`, :class:`str`, :class:`~.melisa.Snowflake`]]
Get messages around this message ID.
before : Optional[Union[:class:`int`, :class:`str`, :class:`~.melisa.Snowflake`]]
Get messages before this message ID.
after : Optional[Union[:class:`int`, :class:`str`, :class:`~.melisa.Snowflake`]]
Get messages after this message ID.
reason: Optional[:class:`str`]
The reason of the channel purge operation.
Raises
-------
BadRequestError
if any message provided is older than that or if any duplicate message IDs are provided.
ForbiddenError
You do not have proper permissions to do the actions required.
(You must have `MANAGE_MESSAGES` permission)
"""
iterator = self.history(
limit,
around=around,
before=before,
after=after,
)
message_ids = []
count = 0
async for message in iterator:
message_ids.append(message["id"])
count += 1
if count == 100:
print("abobatelecom")
await self.bulk_delete_messages(message_ids, reason=reason)
message_ids = []
count = 0
await asyncio.sleep(1)
await asyncio.sleep(1)
if count > 1:
await self.bulk_delete_messages(message_ids, reason=reason)
return
if count == 0:
await self.delete_message(message_ids[0], reason=reason)
return
# noinspection PyTypeChecker
channel_types_for_converting: Dict[ChannelType, Any] = {
channel_types_for_converting: Dict[ChannelType, Channel] = {
ChannelType.GUILD_TEXT: TextChannel
}