Feat: Remove pylance warnings in app.py.
Also removed old database setup.
This commit is contained in:
+17
-11
@@ -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/<module_name>
|
||||
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)
|
||||
|
||||
Reference in New Issue
Block a user