72 lines
2.1 KiB
Python
72 lines
2.1 KiB
Python
import configparser
|
|
import logging
|
|
from colorlog import ColoredFormatter
|
|
import discord
|
|
import openai
|
|
|
|
|
|
log_level = logging.DEBUG
|
|
log_format = (
|
|
" %(log_color)s%(levelname)-8s%(reset)s | %(log_color)s%(message)s%(reset)s"
|
|
)
|
|
|
|
logging.root.setLevel(log_level)
|
|
formatter = ColoredFormatter(log_format)
|
|
|
|
stream = logging.StreamHandler()
|
|
stream.setLevel(log_level)
|
|
stream.setFormatter(formatter)
|
|
|
|
LOG = logging.getLogger("pythonConfig")
|
|
LOG.setLevel(log_level)
|
|
LOG.addHandler(stream)
|
|
|
|
try:
|
|
with open("config.ini", "r") as f:
|
|
file_contents = f.read()
|
|
except FileNotFoundError:
|
|
config = configparser.ConfigParser()
|
|
config["BOT_INFO"] = {
|
|
"TOKEN": "",
|
|
"BOT_COLOR": "",
|
|
"FEEDBACK_CHANNEL_ID": "",
|
|
"SPOTIFY_CLIENT_ID": "",
|
|
"SPOTIFY_CLIENT_SECRET": "",
|
|
"OPENAI_API_KEY": "",
|
|
"BUG_CHANNEL_ID": "",
|
|
"BOT_INVITE_LINK": ""
|
|
}
|
|
|
|
config["LAVALINK"] = {"HOST": "", "PORT": "", "PASSWORD": ""}
|
|
|
|
with open("config.ini", "w") as configfile:
|
|
config.write(configfile)
|
|
|
|
LOG.error(
|
|
"Configuration file `config.ini` has been generated. Please fill out all of the necessary information. Refer to the docs for information on what a specific configuration option is."
|
|
)
|
|
exit()
|
|
|
|
|
|
config = configparser.ConfigParser()
|
|
config.read_string(file_contents)
|
|
|
|
BOT_TOKEN = config["BOT_INFO"]["TOKEN"]
|
|
BOT_COLOR = discord.Color(int((config["BOT_INFO"]["BOT_COLOR"]).replace("#", ""), 16))
|
|
FEEDBACK_CHANNEL_ID = int(config["BOT_INFO"]["FEEDBACK_CHANNEL_ID"])
|
|
SPOTIFY_CLIENT_ID = config["BOT_INFO"]["SPOTIFY_CLIENT_ID"]
|
|
SPOTIFY_CLIENT_SECRET = config["BOT_INFO"]["SPOTIFY_CLIENT_SECRET"]
|
|
CLIENT = openai.OpenAI(api_key=config["BOT_INFO"]["OPENAI_API_KEY"])
|
|
BUG_CHANNEL_ID = int(config["BOT_INFO"]["BUG_CHANNEL_ID"])
|
|
BOT_INVITE_LINK = config["BOT_INFO"]["BOT_INVITE_LINK"]
|
|
|
|
LAVALINK_HOST = config["LAVALINK"]["HOST"]
|
|
LAVALINK_PORT = config["LAVALINK"]["PORT"]
|
|
LAVALINK_PASSWORD = config["LAVALINK"]["PASSWORD"]
|
|
|
|
|
|
class CheckPlayerError(discord.app_commands.AppCommandError):
|
|
def __init__(self, info) -> None:
|
|
self.info = info
|
|
super().__init__()
|