From 57e19f3036e491100834426b58994fa23f70e8df Mon Sep 17 00:00:00 2001 From: imcompiling Date: Wed, 16 Jul 2025 10:54:29 -0400 Subject: [PATCH] Feat: Remove pylance warnings in app.py. Also removed old database setup. --- dakiny_bot/app.py | 28 +++++++++++++++++----------- 1 file changed, 17 insertions(+), 11 deletions(-) diff --git a/dakiny_bot/app.py b/dakiny_bot/app.py index d8aa635..706b048 100644 --- a/dakiny_bot/app.py +++ b/dakiny_bot/app.py @@ -1,6 +1,6 @@ """Main file for bot.""" -import os +import pathlib import nextcord from nextcord.ext import commands @@ -9,7 +9,7 @@ from dakiny_bot.classes.nominate_highcouncil_views import NominateHighCouncilVie from dakiny_bot.classes.nominate_views import NominateView from dakiny_bot.classes.voting_highcouncil_views import VoteHighcouncilView from dakiny_bot.classes.voting_views import VoteView -from dakiny_bot.database import create_database +from dakiny_bot.logger import dakiny_logger from .DakinySettings import settings @@ -31,19 +31,25 @@ def run() -> None: bot = commands.Bot(intents=intents) # Get the modules of all cogs whose directory structure is ./cogs/ - for folder in os.listdir("dakiny_bot/cogs"): - bot.load_extension(f"dakiny_bot.cogs.{folder}") + for folder in pathlib.Path("dakiny_bot/cogs").iterdir(): + bot.load_extension(f"dakiny_bot.cogs.{folder.name}") @bot.listen() async def on_ready() -> None: """When Discord is connected.""" - assert bot.user is not None # noqa: S101 documentation does this - create_database # noqa: B018 - bot.add_view(NominateView(bot.get_guild(settings.params.guild_id))) - bot.add_view(NominateHighCouncilView(bot.get_guild(settings.params.guild_id))) - bot.add_view(VoteView(bot.get_guild(settings.params.guild_id))) - bot.add_view(VoteHighcouncilView(bot.get_guild(settings.params.guild_id))) - print(f"{bot.user.name} has connected to Discord!") # noqa: T201 TODO make logging to print my stuff to console + if bot.user is None: + dakiny_logger.error("Discord connection failed.") + return + + if (guild := bot.get_guild(settings.params.guild_id)) is None: + dakiny_logger.error("Could not find Dakiny guild object.") + return + + bot.add_view(NominateView(guild)) + bot.add_view(NominateHighCouncilView(guild)) + bot.add_view(VoteView(guild)) + bot.add_view(VoteHighcouncilView(guild)) + dakiny_logger.info("%s has connected to Discord!", bot.user.name) # Run Discord bot bot.run(settings.params.discord_token)