Add module responsible for loading env-variables

This module tries to load 'optional' env-variables and falls back to pre-set default values if no env-variable exists.

Variables can simply be imported like `from environment import PREFIX`

Loading the prefix was also moved there.

Author: nonchris

src/environment.py

@@ -0,0 +1,25 @@
+import os
+
+
+def load_env(key: str, default: str) -> str:
+    """
+    os.getenv() wrapper that handles the case of None-types for not-set env-variables\n
+
+    :param key: name of env variable to load
+    :param default: default value if variable couldn't be loaded
+    :return: value of env variable or default value
+    """
+    value = os.getenv(key)
+    if value:
+        return value
+    print(f"Can't load env-variable for: '{key}' - falling back to DEFAULT {key}='{default}'")
+    return default
+
+
+TOKEN = os.getenv("TOKEN")  # reading in the token from config.py file
+
+# loading optional env variables
+PREFIX = load_env("PREFIX", "b!")
+VERSION = load_env("VERSION", "unknown")  # version of the bot
+OWNER_NAME = load_env("OWNER_NAME", "unknown")   # owner name with tag e.g. pi#3141
+OWNER_ID = int(load_env("OWNER", "100000000000000000"))  # discord id of the owner

src/main.py

@@ -4,9 +4,7 @@
 import discord
 from discord.ext import commands
 
-intents = discord.Intents.all()
-
-TOKEN = os.getenv("TOKEN")  # reading in the token from config.py file
+from environment import PREFIX, TOKEN
 
 # path for databases or config files
 if not os.path.exists('data/'):
@@ -33,9 +31,8 @@
 logger.addHandler(file_logger)
 logger.addHandler(console_logger)
 
-prefix = "b!"
-
-bot = commands.Bot(command_prefix=prefix, intents=intents)
+intents = discord.Intents.all()
+bot = commands.Bot(command_prefix=PREFIX, intents=intents)
 
 
 # login message