parker/guava
parker/guava
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add return for environment variables

Browse Source
This commit is contained in:
Parker M. 2024-06-26 18:48:53 -05:00
parent e276ef9486
commit 0c741e3c2b
No known key found for this signature in database
GPG Key ID: 95CD2E0C7E329F2A
1 changed files with 15 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
code/config.py
View File

@ -46,7 +46,7 @@ def load_config():
# Look for variables in the environment # Look for variables in the environment
if "TOKEN" in os.environ or "BOT_COLOR" in os.environ or "BOT_INVITE_LINK" in os.environ: if "TOKEN" in os.environ or "BOT_COLOR" in os.environ or "BOT_INVITE_LINK" in os.environ:
LOG.info("Detected environment variables. Checking for configuration options.") LOG.info("Detected environment variables. Checking for configuration options.")
validate_env_vars() return validate_env_vars()
else: else:
LOG.info("Detected local environment. Checking for config.ini file.") LOG.info("Detected local environment. Checking for config.ini file.")
@ -172,13 +172,20 @@ def validate_config(file_contents):
BUG_CHANNEL_ID = int(config["BOT_INFO"]["BUG_CHANNEL_ID"]) BUG_CHANNEL_ID = int(config["BOT_INFO"]["BUG_CHANNEL_ID"])
# Assign the rest of the variables # Assign the rest of the variables
TOKEN = config["BOT_INFO"]["TOKEN"] try:
SPOTIFY_CLIENT_ID = config["SPOTIFY"]["SPOTIFY_CLIENT_ID"] TOKEN = config["BOT_INFO"]["TOKEN"]
SPOTIFY_CLIENT_SECRET = config["SPOTIFY"]["SPOTIFY_CLIENT_SECRET"] SPOTIFY_CLIENT_ID = config["SPOTIFY"]["SPOTIFY_CLIENT_ID"]
CLIENT = openai.OpenAI(api_key=config["OPENAI"]["OPENAI_API_KEY"]) SPOTIFY_CLIENT_SECRET = config["SPOTIFY"]["SPOTIFY_CLIENT_SECRET"]
LAVALINK_HOST = config["LAVALINK"]["HOST"] CLIENT = openai.OpenAI(api_key=config["OPENAI"]["OPENAI_API_KEY"])
LAVALINK_PORT = config["LAVALINK"]["PORT"] LAVALINK_HOST = config["LAVALINK"]["HOST"]
LAVALINK_PASSWORD = config["LAVALINK"]["PASSWORD"] LAVALINK_PORT = config["LAVALINK"]["PORT"]
LAVALINK_PASSWORD = config["LAVALINK"]["PASSWORD"]
except KeyError:
sys.exit(
LOG.critical(
"One or more requires environment variables are missing. Please check the docs."
)
)
if errors > 0: if errors > 0:
sys.exit( sys.exit(