update

1 files changed, 9 insertions, 0 deletions

diff --git a/server/db.py b/server/db.py
index e69de29..a73d08f 100644
--- a/server/db.py
+++ b/server/db.py
@@ -0,0 +1,9 @@
+from sqlalchemy import create_engine
+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})
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+Base = declarative_base()