diff --git a/jarvis/client.py b/jarvis/client.py index 04d44bb..01e2cf5 100644 --- a/jarvis/client.py +++ b/jarvis/client.py @@ -53,6 +53,7 @@ class Jarvis(Snake): super().__init__(*args, **kwargs) self.logger = logging.getLogger(__name__) self.phishing_domains = [] + self.pre_run_callback = self._prerun @Task.create(IntervalTrigger(days=1)) async def _update_domains(self) -> None: @@ -72,6 +73,11 @@ class Jarvis(Snake): if update["domain"] in self.phishing_domains: self.phishing_domains.remove(update["domain"]) + async def _prerun(self, ctx: Context, *args, **kwargs) -> None: + name = ctx.invoked_name + args = " ".join(f"{k}:{v}" for k, v in kwargs.items()) + self.logger.debug(f"Running command `{name}` with args: {args or 'None'}") + async def _sync_domains(self) -> None: self.logger.debug("Loading phishing domains") async with ClientSession(headers={"X-Identity": "Discord: zevaryx#5779"}) as session: diff --git a/jarvis/cogs/admin/__init__.py b/jarvis/cogs/admin/__init__.py index daaca31..2022554 100644 --- a/jarvis/cogs/admin/__init__.py +++ b/jarvis/cogs/admin/__init__.py @@ -1,4 +1,6 @@ """J.A.R.V.I.S. Admin Cogs.""" +import logging + from dis_snek import Snake from jarvis.cogs.admin import ban, kick, lock, lockdown, mute, purge, roleping, warning @@ -6,11 +8,21 @@ from jarvis.cogs.admin import ban, kick, lock, lockdown, mute, purge, roleping, def setup(bot: Snake) -> None: """Add admin cogs to J.A.R.V.I.S.""" + logger = logging.getLogger(__name__) + msg = "Loaded jarvis.cogs.admin.{}" ban.BanCog(bot) + logger.debug(msg.format("ban")) kick.KickCog(bot) + logger.debug(msg.format("kick")) lock.LockCog(bot) + logger.debug(msg.format("lock")) lockdown.LockdownCog(bot) + logger.debug(msg.format("ban")) mute.MuteCog(bot) + logger.debug(msg.format("mute")) purge.PurgeCog(bot) + logger.debug(msg.format("purge")) roleping.RolepingCog(bot) + logger.debug(msg.format("roleping")) warning.WarningCog(bot) + logger.debug(msg.format("warning"))