Skip to content

Commit e101f0b

Browse files
Improve handler logic
1 parent 41bd350 commit e101f0b

1 file changed

Lines changed: 19 additions & 17 deletions

File tree

cogs/startup.py

Lines changed: 19 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -40,24 +40,26 @@ async def on_ready(self) -> None:
4040
Shortcut accessors should only be populated once TeX-Bot is ready to make API requests.
4141
"""
4242
if settings["DISCORD_LOG_CHANNEL_WEBHOOK_URL"]:
43-
if isinstance(logger.handlers[1], DiscordHandler):
44-
logger.removeHandler(logger.handlers[1])
45-
46-
discord_logging_handler: DiscordHandler = DiscordHandler( # type: ignore[no-any-unimported]
47-
service_name=self.bot.user.name if self.bot.user else "TeX-Bot",
48-
webhook_url=settings["DISCORD_LOG_CHANNEL_WEBHOOK_URL"],
49-
avatar_url=(
50-
self.bot.user.avatar.url
51-
if self.bot.user and self.bot.user.avatar
52-
else None
53-
),
54-
)
55-
discord_logging_handler.setLevel(logging.WARNING)
56-
discord_logging_handler.setFormatter(
57-
logging.Formatter("{levelname} | {message}", style="{"),
58-
)
43+
for handler in logger.handlers:
44+
if isinstance(handler, DiscordHandler):
45+
logger.removeHandler(handler)
46+
47+
discord_logging_handler: DiscordHandler = DiscordHandler( # type: ignore[no-any-unimported]
48+
service_name=self.bot.user.name if self.bot.user else "TeX-Bot",
49+
webhook_url=settings["DISCORD_LOG_CHANNEL_WEBHOOK_URL"],
50+
avatar_url=(
51+
self.bot.user.avatar.url
52+
if self.bot.user and self.bot.user.avatar
53+
else None
54+
),
55+
)
56+
57+
discord_logging_handler.setLevel(logging.WARNING)
58+
discord_logging_handler.setFormatter(
59+
logging.Formatter("{levelname} | {message}", style="{"),
60+
)
5961

60-
logger.addHandler(discord_logging_handler)
62+
logger.addHandler(discord_logging_handler)
6163
else:
6264
logger.warning(
6365
"DISCORD_LOG_CHANNEL_WEBHOOK_URL was not set, "

0 commit comments

Comments
 (0)