Create db path if it does not exist

NickM-27 · May 28, 2022 · c1dd25f · c1dd25f
1 parent 7085fd4
commit c1dd25f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/swatch/app.py b/swatch/app.py
@@ -60,6 +60,10 @@ def __init_db__(self):
         """Init the Swatch database."""
         db_file = os.environ.get("DB_FILE", CONST_DB_FILE)
 
+        if not os.path.exists(db_file):
+            logging.debug(f"{db_file} doesn't exist, creating...")
+            os.makedirs(db_file)
+
         swatch_db = SqliteExtDatabase(db_file)
 
         router = Router(swatch_db)

diff --git a/swatch/snapshot.py b/swatch/snapshot.py
@@ -54,7 +54,6 @@ def save_snapshot(
         if not os.path.exists(file_dir):
             logging.debug(f"{file_dir} doesn't exist, creating...")
             os.makedirs(file_dir)
-            return False
 
         file = f"{file_dir}/{file_name}"