Cogs management
- Cogs file renamed (easier managing) - Cogs commands in admin cog: cog load/unload/reload/list
This commit is contained in:
parent
75f8e041e8
commit
3184035346
|
@ -0,0 +1,116 @@
|
|||
import discord
|
||||
from discord import app_commands
|
||||
from discord.app_commands import Choice
|
||||
from discord.ext import commands
|
||||
|
||||
from classes.my_cog import MyCog
|
||||
|
||||
|
||||
class CogGroup(app_commands.Group, name="cog", description="Gérer les cogs"):
|
||||
def __init__(self, bot: commands.Bot):
|
||||
super().__init__()
|
||||
self.bot = bot
|
||||
|
||||
@app_commands.command(name="load")
|
||||
async def cog_load(self, interaction: discord.Interaction, extension: str):
|
||||
"""
|
||||
Charge une extension
|
||||
:param interaction: discord interaction
|
||||
:param extension: Nom de l'extension
|
||||
"""
|
||||
try:
|
||||
await self.bot.load_extension(f"cogs.{extension.lower()}")
|
||||
await interaction.response.send_message(f"Extension `{extension}` chargée !")
|
||||
|
||||
except commands.ExtensionAlreadyLoaded:
|
||||
await interaction.response.send_message(f"Extension `{extension}` déjà chargée...")
|
||||
|
||||
except commands.ExtensionNotFound:
|
||||
await interaction.response.send_message(f"Extension `{extension}` non trouvée...")
|
||||
|
||||
except Exception as e:
|
||||
print(e)
|
||||
await interaction.response.send_message(f"Une erreur inconnue est survenue:\n`{e}`")
|
||||
|
||||
@app_commands.command(name="unload")
|
||||
async def cog_unload(self, interaction: discord.Interaction, extension: str):
|
||||
"""
|
||||
Décharge une extension
|
||||
:param interaction: discord interaction
|
||||
:param extension: Nom de l'extension
|
||||
:return:
|
||||
"""
|
||||
try:
|
||||
await self.bot.unload_extension(f"cogs.{extension.lower()}")
|
||||
await interaction.response.send_message(f"Extension `{extension}` déchargée !")
|
||||
|
||||
except commands.ExtensionNotLoaded:
|
||||
await interaction.response.send_message(f"Extension `{extension}` non chargée...")
|
||||
|
||||
except commands.ExtensionNotFound:
|
||||
await interaction.response.send_message(f"Extension `{extension}` non trouvée...")
|
||||
|
||||
except Exception as e:
|
||||
print(e)
|
||||
await interaction.response.send_message(f"Une erreur inconnue est survenue:\n`{e}`")
|
||||
|
||||
@app_commands.command(name="reload")
|
||||
async def cog_reload(self, interaction: discord.Interaction, extension: str):
|
||||
"""
|
||||
Recharge une extension
|
||||
:param interaction: discord interaction
|
||||
:param extension: Nom de l'extension
|
||||
"""
|
||||
|
||||
async def fun():
|
||||
await self.bot.load_extension(f"cogs.{extension.lower()}")
|
||||
await interaction.response.send_message(f"Extension `{extension}` rechargée !")
|
||||
|
||||
try:
|
||||
await self.bot.unload_extension(f"cogs.{extension.lower()}")
|
||||
await fun()
|
||||
|
||||
except (commands.ExtensionAlreadyLoaded, commands.ExtensionNotLoaded):
|
||||
await fun()
|
||||
|
||||
except commands.ExtensionNotFound:
|
||||
await interaction.response.send_message(f"Extension `{extension}` non trouvée...")
|
||||
|
||||
except Exception as e:
|
||||
print(e)
|
||||
await interaction.response.send_message(f"Une erreur inconnue est survenue:\n`{e}`")
|
||||
|
||||
@app_commands.command(name="list")
|
||||
async def cog_list(self, interaction: discord.Interaction):
|
||||
"""
|
||||
Voir la liste des extensions
|
||||
:param interaction: discord interaction
|
||||
"""
|
||||
enabled = [str(ext) for ext in self.bot.extensions]
|
||||
total_list = self.bot.initial_extensions
|
||||
disabled = list(set(total_list).difference(enabled))
|
||||
|
||||
msg = "✅ **Enabled**\n"
|
||||
for ext in enabled:
|
||||
msg += f"* `{ext[5:]}`\n"
|
||||
msg += "\n❌ **Disabled**\n"
|
||||
for ext in disabled:
|
||||
msg += f"* `{ext[5:]}`\n"
|
||||
|
||||
embed = discord.Embed(title="Liste des extensions",
|
||||
description=msg,
|
||||
color=discord.Color.blurple())
|
||||
await interaction.response.send_message(embed=embed)
|
||||
|
||||
|
||||
class Admin(MyCog):
|
||||
def __init__(self, bot):
|
||||
super().__init__(bot)
|
||||
self.bot = bot
|
||||
|
||||
# Add cogs commands group to cog
|
||||
self.bot.tree.add_command(CogGroup(self.bot))
|
||||
|
||||
|
||||
async def setup(bot):
|
||||
await bot.add_cog(Admin(bot))
|
|
@ -1,16 +0,0 @@
|
|||
import discord
|
||||
from discord import app_commands
|
||||
from discord.app_commands import Choice
|
||||
from discord.ext import commands
|
||||
|
||||
from classes.my_cog import MyCog
|
||||
|
||||
|
||||
class Admin(MyCog):
|
||||
def __init__(self, bot):
|
||||
super().__init__(bot)
|
||||
self.bot = bot
|
||||
|
||||
|
||||
async def setup(bot):
|
||||
await bot.add_cog(Admin(bot))
|
Loading…
Reference in New Issue