aboutsummaryrefslogtreecommitdiff
path: root/server/db.py
diff options
context:
space:
mode:
Diffstat (limited to 'server/db.py')
-rw-r--r--server/db.py8
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()