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."""
|
"""Main file for bot."""
|
||||||
|
|
||||||
import os
|
import pathlib
|
||||||
|
|
||||||
import nextcord
|
import nextcord
|
||||||
from nextcord.ext import commands
|
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.nominate_views import NominateView
|
||||||
from dakiny_bot.classes.voting_highcouncil_views import VoteHighcouncilView
|
from dakiny_bot.classes.voting_highcouncil_views import VoteHighcouncilView
|
||||||
from dakiny_bot.classes.voting_views import VoteView
|
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
|
from .DakinySettings import settings
|
||||||
|
|
||||||
@@ -31,19 +31,25 @@ def run() -> 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("dakiny_bot/cogs"):
|
for folder in pathlib.Path("dakiny_bot/cogs").iterdir():
|
||||||
bot.load_extension(f"dakiny_bot.cogs.{folder}")
|
bot.load_extension(f"dakiny_bot.cogs.{folder.name}")
|
||||||
|
|
||||||
@bot.listen()
|
@bot.listen()
|
||||||
async def on_ready() -> None:
|
async def on_ready() -> None:
|
||||||
"""When Discord is connected."""
|
"""When Discord is connected."""
|
||||||
assert bot.user is not None # noqa: S101 documentation does this
|
if bot.user is None:
|
||||||
create_database # noqa: B018
|
dakiny_logger.error("Discord connection failed.")
|
||||||
bot.add_view(NominateView(bot.get_guild(settings.params.guild_id)))
|
return
|
||||||
bot.add_view(NominateHighCouncilView(bot.get_guild(settings.params.guild_id)))
|
|
||||||
bot.add_view(VoteView(bot.get_guild(settings.params.guild_id)))
|
if (guild := bot.get_guild(settings.params.guild_id)) is None:
|
||||||
bot.add_view(VoteHighcouncilView(bot.get_guild(settings.params.guild_id)))
|
dakiny_logger.error("Could not find Dakiny guild object.")
|
||||||
print(f"{bot.user.name} has connected to Discord!") # noqa: T201 TODO make logging to print my stuff to console
|
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
|
# Run Discord bot
|
||||||
bot.run(settings.params.discord_token)
|
bot.run(settings.params.discord_token)
|
||||||
|
|||||||
Reference in New Issue
Block a user