Set log level more directly

stanfordio · Oct 14, 2024 · 236a8ff · 236a8ff
1 parent 4f3860a
commit 236a8ff
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/truthbrush/api.py b/truthbrush/api.py
@@ -12,13 +12,9 @@
 
 load_dotenv()  # take environment variables from .env.
 
-logging.basicConfig(
-    level=(
-        logging.DEBUG
-        if os.getenv("DEBUG") and os.getenv("DEBUG").lower() != "false"
-        else logging.INFO
-    )
-)
+# LOG_LEVEL: CRITICAL, ERROR, WARNING, INFO, DEBUG, NOTSET
+LOG_LEVEL = os.getenv("LOG_LEVEL", default="INFO").upper()
+logging.basicConfig(level=eval("logging.{LOG_LEVEL}")),
 
 BASE_URL = "https://truthsocial.com"
 API_BASE_URL = "https://truthsocial.com/api"