Make sure DATA_DIR exists

diff --git a/main.py b/main.py
index e8de3e7b9413112256891521fe133c477376590c..67ca490b4452d5a45ba0cd38f94f8c4ac3b09a8e 100644 (file)
--- a/main.py
+++ b/main.py
@@ -32,6 +32,9 @@ DB_VERSION = 4
 DATA_DIR = Path(os.environ["DATA_DIR"]) if "DATA_DIR" in os.environ else Path(".data")
 SCRIPT_DIR = Path(__file__).parent
 
+# Make sure DATA_DIR exists
+os.makedirs(DATA_DIR, exist_ok=True)
+
 BGCOLOR = "#e8e6b5"
 FGCOLOR = "#695c73"
 ACCOLOR = "#9373a9"