110 lines
3.2 KiB
Python
110 lines
3.2 KiB
Python
import asyncio
|
|
import random
|
|
|
|
import discord
|
|
from discord import Embed
|
|
from discord.ext import commands
|
|
|
|
from rich.console import Console
|
|
|
|
console = Console()
|
|
|
|
TOKEN_REPLACEMENT = "\\*" * random.randint(3, 15)
|
|
PASSWORD_REPLACEMENT = "\\*" * random.randint(3, 15)
|
|
|
|
|
|
class ContextPlus(commands.Context):
|
|
async def send(
|
|
self,
|
|
content=None,
|
|
*,
|
|
tts=False,
|
|
embed=None,
|
|
file=None,
|
|
files=None,
|
|
delete_after=None,
|
|
nonce=None,
|
|
allowed_mentions=None,
|
|
deletable=False
|
|
): # i know *args and **kwargs but, i prefer work with same values
|
|
if content:
|
|
content = content.replace(
|
|
self.bot.config.Core.token, TOKEN_REPLACEMENT
|
|
).replace(
|
|
self.bot.config.Core.Database.password, PASSWORD_REPLACEMENT
|
|
)
|
|
if embed:
|
|
e = embed.to_dict()
|
|
for key, value in e.items():
|
|
if isinstance(value, (str, bytes)):
|
|
e[key] = value.replace(
|
|
self.bot.config.Core.token, TOKEN_REPLACEMENT
|
|
).replace(
|
|
self.bot.config.Core.Database.password,
|
|
PASSWORD_REPLACEMENT,
|
|
)
|
|
embed = Embed.from_dict(e)
|
|
|
|
if (
|
|
hasattr(self.command, "deletable") and self.command.deletable
|
|
) or deletable:
|
|
message = await super().send(
|
|
content=content,
|
|
tts=tts,
|
|
embed=embed,
|
|
file=file,
|
|
files=files,
|
|
delete_after=delete_after,
|
|
nonce=nonce,
|
|
allowed_mentions=allowed_mentions,
|
|
)
|
|
await message.add_reaction("🗑")
|
|
|
|
def check(reaction: discord.Reaction, user: discord.User):
|
|
return (
|
|
user == self.author
|
|
and str(reaction.emoji) == "🗑"
|
|
and reaction.message.id == message.id
|
|
)
|
|
|
|
try:
|
|
await self.bot.wait_for(
|
|
"reaction_add", timeout=42.0, check=check
|
|
)
|
|
except asyncio.TimeoutError:
|
|
await message.remove_reaction("🗑", self.bot.user)
|
|
else:
|
|
await message.delete()
|
|
return message
|
|
|
|
return await super().send(
|
|
content=content,
|
|
tts=tts,
|
|
embed=embed,
|
|
file=file,
|
|
files=files,
|
|
delete_after=delete_after,
|
|
nonce=nonce,
|
|
allowed_mentions=allowed_mentions,
|
|
)
|
|
|
|
|
|
class CommandPLus(commands.Command):
|
|
def __init__(self, function, **kwargs):
|
|
super().__init__(function, **kwargs)
|
|
self.deletable = kwargs.pop("deletable", True)
|
|
|
|
|
|
def command_extra(*args, **kwargs):
|
|
return commands.command(*args, **kwargs, cls=CommandPLus)
|
|
|
|
|
|
class GroupPlus(commands.Group):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.deletable = kwargs.pop("deletable", True)
|
|
|
|
|
|
def group_extra(*args, **kwargs):
|
|
return commands.group(*args, **kwargs, cls=GroupPlus)
|