WAL & retry

1 files changed, 6 insertions, 2 deletions

diff --git a/server/db.py b/server/db.py
index a73d08f..221d240 100644
--- a/server/db.py
+++ b/server/db.py
@@ -1,9 +1,13 @@
-from sqlalchemy import create_engine
+from sqlalchemy import create_engine, text
 from sqlalchemy.orm import sessionmaker, declarative_base
 
 # SQLite DB
 DATABASE_URL = "sqlite:///./virtual_energy.db"
 
-engine = create_engine(DATABASE_URL, connect_args={"check_same_thread": False})
+engine = create_engine(
+    DATABASE_URL, connect_args={"check_same_thread": False, "timeout": 30}
+)
+with engine.connect() as conn:
+    conn.execute(text("PRAGMA journal_mode=WAL;"))
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 Base = declarative_base()