Fix default DB errors

1 files changed, 3 insertions, 0 deletions

diff --git a/database.py b/database.py
index 0166d28..1258e42 100644
--- a/database.py
+++ b/database.py
@@ -1,6 +1,7 @@
 from sqlalchemy import create_engine
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import sessionmaker
+import os
 
 import config
 
@@ -11,6 +12,8 @@ elif config.DB_ENGINE == "postgresql":
     database_url = f"postgresql+psycopg2://{config.DB_USER}:{config.DB_PASSWORD}@{config.DB_HOST}:{config.DB_PORT}/{config.DB_NAME}"
 
 else:
+    if not os.path.exists("data"):
+        os.makedirs("data")
     database_url = "sqlite:///data/data.db"
 
 engine = create_engine(database_url)