async def check_channel(
self, ctx: Context,
channel: Optional[discord.TextChannel]) -> discord.TextChannel:
def is_correct(m: discord.Message) -> bool:
return m.author == ctx.author
if channel is None:
await ctx.send("What is the channel?")
get_channel = await self.bot.wait_for("message", check=is_correct)
get_channel.content
channel = await commands.TextChannelConverter().convert(
ctx, get_channel.content)
assert ctx.guild is not None
assert ctx.guild.me is not None
assert isinstance(ctx.author, discord.Member)
perms = channel.permissions_for(ctx.guild.me)
sender_perms = channel.permissions_for(ctx.author)
if not perms.view_channel:
raise errors.InputContentIncorrect(
"I do not have the the `View Messages` permission in that channel!"
)
elif not (perms.send_messages and perms.embed_links):
raise errors.InputContentIncorrect(
"I do not have the the `Send Message` and `Embed Links` permission in that channel!"
)
elif not sender_perms.view_channel:
raise errors.InputContentIncorrect(
"You don't have the `View Messages` permissions in that channel!"
)
return channel
async def set_admin_role_logic(
self, guild: discord.Guild, author: discord.Member,
new_role_id: Optional[str]) -> discord.Embed:
if not author.guild_permissions.administrator:
raise commands.MissingPermissions(["administrator"
]) # type: ignore
db_guild = await self.bot.guild_cache.get(guild.id)
original_role_id = db_guild.management_role
original_role = (guild.get_role(original_role_id)
if original_role_id is not None else None)
if new_role_id is None:
await self.bot.guild_cache.update_management_role(guild.id, None)
embed = discord.Embed(
title="Config updated!",
timestamp=datetime.now(timezone.utc),
colour=discord.Colour(15653155),
)
if original_role is None:
embed.description = "Management role has not changed! It remains None"
else:
embed.description = (
f"Management role updated from {original_role.mention} to None"
)
return embed
else:
if new_role_id[:3] == "<@&":
new_role_id = new_role_id[3:-1]
if isinstance(new_role_id, str):
try:
new_role_id = int(new_role_id) # type: ignore
except ValueError:
raise errors.InputContentIncorrect(
"I could not find that role! Please try again")
assert isinstance(new_role_id, int)
new_role = guild.get_role(new_role_id)
if new_role is None:
raise errors.InputContentIncorrect(
"I could not find that role! Please try again")
await self.bot.guild_cache.update_management_role(
guild.id, new_role_id)
embed = discord.Embed(
title="Config updated!",
timestamp=datetime.now(timezone.utc),
colour=discord.Colour(15653155),
)
if original_role is None:
embed.description = f"Management role updated to {new_role.mention}"
else:
embed.description = f"Management role updated from {original_role.mention} to {new_role.mention}"
return embed
async def message_or_error(
channel: discord.TextChannel,
message_id: Union[int, str]) -> discord.Message:
try:
return await channel.fetch_message(int(message_id))
except Exception as e:
if isinstance(e, discord.NotFound):
raise errors.InputContentIncorrect(
"I can't find that message!")
elif isinstance(e, ValueError):
raise errors.InputContentIncorrect(
"That is not a message id!")
else:
raise e
async def set_prefix_logic(self, guild: discord.Guild,
author: discord.Member,
new_prefix: Optional[str]) -> discord.Embed:
if not author.guild_permissions.administrator:
raise commands.MissingPermissions(["administrator"
]) # type: ignore
guild_data = await self.bot.guild_cache.get(guild.id)
current_prefix = guild_data.prefix
if new_prefix is None:
await self.bot.guild_cache.update_prefix(guild.id,
self.bot.default_prefix)
return discord.Embed(
title="Config updated!",
description=
f"Server prefix updated from `{current_prefix}` to `{self.bot.default_prefix}`",
timestamp=datetime.now(timezone.utc),
colour=discord.Colour(15653155),
)
else:
if len(new_prefix) > 1:
raise errors.InputContentIncorrect(
"Prefix's can only be 1 character long!")
else:
await self.bot.guild_cache.update_prefix(guild.id, new_prefix)
return discord.Embed(
title="Config updated!",
description=
f"Server prefix updated from `{current_prefix}` to `{new_prefix}`",
timestamp=datetime.now(timezone.utc),
colour=discord.Colour(15653155),
)
async def set_logging_logic(
self,
guild: discord.Guild,
author: discord.Member,
channel_input: Optional[str],
) -> Union[discord.Embed, str]:
if not author.guild_permissions.administrator:
raise commands.MissingPermissions(["administrator"
]) # type: ignore
original_logging_channel = await self.bot.db.get_logger(
guild.id, "main")
embed = discord.Embed(
title="Config updated!",
timestamp=datetime.now(timezone.utc),
colour=discord.Colour(15653155),
)
if channel_input is None:
if original_logging_channel is None:
embed.description = "Logging channel not updated! It remains None"
else:
await self.bot.db.remove_logger(guild, "main")
embed.description = f"Logging channel updated from <#{original_logging_channel.channel_id}> to None"
return embed
else:
if channel_input[:2] == "<#":
channel_input = channel_input[2:-1]
try:
channel_id = int(channel_input)
channel = guild.get_channel(channel_id)
if channel is None:
raise errors.InputContentIncorrect(
"I could not find that channel! Please try again")
except ValueError:
raise errors.InputContentIncorrect(
"I could not find that channel! Please try again")
if not isinstance(channel, discord.TextChannel):
raise errors.WebhookChannelNotTextChannel(
"That channel is not a text channel! "
"Try again with a text channel.")
await self.bot.db.update_logger(guild, channel.id, "main")
if original_logging_channel is None:
embed.description = f"Logging channel updated to {channel.mention}"
else:
embed.description = f"Logging channel updated from <#{original_logging_channel.channel_id}> to {channel.mention}"
return embed
async def json_edit(
self,
ctx: Context,
channel: Optional[discord.TextChannel] = None,
message_id: Optional[int] = None,
*,
json_content: Optional[str] = None,
) -> None:
print(type(channel))
channel = await self.check_channel(ctx, channel)
if channel.guild != ctx.guild:
raise errors.DifferentServer()
msg = await self.check_message_id(ctx, channel, message_id)
if msg.author != ctx.guild.me:
raise errors.DifferentAuthor()
if len(msg.embeds) == 0:
raise errors.InputContentIncorrect(
f"That message does not have an embed! Try {self.bot.command_with_prefix(ctx, 'edit')} instead"
)
elif len(msg.embeds) > 1:
raise errors.InputContentIncorrect(
"That message has more than one embed! I don't support that right now 😔"
)
old_embed = msg.embeds[0].to_dict()
new_json_content = await self.check_content(
ctx,
json_content,
ask_message="What is the new JSON content of the embed?")
if new_json_content[0:7] == "```json" and new_json_content[
-3:] == "```":
new_json_content = new_json_content[7:-3]
elif new_json_content[0:3] == "```" and new_json_content[-3:] == "```":
new_json_content = new_json_content[3:-3]
try:
new_dict_content = json.loads(new_json_content)
except json.decoder.JSONDecodeError as e:
raise errors.JSONFailure(
"The json that you specified was not correct, please check and try again.\n"
f"Get support from the docs or the support server at {self.bot.command_with_prefix(ctx, 'support')}\n"
f"Error message: {e}")
log_embed = discord.Embed(
title="Edited the embed!",
colour=discord.Colour(0xC387C1),
description=
"The original message and the new message are attached above.",
timestamp=datetime.now(timezone.utc),
)
log_embed.add_field(name="Editor", value=ctx.author.mention)
log_embed.add_field(name="channel", value=channel.mention)
if "embeds" in new_dict_content:
embeds = new_dict_content["embeds"]
content = new_dict_content.pop("content", None)
if len(embeds) > 1:
raise errors.InputContentIncorrect(
"You can't edit more than one embed at once!\nTry again with only one embed in the JSON data"
)
if content is not None and content != "":
await ctx.send(
"WARNING!\nPlease update the content separately, the content value has been ignored"
)
embed = embeds[0]
try:
if isinstance(embed["timestamp"], str):
embed.pop("timestamp")
except KeyError:
pass
await msg.edit(embed=discord.Embed.from_dict(embed))
else:
try:
if isinstance(new_dict_content["timestamp"], str):
new_dict_content.pop("timestamp")
except KeyError:
pass
await msg.edit(embed=discord.Embed.from_dict(new_dict_content))
old_file_name = f"{ctx.author.id}-{datetime.utcnow()}-old-content.json"
with open(old_file_name, "w+") as f:
json.dump(old_embed, f)
new_file_name = f"{ctx.author.id}-{datetime.utcnow()}-new-content.json"
with open(new_file_name, "w+") as f:
json.dump(new_dict_content, f)
await send_log_once(
guild_id=ctx.guild.id,
bot=self.bot,
logger_type="main",
embeds=[log_embed],
files=[
discord.File(new_file_name, filename="new-content.json"),
discord.File(old_file_name, "old-content.json"),
],
)
await ctx.send(embed=discord.Embed(
title="Edited the message!",
description=
"Wondering where the more informative message went? [Click here](https://docs.messagemanager.xyz/features/logging) for more info.",
))
os.remove(new_file_name)
os.remove(old_file_name)
async def edit_embed(
self,
ctx: Context,
channel: Optional[discord.TextChannel] = None,
message_id: Optional[int] = None,
) -> None:
channel = await self.check_channel(ctx, channel)
if channel.guild != ctx.guild:
raise errors.DifferentServer()
msg = await self.check_message_id(ctx, channel, message_id)
if msg.author != ctx.guild.me:
raise errors.DifferentAuthor()
if len(msg.embeds) == 0:
raise errors.InputContentIncorrect(
f"That message does not have an embed! Try {self.bot.command_with_prefix(ctx, 'edit')} instead"
)
elif len(msg.embeds) > 1:
raise errors.InputContentIncorrect(
"That message has more than one embed! I don't support that right now 😔"
)
title = await self.check_content(
ctx, None, ask_message="Enter the new title of the embed:")
if title[0:3] == "```" and title[-3:] == "```":
title = title[3:-3]
description = await self.check_content(
ctx,
None,
ask_message="Enter the new description (main body) of the embed:",
)
if description[0:3] == "```" and description[-3:] == "```":
description = description[3:-3]
old_embed = msg.embeds[0].to_dict()
new_embed = msg.embeds[0]
new_embed.description = description
new_embed.title = title
log_embed = discord.Embed(
title="Edited the embed!",
colour=discord.Colour(0xC387C1),
description=
"The original message and the new message are attached above.",
timestamp=datetime.now(timezone.utc),
)
log_embed.add_field(name="Editor", value=ctx.author.mention)
log_embed.add_field(name="Channel", value=channel.mention)
old_file_name = f"{ctx.author.id}-{datetime.utcnow()}-old-content.json"
with open(old_file_name, "w+") as f:
json.dump(old_embed, f)
new_file_name = f"{ctx.author.id}-{datetime.utcnow()}-new-content.json"
with open(new_file_name, "w+") as f:
json.dump(new_embed.to_dict(), f)
await send_log_once(
guild_id=ctx.guild.id,
bot=self.bot,
logger_type="main",
embeds=[log_embed],
files=[
discord.File(new_file_name, filename="new-content.json"),
discord.File(old_file_name, "old-content.json"),
],
)
await ctx.send(embed=discord.Embed(
title="Edited the message!",
description=
"Wondering where the more informative message went? [Click here](https://docs.messagemanager.xyz/features/logging) for more info.",
))
os.remove(new_file_name)
os.remove(old_file_name)
await msg.edit(embed=new_embed)