WIP: Refactor Aurora (3.0.0) #29
5 changed files with 124 additions and 4 deletions
|
@ -15,6 +15,7 @@ from math import ceil
|
||||||
import discord
|
import discord
|
||||||
from dateutil.parser import ParserError, parse
|
from dateutil.parser import ParserError, parse
|
||||||
from discord.ext import tasks
|
from discord.ext import tasks
|
||||||
|
from menus.types import Types
|
||||||
from redbot.core import app_commands, commands, data_manager
|
from redbot.core import app_commands, commands, data_manager
|
||||||
from redbot.core.app_commands import Choice
|
from redbot.core.app_commands import Choice
|
||||||
from redbot.core.bot import Red
|
from redbot.core.bot import Red
|
||||||
|
@ -31,7 +32,7 @@ from .models.change import Change
|
||||||
from .models.moderation import Moderation
|
from .models.moderation import Moderation
|
||||||
from .models.type import type_registry
|
from .models.type import type_registry
|
||||||
from .utilities.config import config, register_config
|
from .utilities.config import config, register_config
|
||||||
from .utilities.factory import addrole_embed, case_factory, changes_factory, evidenceformat_factory, guild_embed, immune_embed, overrides_embed
|
from .utilities.factory import addrole_embed, case_factory, changes_factory, evidenceformat_factory, guild_embed, immune_embed, overrides_embed, type_embed
|
||||||
from .utilities.json import dump
|
from .utilities.json import dump
|
||||||
from .utilities.logger import logger
|
from .utilities.logger import logger
|
||||||
from .utilities.moderate import moderate
|
from .utilities.moderate import moderate
|
||||||
|
@ -44,7 +45,7 @@ class Aurora(commands.Cog):
|
||||||
This cog stores all of its data in an SQLite database."""
|
This cog stores all of its data in an SQLite database."""
|
||||||
|
|
||||||
__author__ = ["SeaswimmerTheFsh"]
|
__author__ = ["SeaswimmerTheFsh"]
|
||||||
__version__ = "2.4.1"
|
__version__ = "2.4.2"
|
||||||
__documentation__ = "https://seacogs.coastalcommits.com/aurora/"
|
__documentation__ = "https://seacogs.coastalcommits.com/aurora/"
|
||||||
|
|
||||||
async def red_delete_data_for_user(self, *, requester, user_id: int):
|
async def red_delete_data_for_user(self, *, requester, user_id: int):
|
||||||
|
@ -1115,6 +1116,21 @@ class Aurora(commands.Cog):
|
||||||
msg = await ctx.send(embed=await guild_embed(ctx))
|
msg = await ctx.send(embed=await guild_embed(ctx))
|
||||||
await msg.edit(view=Guild(ctx, msg, 60))
|
await msg.edit(view=Guild(ctx, msg, 60))
|
||||||
|
|
||||||
|
@aurora_settings.command(name="type")
|
||||||
|
@commands.admin_or_permissions(manage_guild=True)
|
||||||
|
@commands.guild_only()
|
||||||
|
async def aurora_settings_type(self, ctx: commands.Context, moderation_type: str):
|
||||||
|
"""Manage configuration options for specific moderation types.
|
||||||
|
|
||||||
|
See [the documentation](https://seacogs.coastalcommits.com/Aurora/Types) for a list of built-in moderation types."""
|
||||||
|
registered_type = type_registry.get(moderation_type)
|
||||||
|
if not registered_type:
|
||||||
|
types = "`, `".join(type_registry.keys())
|
||||||
|
await ctx.send(error("`moderation_type` is not a valid moderation type.\nValid types are:\n" + types))
|
||||||
|
return
|
||||||
|
msg = await ctx.send(embed=await type_embed(ctx, registered_type))
|
||||||
|
await msg.edit(view=Types(ctx, msg, registered_type))
|
||||||
|
|
||||||
@aurora_settings.command(name="addrole", aliases=["removerole"])
|
@aurora_settings.command(name="addrole", aliases=["removerole"])
|
||||||
@commands.admin_or_permissions(manage_guild=True)
|
@commands.admin_or_permissions(manage_guild=True)
|
||||||
@commands.guild_only()
|
@commands.guild_only()
|
||||||
|
|
58
aurora/menus/types.py
Normal file
58
aurora/menus/types.py
Normal file
|
@ -0,0 +1,58 @@
|
||||||
|
from discord import ButtonStyle, Interaction, Message, ui
|
||||||
|
from redbot.core import commands
|
||||||
|
|
||||||
|
from ..models.type import Type
|
||||||
|
from ..utilities.config import config
|
||||||
|
from ..utilities.factory import type_embed
|
||||||
|
|
||||||
|
|
||||||
|
class Types(ui.View):
|
||||||
|
def __init__(self, ctx: commands.Context, message: Message, moderation_type: Type, timeout: int | None = None):
|
||||||
|
super().__init__()
|
||||||
|
self.ctx = ctx
|
||||||
|
self.message = message
|
||||||
|
self.type = moderation_type
|
||||||
|
self.timeout = timeout
|
||||||
|
|
||||||
|
async def on_timeout(self):
|
||||||
|
await self.message.edit(view=None)
|
||||||
|
|
||||||
|
@ui.button(label="Show in History", style=ButtonStyle.green, row=0)
|
||||||
|
async def show_in_history(self, interaction: Interaction, button: ui.Button): # pylint: disable=unused-argument
|
||||||
|
if not interaction.user.guild_permissions.manage_guild and not interaction.user.guild_permissions.administrator:
|
||||||
|
await interaction.response.send_message("You must have the manage guild permission to change this setting.", ephemeral=True)
|
||||||
|
return
|
||||||
|
await interaction.response.defer()
|
||||||
|
current_setting = await config.custom("type", interaction.guild.id, self.type.key).show_in_history()
|
||||||
|
await config.custom("type", interaction.guild.id, self.type.key).show_in_history.set(not current_setting)
|
||||||
|
await interaction.message.edit(embed=await type_embed(self.ctx, self.type))
|
||||||
|
|
||||||
|
@ui.button(label="Show Moderator", style=ButtonStyle.green, row=0)
|
||||||
|
async def show_moderator(self, interaction: Interaction, button: ui.Button): # pylint: disable=unused-argument
|
||||||
|
if not interaction.user.guild_permissions.manage_guild and not interaction.user.guild_permissions.administrator:
|
||||||
|
await interaction.response.send_message("You must have the manage guild permission to change this setting.", ephemeral=True)
|
||||||
|
return
|
||||||
|
await interaction.response.defer()
|
||||||
|
current_setting = await config.custom("type", interaction.guild.id, self.type.key).show_moderator()
|
||||||
|
await config.custom("type", interaction.guild.id, self.type.key).show_moderator.set(not current_setting)
|
||||||
|
await interaction.message.edit(embed=await type_embed(self.ctx, self.type))
|
||||||
|
|
||||||
|
@ui.button(label="Use Discord Permissions", style=ButtonStyle.green, row=0)
|
||||||
|
async def use_discord_permissions(self, interaction: Interaction, button: ui.Button): # pylint: disable=unused-argument
|
||||||
|
if not interaction.user.guild_permissions.manage_guild and not interaction.user.guild_permissions.administrator:
|
||||||
|
await interaction.response.send_message("You must have the manage guild permission to change this setting.", ephemeral=True)
|
||||||
|
return
|
||||||
|
await interaction.response.defer()
|
||||||
|
current_setting = await config.custom("type", interaction.guild.id, self.type.key).use_discord_permissions()
|
||||||
|
await config.custom("type", interaction.guild.id, self.type.key).use_discord_permissions.set(not current_setting)
|
||||||
|
await interaction.message.edit(embed=await type_embed(self.ctx, self.type))
|
||||||
|
|
||||||
|
@ui.button(label="DM Users", style=ButtonStyle.green, row=0)
|
||||||
|
async def dm_users(self, interaction: Interaction, button: ui.Button): # pylint: disable=unused-argument
|
||||||
|
if not interaction.user.guild_permissions.manage_guild and not interaction.user.guild_permissions.administrator:
|
||||||
|
await interaction.response.send_message("You must have the manage guild permission to change this setting.", ephemeral=True)
|
||||||
|
return
|
||||||
|
await interaction.response.defer()
|
||||||
|
current_setting = await config.custom("type", interaction.guild.id, self.type.key).dm_users()
|
||||||
|
await config.custom("type", interaction.guild.id, self.type.key).dm_users.set(not current_setting)
|
||||||
|
await interaction.message.edit(embed=await type_embed(self.ctx, self.type))
|
|
@ -1,12 +1,12 @@
|
||||||
from abc import abstractmethod
|
from abc import abstractmethod
|
||||||
from typing import Any, List, Tuple
|
from typing import Any, Dict, Tuple
|
||||||
|
|
||||||
from class_registry import AutoRegister, ClassRegistry
|
from class_registry import AutoRegister, ClassRegistry
|
||||||
from discord import Interaction, Member, User
|
from discord import Interaction, Member, User
|
||||||
from discord.abc import Messageable
|
from discord.abc import Messageable
|
||||||
from redbot.core import commands
|
from redbot.core import commands
|
||||||
|
|
||||||
type_registry: List['Type'] = ClassRegistry(attr_name='key', unique=True)
|
type_registry: Dict['str', 'Type'] = ClassRegistry(attr_name='key', unique=True)
|
||||||
|
|
||||||
class Type(metaclass=AutoRegister(type_registry)):
|
class Type(metaclass=AutoRegister(type_registry)):
|
||||||
"""This is a base class for moderation types.
|
"""This is a base class for moderation types.
|
||||||
|
|
|
@ -27,3 +27,13 @@ def register_config(config_obj: Config):
|
||||||
history_inline_pagesize=None,
|
history_inline_pagesize=None,
|
||||||
auto_evidenceformat=None,
|
auto_evidenceformat=None,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
moderation_type = {
|
||||||
|
"show_in_history": bool,
|
||||||
|
"show_moderator": bool,
|
||||||
|
"use_discord_permissions": bool,
|
||||||
|
"dm_users": bool,
|
||||||
|
}
|
||||||
|
|
||||||
|
config_obj.init_custom("type", 2)
|
||||||
|
config_obj.register_custom("type", **moderation_type)
|
||||||
|
|
|
@ -544,3 +544,39 @@ async def immune_embed(ctx: commands.Context) -> Embed:
|
||||||
e.description += "\n\n" + immune_str
|
e.description += "\n\n" + immune_str
|
||||||
|
|
||||||
return e
|
return e
|
||||||
|
|
||||||
|
async def type_embed(ctx: commands.Context, moderation_type = Type) -> Embed:
|
||||||
|
"""Generates a configuration menu field value for a guild's settings."""
|
||||||
|
|
||||||
|
type_settings = {
|
||||||
|
"show_in_history": await config.custom("type", ctx.guild.id, moderation_type.key).show_in_history(),
|
||||||
|
"show_moderator": await config.custom("type", ctx.guild.id, moderation_type.key).show_moderator(),
|
||||||
|
"use_discord_permissions": await config.custom("type", ctx.guild.id, moderation_type.key).use_discord_permissions(),
|
||||||
|
"dm_users": await config.custom("type", ctx.guild.id, moderation_type.key).dm_users(),
|
||||||
|
}
|
||||||
|
|
||||||
|
guild_str = [
|
||||||
|
"- "
|
||||||
|
+ bold("Show in History: ")
|
||||||
|
+ get_bool_emoji(type_settings["show_in_history"]),
|
||||||
|
"- "
|
||||||
|
+ bold("Show Moderator: ")
|
||||||
|
+ get_bool_emoji(type_settings["show_moderator"]),
|
||||||
|
"- "
|
||||||
|
+ bold("Use Discord Permissions: ")
|
||||||
|
+ get_bool_emoji(type_settings["use_discord_permissions"]),
|
||||||
|
"- "
|
||||||
|
+ bold("DM Users: ")
|
||||||
|
+ get_bool_emoji(type_settings["dm_users"]),
|
||||||
|
]
|
||||||
|
guild_str = "\n".join(guild_str)
|
||||||
|
|
||||||
|
e = await _config(ctx)
|
||||||
|
e.title += ": Server Configuration"
|
||||||
|
e.description = (
|
||||||
|
"""
|
||||||
|
Use the buttons below to manage Aurora's server configuration.\n
|
||||||
|
"""
|
||||||
|
+ guild_str
|
||||||
|
)
|
||||||
|
return e
|
||||||
|
|
Loading…
Reference in a new issue