diff --git a/jarvis/cogs/admin/__init__.py b/jarvis/cogs/admin/__init__.py index 6e2a8f8..e524d89 100644 --- a/jarvis/cogs/admin/__init__.py +++ b/jarvis/cogs/admin/__init__.py @@ -3,7 +3,17 @@ import logging from naff import Client -from jarvis.cogs.admin import ban, kick, lock, lockdown, mute, purge, roleping, warning +from jarvis.cogs.admin import ( + ban, + kick, + lock, + lockdown, + modcase, + mute, + purge, + roleping, + warning, +) def setup(bot: Client) -> None: @@ -17,7 +27,9 @@ def setup(bot: Client) -> None: lock.LockCog(bot) logger.debug(msg.format("lock")) lockdown.LockdownCog(bot) - logger.debug(msg.format("ban")) + logger.debug(msg.format("lockdown")) + modcase.CaseCog(bot) + logger.debug(msg.format("modcase")) mute.MuteCog(bot) logger.debug(msg.format("mute")) purge.PurgeCog(bot) diff --git a/jarvis/cogs/admin/modcase.py b/jarvis/cogs/admin/modcase.py index cc1bb05..295aa34 100644 --- a/jarvis/cogs/admin/modcase.py +++ b/jarvis/cogs/admin/modcase.py @@ -3,7 +3,7 @@ from typing import TYPE_CHECKING, Optional from jarvis_core.db import q from jarvis_core.db.models import Modlog, actions -from naff import Client, Cog, InteractionContext, Permissions +from naff import Cog, InteractionContext, Permissions from naff.ext.paginators import Paginator from naff.models.discord.embed import Embed, EmbedField from naff.models.discord.user import Member @@ -165,8 +165,3 @@ class CaseCog(Cog): pages = [await self.get_embed(c, ctx.guild) for c in cases] paginator = Paginator.create_from_embeds(self.bot, *pages, timeout=300) await paginator.send(ctx) - - -def setup(bot: Client) -> None: - """Add CaseCog to JARVIS""" - CaseCog(bot)