flifloo/Administrator-py
Archived
1
0
Fork 0
Code Activity

Update check and send error with traceback on fail

Browse source
This commit is contained in:
Ethanell 2020-11-05 00:53:12 +01:00
parent 1721dfbd69
commit d740869609
2 changed files with 9 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
administrator/check.py
View file

@ -1,10 +0,0 @@
from discord.ext import commands
from administrator import config
class NotOwner(commands.CheckFailure):
pass
async def is_owner(ctx: commands.Context):
return ctx.author.id == config.get("admin_id")

14
extensions/extension.py
View file

@ -1,6 +1,7 @@
from traceback import format_exc
from discord.ext import commands from discord.ext import commands
from discord import Embed from discord import Embed
from administrator.check import is_owner
from administrator.logger import logger from administrator.logger import logger
@ -16,7 +17,7 @@ class Extension(commands.Cog):
return "Manage bot's extensions" return "Manage bot's extensions"
@commands.group("extension", pass_context=True) @commands.group("extension", pass_context=True)
@commands.check(is_owner) @commands.is_owner()
async def extension(self, ctx: commands.Context): async def extension(self, ctx: commands.Context):
if ctx.invoked_subcommand is None: if ctx.invoked_subcommand is None:
embed = Embed(title="Extensions") embed = Embed(title="Extensions")
@ -25,33 +26,36 @@ class Extension(commands.Cog):
await ctx.send(embed=embed) await ctx.send(embed=embed)
@extension.group("load", pass_context=True) @extension.group("load", pass_context=True)
@commands.check(is_owner) @commands.is_owner()
async def extension_load(self, ctx: commands.Context, name: str): async def extension_load(self, ctx: commands.Context, name: str):
try: try:
self.bot.load_extension(name) self.bot.load_extension(name)
except Exception as e: except Exception as e:
await ctx.message.add_reaction("\u26a0") await ctx.message.add_reaction("\u26a0")
await ctx.send(f"{e.__class__.__name__}: {e}\n```{format_exc()}```")
else: else:
await ctx.message.add_reaction("\U0001f44d") await ctx.message.add_reaction("\U0001f44d")
@extension.group("unload", pass_context=True) @extension.group("unload", pass_context=True)
@commands.check(is_owner) @commands.is_owner()
async def extension_unload(self, ctx: commands.Context, name: str): async def extension_unload(self, ctx: commands.Context, name: str):
try: try:
self.bot.unload_extension(name) self.bot.unload_extension(name)
except Exception as e: except Exception as e:
await ctx.message.add_reaction("\u26a0") await ctx.message.add_reaction("\u26a0")
await ctx.send(f"{e.__class__.__name__}: {e}\n```{format_exc()}```")
else: else:
await ctx.message.add_reaction("\U0001f44d") await ctx.message.add_reaction("\U0001f44d")
@extension.group("reload", pass_context=True) @extension.group("reload", pass_context=True)
@commands.check(is_owner) @commands.is_owner()
async def extension_reload(self, ctx: commands.Context, name: str): async def extension_reload(self, ctx: commands.Context, name: str):
try: try:
self.bot.unload_extension(name) self.bot.unload_extension(name)
self.bot.load_extension(name) self.bot.load_extension(name)
except Exception as e: except Exception as e:
await ctx.message.add_reaction("\u26a0") await ctx.message.add_reaction("\u26a0")
await ctx.send(f"{e.__class__.__name__}: {e}\n```{format_exc()}```")
else: else:
await ctx.message.add_reaction("\U0001f44d") await ctx.message.add_reaction("\U0001f44d")