class MyContext(commands.Context):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.bot: 'MyBot'
self.logger = LoggerConstant(self.bot.logger, self.guild, self.channel, self.author)
async def send(self, content=None, *, delete_on_invoke_removed=True, file=None, files=None, **kwargs) -> Message:
# Case for a too-big message
if content and len(content) > 1990:
self.logger.warning("Message content is too big to be sent, putting in a text file for sending.")
message_file = discord.File(io.BytesIO(content.encode()), filename="message.txt")
content = None
if file is not None and files is not None:
raise InvalidArgument('Cannot pass both file and files parameter to send()')
elif file is not None:
files = [message_file, file]
file = None
elif files is not None:
if len(files) == 10:
raise InvalidArgument('Content is too big, and too many files were provided')
else:
files = [message_file] + files
else:
file = message_file
message = await super().send(content, file=file, files=files, **kwargs)
# Message deletion if source is deleted
if delete_on_invoke_removed:
asyncio.ensure_future(delete_messages_if_message_removed(self.bot, self.message, message))
return message
def author_permissions(self):
return self.channel.permissions_for(self.author)
async def get_language_code(self):
if self.guild:
db_guild = await get_from_db(self.guild)
language = db_guild.language
else:
db_user = await get_from_db(self.author, as_user=True)
language = db_user.language
return language
async def translate(self, message):
language_code = await self.get_language_code()
return translate(message, language_code)
async def get_translate_function(self):
language_code = await self.get_language_code()
def _(message, **kwargs):
return translate(message, language_code).format(**kwargs)
return _
class CustomContext(commands.Context):
bot: 'CustomBot'
def __init__(self, **attrs):
super().__init__(**attrs)
self.logger = LoggerConstant(self.bot.logger, self.guild, self.channel,
self.author)
async def send_to(self,
message: str,
user: typing.Optional[discord.User] = None,
**kwargs):
if user is None:
user = self.author
message = f"{user.mention} > {message}"
await self.send(message, **kwargs)
async def send(self,
content=None,
*,
file=None,
files=None,
**kwargs) -> Message:
# Case for a too-big message
if content and len(content) > 1990:
self.logger.warning(
"Message content is too big to be sent, putting in a text file for sending."
)
message_file = discord.File(io.BytesIO(content.encode()),
filename="message.txt")
content = None
if file is not None and files is not None:
raise InvalidArgument(
'Cannot pass both file and files parameter to send()')
elif file is not None:
files = [message_file, file]
file = None
elif files is not None:
if len(files) == 10:
raise InvalidArgument(
'Content is too big, and too many files were provided')
else:
files = [message_file] + files
else:
file = message_file
message = await super().send(content, file=file, files=files, **kwargs)
return message
class MyContext(commands.Context):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.bot: 'MyBot'
self.logger = LoggerConstant(self.bot.logger, self.guild, self.channel,
self.author)
async def reply(self, *args, **kwargs) -> discord.Message:
# pip3.9 install git+https://github.com/Rapptz/discord.py.git
return await self.send(*args, **kwargs, reply=True)
async def send(self,
content=None,
*,
delete_on_invoke_removed=True,
file=None,
files=None,
reply=False,
**kwargs) -> Message:
# Case for a too-big message
if content and len(content) > 1990:
self.logger.warning(
"Message content is too big to be sent, putting in a text file for sending."
)
message_file = discord.File(io.BytesIO(content.encode()),
filename="message.txt")
content = None
if file is not None and files is not None:
raise InvalidArgument(
'Cannot pass both file and files parameter to send()')
elif file is not None:
files = [message_file, file]
file = None
elif files is not None:
if len(files) == 10:
raise InvalidArgument(
'Content is too big, and too many files were provided')
else:
files = [message_file] + files
else:
file = message_file
if reply:
db_user = await get_from_db(self.author, as_user=True)
try:
message = await super().reply(
content,
file=file,
files=files,
allowed_mentions=discord.AllowedMentions(
replied_user=db_user.ping_friendly),
**kwargs)
except discord.errors.HTTPException:
# Can't reply, probably that the message we are replying to was deleted.
message = await super().send(content,
file=file,
files=files,
**kwargs)
else:
message = await super().send(content,
file=file,
files=files,
**kwargs)
# Message deletion if source is deleted
if delete_on_invoke_removed:
asyncio.ensure_future(
delete_messages_if_message_removed(self.bot, self.message,
message))
return message
def ducks(self):
self.bot: 'MyBot'
return self.bot.ducks_spawned[self.channel]
async def target_next_duck(self):
ducks = self.ducks()
try:
myduck = ducks[0]
except IndexError:
return None
if myduck.fake and len(ducks) > 1:
myduck.despawn()
return await self.target_next_duck()
await myduck.target(self.author)
if not await myduck.is_killed():
return myduck
else:
return await myduck.target(self.author)
def author_permissions(self):
return self.channel.permissions_for(self.author)
async def get_language_code(self, user_language=False):
if self.guild and not user_language:
db_guild = await get_from_db(self.guild)
language = db_guild.language
else:
db_user = await get_from_db(self.author, as_user=True)
language = db_user.language
if language == "zh-Hans":
return "zh" # Babel don't know about Simplified Chinese
else:
language = language.replace('-', '_')
return language
async def translate(self, message):
language_code = await self.get_language_code()
return translate(message, language_code)
async def ntranslate(self, singular, plural, n):
language_code = await self.get_language_code()
return ntranslate(singular, plural, n, language_code)
async def get_translate_function(self, user_language=False):
language_code = await self.get_language_code(
user_language=user_language)
return get_translate_function(self,
language_code,
additional_kwargs={'ctx': self})
async def get_ntranslate_function(self, user_language=False):
language_code = await self.get_language_code(
user_language=user_language)
return get_ntranslate_function(self, language_code, {'ctx': self})
async def is_channel_enabled(self):
db_channel = await get_from_db(self.channel)
if db_channel:
return db_channel.enabled
else:
return False