evaluate XDG_CONFIG_HOME for CONFIG_DIR

csfore · Jun 23, 2024 · 90dc0ab · 90dc0ab
1 parent 6de3c72
commit 90dc0ab
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/gentodo/config.py b/src/gentodo/config.py
@@ -7,14 +7,16 @@
 import os
 import subprocess
 
-STORAGE_DIR = os.path.expanduser("~/.config/gentodo")
-CONFIG_FILE = os.path.join(STORAGE_DIR, "config.toml")
+CONFIG_DIR = os.path.join(
+    os.environ.get("XDG_CONFIG_HOME", os.path.expanduser("~/.config")),
+    "gentodo")
+CONFIG_FILE = os.path.join(CONFIG_DIR, "config.toml")
 
 class Config:
     '''Class to handle the configuration file settings'''
     def __init__(self):
         if not os.path.isfile(CONFIG_FILE):
-            os.makedirs(STORAGE_DIR)
+            os.makedirs(CONFIG_DIR)
             with open(CONFIG_FILE, "w", encoding="utf_8") as config:
                 config.write("[gentodo]\n")
         self.data = self.load_config()