forked from cswimr/SeaCogs
feat(aurora): splitting out the role configuration into different commands
This commit is contained in:
parent
606cd76ec4
commit
e28e32d0c0
GPG key ID:
D74DDDDF420E13DF
3 changed files with 63 additions and 17 deletions
|
|
@ -18,6 +18,7 @@ from discord.ext import tasks
|
|||
from pytimeparse2 import disable_dateutil, parse
|
||||
from redbot.core import app_commands, commands, data_manager
|
||||
from redbot.core.app_commands import Choice
|
||||
from redbot.core.bot import Red
|
||||
from redbot.core.utils.chat_formatting import box, error, warning
|
||||
|
||||
from .abc import CompositeMetaClass
|
||||
|
|
@ -65,7 +66,7 @@ class Aurora(Configuration, commands.Cog, metaclass=CompositeMetaClass): # pylin
|
|||
else:
|
||||
logger.warning("Invalid requester passed to red_delete_data_for_user: %s", requester)
|
||||
|
||||
def __init__(self, bot):
|
||||
def __init__(self, bot: Red):
|
||||
super().__init__()
|
||||
self.bot = bot
|
||||
register_config(config)
|
||||
|
|
|
|||
|
|
@ -1,24 +1,39 @@
|
|||
from redbot.core import commands
|
||||
from redbot.core.utils.chat_formatting import error, warning
|
||||
|
||||
from .embed import embed
|
||||
from .embed import addrole_embed, embed, immune_embed
|
||||
from ..abc import Mixin
|
||||
from ..importers.aurora import ImportAuroraView
|
||||
from ..importers.galacticbot import ImportGalacticBotView
|
||||
|
||||
|
||||
class Configuration(Mixin):
|
||||
"""Configuration commands for Aurora."""
|
||||
|
||||
@commands.group(autohelp=True, aliases=['moderation', 'mod'])
|
||||
@commands.group(autohelp=True, aliases=["moderation", "mod"])
|
||||
async def aurora(self, ctx: commands.Context):
|
||||
"""Settings and miscellaneous commands for Aurora."""
|
||||
|
||||
@aurora.command(name="settings", aliases=['config', 'options', 'set'])
|
||||
@aurora.command(name="settings", aliases=["config", "options", "set"])
|
||||
async def aurora_settings(self, ctx: commands.Context):
|
||||
"""View Aurora configuration settings."""
|
||||
await ctx.send(embed=await embed(ctx))
|
||||
|
||||
@aurora.group(autohelp=True, name='import')
|
||||
@aurora_settings.command(name="addrole", aliases=["removerole"])
|
||||
@commands.admin_or_permissions(manage_guild=True)
|
||||
async def aurora_settings_addrole(self, ctx: commands.Context):
|
||||
"""Manage the addrole whitelist.
|
||||
|
||||
Roles added to this list are also applied to `/removerole`."""
|
||||
await ctx.send(embed=await addrole_embed(ctx))
|
||||
|
||||
@aurora_settings.command(name="immunity")
|
||||
@commands.admin_or_permissions(manage_guild=True)
|
||||
async def aurora_settings_immunity(self, ctx: commands.Context):
|
||||
"""Manage the immunity whitelist."""
|
||||
await ctx.send(embed=await immune_embed(ctx))
|
||||
|
||||
@aurora.group(autohelp=True, name="import")
|
||||
@commands.admin()
|
||||
@commands.guild_only()
|
||||
async def aurora_import(self, ctx: commands.Context):
|
||||
|
|
@ -28,8 +43,16 @@ class Configuration(Mixin):
|
|||
@commands.admin()
|
||||
async def aurora_import_aurora(self, ctx: commands.Context):
|
||||
"""Import moderations from another bot using Aurora."""
|
||||
if ctx.message.attachments and ctx.message.attachments[0].content_type == 'application/json; charset=utf-8':
|
||||
message = await ctx.send(warning("Are you sure you want to import moderations from another bot?\n**This will overwrite any moderations that already exist in this guild's moderation table.**\n*The import process will block the rest of your bot until it is complete.*"))
|
||||
if (
|
||||
ctx.message.attachments
|
||||
and ctx.message.attachments[0].content_type
|
||||
== "application/json; charset=utf-8"
|
||||
):
|
||||
message = await ctx.send(
|
||||
warning(
|
||||
"Are you sure you want to import moderations from another bot?\n**This will overwrite any moderations that already exist in this guild's moderation table.**\n*The import process will block the rest of your bot until it is complete.*"
|
||||
)
|
||||
)
|
||||
await message.edit(view=ImportAuroraView(60, ctx, message))
|
||||
else:
|
||||
await ctx.send(error("Please provide a valid Aurora export file."))
|
||||
|
|
@ -38,8 +61,18 @@ class Configuration(Mixin):
|
|||
@commands.admin()
|
||||
async def aurora_import_galacticbot(self, ctx: commands.Context):
|
||||
"""Import moderations from GalacticBot."""
|
||||
if ctx.message.attachments and ctx.message.attachments[0].content_type == 'application/json; charset=utf-8':
|
||||
message = await ctx.send(warning("Are you sure you want to import GalacticBot moderations?\n**This will overwrite any moderations that already exist in this guild's moderation table.**\n*The import process will block the rest of your bot until it is complete.*"))
|
||||
if (
|
||||
ctx.message.attachments
|
||||
and ctx.message.attachments[0].content_type
|
||||
== "application/json; charset=utf-8"
|
||||
):
|
||||
message = await ctx.send(
|
||||
warning(
|
||||
"Are you sure you want to import GalacticBot moderations?\n**This will overwrite any moderations that already exist in this guild's moderation table.**\n*The import process will block the rest of your bot until it is complete.*"
|
||||
)
|
||||
)
|
||||
await message.edit(view=ImportGalacticBotView(60, ctx, message))
|
||||
else:
|
||||
await ctx.send(error("Please provide a valid GalacticBot moderation export file."))
|
||||
await ctx.send(
|
||||
error("Please provide a valid GalacticBot moderation export file.")
|
||||
)
|
||||
|
|
|
|||
|
|
@ -78,15 +78,15 @@ async def _guild(guild: Guild) -> str:
|
|||
guild_str = '\n'.join(guild_str)
|
||||
return guild_str
|
||||
|
||||
async def _blacklist(guild: Guild) -> str:
|
||||
"""Generates a configuration menu field value for a guild's blacklist."""
|
||||
async def _addrole(guild: Guild) -> str:
|
||||
"""Generates a configuration menu field value for a guild's addrole whitelist."""
|
||||
|
||||
blacklist = await config.guild(guild).blacklist_roles()
|
||||
whitelist = await config.guild(guild).addrole_roles()
|
||||
if blacklist:
|
||||
blacklist = [guild.get_role(role).mention or error(f"`{role}` (Not Found)") for role in blacklist]
|
||||
blacklist = [guild.get_role(role).mention or error(f"`{role}` (Not Found)") for role in whitelist]
|
||||
blacklist = '\n'.join(blacklist)
|
||||
else:
|
||||
blacklist = warning("No roles are set as blacklist roles!")
|
||||
blacklist = warning("No roles are on the addrole whitelist!")
|
||||
return blacklist
|
||||
|
||||
async def _immune(guild: Guild) -> str:
|
||||
|
|
@ -106,6 +106,18 @@ async def embed(ctx: commands.Context) -> Embed:
|
|||
e.add_field(name="User Overrides", value=await _overrides(ctx.author), inline=False)
|
||||
if ctx.guild is not None and (ctx.author.guild_permissions.administrator or ctx.author.guild_permissions.manage_guild):
|
||||
e.add_field(name="Guild Settings", value=await _guild(ctx.guild), inline=False)
|
||||
e.add_field(name="Blacklist Roles", value=await _blacklist(ctx.guild), inline=False)
|
||||
e.add_field(name="Immune Roles", value=await _immune(ctx.guild), inline=False)
|
||||
return e
|
||||
|
||||
async def addrole_embed(ctx: commands.Context) -> Embed:
|
||||
"""Generates the addrole embed for a guild."""
|
||||
e = await _core(ctx)
|
||||
e.title += ": Addrole Whitelist"
|
||||
e.description = "Use the buttons below to manage the addrole whitelist.\n\n" + await _addrole(ctx.guild)
|
||||
return e
|
||||
|
||||
async def immune_embed(ctx: commands.Context) -> Embed:
|
||||
"""Generates the immune roles embed for a guild."""
|
||||
e = await _core(ctx)
|
||||
e.title += ": Immune Roles"
|
||||
e.description = "Use the buttons below to manage the immune roles.\n\n" + await _immune(ctx.guild)
|
||||
return e
|
||||
|
|
|
|||
Loading…
Reference in a new issue