Fix some smells.

This commit is contained in:
2025-03-17 17:40:07 -04:00
parent dae0d205e5
commit 7fd3fb8458
+5 -4
View File
@@ -1,11 +1,12 @@
"""Main file for bot.""" """Main file for bot."""
import logging import logging
import os import pathlib
import nextcord import nextcord
from nextcord.ext import commands from nextcord.ext import commands
from logger import ocd_logger
from settings import settings from settings import settings
nextcord_logger = logging.getLogger("nextcord") nextcord_logger = logging.getLogger("nextcord")
@@ -33,16 +34,16 @@ def main() -> None:
bot = commands.Bot(intents=intents) bot = commands.Bot(intents=intents)
# Get the modules of all cogs whose directory structure is ./cogs/<module_name> # Get the modules of all cogs whose directory structure is ./cogs/<module_name>
for folder in os.listdir("cogs"): for folder in pathlib.Path("cogs").iterdir():
bot.load_extension(f"cogs.{folder}") bot.load_extension(f"cogs.{folder}")
@bot.listen() @bot.listen()
async def on_ready() -> None: async def on_ready() -> None:
"""When Discord is connected.""" """When Discord is connected."""
if bot.user is None: if bot.user is None:
print("Discord connection failed.") ocd_logger.critical("Discord connection failed.")
return return
print(f"{bot.user.name} has connected to Discord!") ocd_logger.info(f"{bot.user.name} has connected to Discord!")
# Run Discord bot # Run Discord bot
bot.run(settings.params.discord_token) bot.run(settings.params.discord_token)