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

Add support of disabled extension for reminder

Browse source
This commit is contained in:
Ethanell 2020-11-05 14:56:18 +01:00
parent 234e8e1877
commit b24aebc906
2 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
administrator/check.py
View file

@ -9,7 +9,7 @@ class ExtensionDisabled(commands.CheckFailure):
def is_enabled(): def is_enabled():
async def check(ctx: commands.Context): async def check(ctx: commands.Context):
if ctx.command.cog: if ctx.command.cog and ctx.guild:
s = db.Session() s = db.Session()
es = s.query(db.ExtensionState).get((ctx.command.cog.qualified_name, ctx.guild.id)) es = s.query(db.ExtensionState).get((ctx.command.cog.qualified_name, ctx.guild.id))
s.close() s.close()

2
extensions/reminder.py
View file

@ -6,6 +6,7 @@ from discord import Embed
from discord.ext.commands import BadArgument from discord.ext.commands import BadArgument
from discord.ext import tasks from discord.ext import tasks
from administrator.check import is_enabled
from administrator.logger import logger from administrator.logger import logger
from administrator import db from administrator import db
from administrator.utils import time_pars, seconds_to_time_string from administrator.utils import time_pars, seconds_to_time_string
@ -22,6 +23,7 @@ class Reminders(commands.Cog, name="Reminder"):
return "Create and manage reminders" return "Create and manage reminders"
@commands.group("reminder", pass_context=True) @commands.group("reminder", pass_context=True)
@is_enabled()
async def reminder(self, ctx: commands.Context): async def reminder(self, ctx: commands.Context):
if ctx.invoked_subcommand is None: if ctx.invoked_subcommand is None:
await ctx.invoke(self.reminder_help) await ctx.invoke(self.reminder_help)