1
0
Fork 0

Switch extension to slash commands

This commit is contained in:
Ethanell 2021-01-06 17:53:30 +01:00
parent d349a398c5
commit eb34ebd947
2 changed files with 49 additions and 46 deletions

View file

@ -6,7 +6,7 @@ import db
from discord.ext import commands from discord.ext import commands
bot = commands.Bot(command_prefix=config.get("prefix"), intents=Intents.all()) bot = commands.Bot(command_prefix=config.get("prefix"), intents=Intents.all())
slash = SlashCommand(bot, auto_register=True) slash = SlashCommand(bot, auto_register=True, auto_delete=True)
import extensions import extensions

View file

@ -2,9 +2,12 @@ from traceback import format_exc
from discord.ext import commands from discord.ext import commands
from discord import Embed, Guild from discord import Embed, Guild
from discord.ext.commands import MissingPermissions, BadArgument, CommandError from discord.ext.commands import BadArgument
from discord_slash import cog_ext, SlashContext, SlashCommandOptionType
from discord_slash.utils import manage_commands
import db import db
from administrator import slash
from administrator.logger import logger from administrator.logger import logger
@ -15,69 +18,66 @@ logger = logger.getChild(extension_name)
class Extension(commands.Cog): class Extension(commands.Cog):
def __init__(self, bot: commands.Bot): def __init__(self, bot: commands.Bot):
self.bot = bot self.bot = bot
slash.get_cog_commands(self)
def description(self): def description(self):
return "Manage bot's extensions" return "Manage bot's extensions"
@commands.group("extension", pass_context=True) @cog_ext.cog_subcommand(base="extension", name="list", description="List all enabled extensions")
async def extension(self, ctx: commands.Context):
if ctx.invoked_subcommand is None:
await ctx.invoke(self.extension_help)
@extension.group("help", pass_context=True)
async def extension_help(self, ctx: commands.Context):
embed = Embed(title="Extension help")
for c, n, v in [[self.extension_list, "extension list", "List all enabled extensions"],
[self.extension_enable, "extension enable", "Enable an extensions"],
[self.extension_disable, "extension disable", "Disable an extensions"],
[self.extension_loaded, "extension loaded", "List all loaded extensions"],
[self.extension_load, "extension load <name>", "Load an extension"],
[self.extension_unload, "extension unload <name>", "Unload an extension"],
[self.extension_reload, "extension reload <name>", "Reload an extension"]]:
try:
if await c.can_run(ctx):
embed.add_field(name=n, value=v, inline=False)
except CommandError:
pass
if not embed.fields:
raise MissingPermissions("")
await ctx.send(embed=embed)
@extension.group("list", pass_context=True)
@commands.has_guild_permissions(administrator=True) @commands.has_guild_permissions(administrator=True)
async def extension_list(self, ctx: commands.Context): async def extension_list(self, ctx: SlashContext):
s = db.Session() s = db.Session()
embed = Embed(title="Extensions list") embed = Embed(title="Extensions list")
for es in s.query(db.ExtensionState).filter(db.ExtensionState.guild_id == ctx.guild.id): for es in s.query(db.ExtensionState).filter(db.ExtensionState.guild_id == ctx.guild.id):
embed.add_field(name=es.extension_name, value="Enable" if es.state else "Disable") embed.add_field(name=es.extension_name, value="Enable" if es.state else "Disable")
await ctx.send(embed=embed) s.close()
await ctx.send(embeds=[embed])
@extension.group("enable", pass_context=True) @cog_ext.cog_subcommand(base="extension",
name="enable",
description="Enable an extensions",
options=[manage_commands.create_option("extension", "The extension to enable",
SlashCommandOptionType.STRING, True)])
@commands.has_guild_permissions(administrator=True) @commands.has_guild_permissions(administrator=True)
async def extension_enable(self, ctx: commands.Context, name: str): async def extension_enable(self, ctx: SlashContext, name: str):
s = db.Session() s = db.Session()
es = s.query(db.ExtensionState).get((name, ctx.guild.id)) es = s.query(db.ExtensionState).get((name, ctx.guild.id))
if not es or es.state: if not es:
raise BadArgument() raise BadArgument()
es.state = True elif es.state:
s.add(es) message = "Extension already enabled"
s.commit() else:
s.close() es.state = True
await ctx.message.add_reaction("\U0001f44d") s.add(es)
s.commit()
s.close()
message = "\U0001f44d"
await ctx.send(content=message)
@extension.group("disable", pass_context=True) @cog_ext.cog_subcommand(base="extension",
name="disable",
description="Disable an extensions",
options=[manage_commands.create_option("extension", "The extension to disable",
SlashCommandOptionType.STRING, True)])
@commands.has_guild_permissions(administrator=True) @commands.has_guild_permissions(administrator=True)
async def extension_disable(self, ctx: commands.Context, name: str): async def extension_disable(self, ctx: SlashContext, name: str):
s = db.Session() s = db.Session()
es = s.query(db.ExtensionState).get((name, ctx.guild.id)) es = s.query(db.ExtensionState).get((name, ctx.guild.id))
if not es or not es.state: if not es:
raise BadArgument() raise BadArgument()
es.state = False elif not es.state:
s.add(es) message = "Extension already disabled"
s.commit() else:
s.close() es.state = False
await ctx.message.add_reaction("\U0001f44d") s.add(es)
s.commit()
s.close()
message = "\U0001f44d"
await ctx.send(content=message)
@commands.group("extension", pass_context=True)
async def extension(self, ctx: commands.Context):
pass
@extension.group("loaded", pass_context=True) @extension.group("loaded", pass_context=True)
@commands.is_owner() @commands.is_owner()
@ -152,6 +152,9 @@ class Extension(commands.Cog):
s.commit() s.commit()
s.close() s.close()
def cog_unload(self):
slash.remove_cog_commands(self)
def setup(bot): def setup(bot):
logger.info(f"Loading...") logger.info(f"Loading...")