1 files changed, 59 insertions, 0 deletions
diff --git a/var.py b/var.py
new file mode 100644
index 0000000..e444629
--- /dev/null
+++ b/var.py
@@ -0,0 +1,59 @@
+import configparser
+import logging
+import os
+from colorlog import ColoredFormatter
+
+
+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)
+
+
+# If the app is running in Docker
+if "BASE_URL" in os.environ or "IP_TO_LOCATION" in os.environ:
+    BASE_URL = os.environ["BASE_URL"]
+    IP_TO_LOCATION = (
+        True if os.environ["IP_TO_LOCATION"].upper() in ["TRUE", "T"] else False
+    )
+    if IP_TO_LOCATION:
+        API_KEY = os.environ["API_KEY"]
+    else:
+        API_KEY = None
+
+# Otherwise, the app is running on bare metal
+try:
+    with open("config.ini", "r") as f:
+        config = configparser.ConfigParser()
+        config.read_string(f.read())
+
+        BASE_URL = config["CONFIG"]["BASE_URL"]
+        IP_TO_LOCATION = (
+            True
+            if config["CONFIG"]["IP_TO_LOCATION"].upper() in ["TRUE", "T"]
+            else False
+        )
+        if IP_TO_LOCATION:
+            API_KEY = config["CONFIG"]["API_KEY"]
+        else:
+            API_KEY = None
+except FileNotFoundError:
+    config = configparser.ConfigParser()
+    config["CONFIG"] = {"BASE_URL": "", "IP_TO_LOCATION": "", "API_KEY": ""}
+
+    with open("config.ini", "w") as configfile:
+        config.write(configfile)
+
+    LOG.error(
+        "`config.ini` has been created. Fill out the necessary information then re-run."
+    )
+    exit()