Connect to db on bot start. Remove message map.

This commit is contained in:
alwayscompiling
2022-11-26 21:36:38 -05:00
parent 77a32b3adc
commit 12a55dd626
2 changed files with 16 additions and 12 deletions
+9 -1
View File
@@ -1,3 +1,7 @@
"""
Loads environment variables for use in bot.
"""
import os import os
from dotenv.main import load_dotenv from dotenv.main import load_dotenv
@@ -5,5 +9,9 @@ from dotenv.main import load_dotenv
load_dotenv() load_dotenv()
# Discord config # Discord config
DISCORD_TOKEN = os.getenv("DISCORD_TOKEN", "")
BOT_PREFIX = "?" BOT_PREFIX = "?"
DISCORD_TOKEN = os.getenv("DISCORD_TOKEN", "")
SQL_DB = os.getenv("SQL_DB", "")
SQL_HOST = os.getenv("SQL_HOST", "")
SQL_PASS = os.getenv("SQL_PASS", "")
SQL_USER = os.getenv("SQL_USER", "")
+7 -11
View File
@@ -1,16 +1,16 @@
""" """
main file for pvm-highscores bot main file for highscores bot
""" """
import logging import logging
import os import os
import sys import sys
from nextcord.ext import commands import mysql.connector
import nextcord import nextcord
from nextcord.ext import commands
import config import config
from classes.highscores_data import HighscoresData
from classes.submission import SubmissionMap from classes.submission import SubmissionMap
logger = logging.getLogger("nextcord") logger = logging.getLogger("nextcord")
@@ -19,14 +19,10 @@ handler = logging.FileHandler(filename="nextcord.log", encoding="utf-8", mode="w
handler.setFormatter(logging.Formatter("%(asctime)s:%(levelname)s:%(name)s: %(message)s")) handler.setFormatter(logging.Formatter("%(asctime)s:%(levelname)s:%(name)s: %(message)s"))
logger.addHandler(handler) logger.addHandler(handler)
highscores_db = mysql.connector.connect(
highscores_data: HighscoresData = HighscoresData() host=config.SQL_HOST, user=config.SQL_USER, password=config.SQL_PASS, database=config.SQL_DB
highscores_message_map = {} )
highscores_cursor = highscores_db.cursor()
for guild in highscores_data.guilds:
highscores_message_map[guild.guild_id] = {}
for table in guild.tables:
highscores_message_map[guild.guild_id][table.message_id] = table.name
submission_objects: SubmissionMap = SubmissionMap() submission_objects: SubmissionMap = SubmissionMap()