From a61f68c683df0e07d8958c6fbd9274410b23f3e1 Mon Sep 17 00:00:00 2001
From: Parker <contact@pkrm.dev>
Date: Sat, 8 Mar 2025 17:19:04 -0600
Subject: Fix default DB errors

---
 database.py | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'database.py')

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)
-- 
cgit v1.2.3-70-g09d2