diff --git a/main.py b/main.py index 00f6978..10b31cc 100644 --- a/main.py +++ b/main.py @@ -1,11 +1,12 @@ """Main file for bot.""" import logging -import os +import pathlib import nextcord from nextcord.ext import commands +from logger import ocd_logger from settings import settings nextcord_logger = logging.getLogger("nextcord") @@ -33,16 +34,16 @@ def main() -> None: bot = commands.Bot(intents=intents) # Get the modules of all cogs whose directory structure is ./cogs/ - for folder in os.listdir("cogs"): + for folder in pathlib.Path("cogs").iterdir(): bot.load_extension(f"cogs.{folder}") @bot.listen() async def on_ready() -> None: """When Discord is connected.""" if bot.user is None: - print("Discord connection failed.") + ocd_logger.critical("Discord connection failed.") return - print(f"{bot.user.name} has connected to Discord!") + ocd_logger.info(f"{bot.user.name} has connected to Discord!") # Run Discord bot bot.run(settings.params.discord_token)