python-app-rocketbot/cogs/configcog.py

from discord import Guild, TextChannel
from discord.ext import commands

from config import CONFIG
from storage import ConfigKey, Storage
from cogs.basecog import BaseCog

class ConfigCog(BaseCog):
	"""
	Cog for handling general bot configuration.
	"""
	def __init__(self, bot):
		super().__init__(bot)

	@commands.group(
		brief='Manages general bot configuration'
	)
	@commands.has_permissions(ban_members=True)
	@commands.guild_only()
	async def config(self, context: commands.Context):
		'Command group'
		if context.invoked_subcommand is None:
			await context.send_help()

	@config.command(
		name='setwarningchannel',
		brief='Sets the channel where mod warnings are posted',
		description='Run this command in the channel where bot messages ' +
			'intended for server moderators should be sent. Other bot messages ' +
			'may still be posted in the channel a command was invoked in. If ' +
			'no output channel is set, mod-related messages will not be posted!',
	)
	async def config_setwarningchannel(self, context: commands.Context) -> None:
		'Command handler'
		guild: Guild = context.guild
		channel: TextChannel = context.channel
		Storage.set_config_value(guild, ConfigKey.WARNING_CHANNEL_ID,
			context.channel.id)
		await context.message.reply(
			f'{CONFIG["success_emoji"]} Warning channel updated to {channel.mention}.',
			mention_author=False)

	@config.command(
		name='getwarningchannel',
		brief='Shows the channel where mod warnings are posted',
	)
	async def config_getwarningchannel(self, context: commands.Context) -> None:
		'Command handler'
		guild: Guild = context.guild
		channel_id = Storage.get_config_value(guild, ConfigKey.WARNING_CHANNEL_ID)
		if channel_id is None:
			await context.message.reply(
				f'{CONFIG["info_emoji"]} No warning channel is configured.',
				mention_author=False)
		else:
			channel = guild.get_channel(channel_id)
			await context.message.reply(
				f'{CONFIG["info_emoji"]} Warning channel is configured as {channel.mention}.',
				mention_author=False)

	@config.command(
		name='setwarningmention',
		brief='Sets a user/role to mention in warning messages',
		usage='<@user|@role>',
		description='Configures an role or other prefix to include at the ' +
			'beginning of warning messages. If the intent is to get the ' +
			'attention of certain users, be sure to specify a properly ' +
			'formed @ tag, not just the name of the user/role.'
	)
	async def config_setwarningmention(self, context: commands.Context, mention: str = None) -> None:
		guild: Guild = context.guild
		Storage.set_config_value(guild, ConfigKey.WARNING_MENTION, mention)
		if mention is None:
			await context.message.reply(
				f'{CONFIG["success_emoji"]} Warning messages will not tag anyone.',
				mention_author=False)
		else:
			await context.message.reply(
				f'{CONFIG["success_emoji"]} Warning messages will now tag {mention}.',
				mention_author=False)

	@config.command(
		name='getwarningmention',
		brief='Shows the user/role to mention in warning messages',
		description='Shows the text, if any, that will be prefixed on any ' +
			'warning messages.'
	)
	async def config_getwarningmention(self, context: commands.Context) -> None:
		guild: Guild = context.guild
		mention: str = Storage.get_config_value(guild, ConfigKey.WARNING_MENTION)
		if mention is None:
			await context.message.reply(
				f'{CONFIG["info_emoji"]} No warning mention configured.',
				mention_author=False)
		else:
			await context.message.reply(
				f'{CONFIG["info_emoji"]} Warning messages will tag {mention}',
				mention_author=False)