Seth0330 commited on
Commit
c5cdbac
·
verified ·
1 Parent(s): e3599e6

Create backend/app/db.py

Browse files
Files changed (1) hide show
  1. backend/app/db.py +18 -0
backend/app/db.py ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from sqlalchemy import create_engine
3
+ from sqlalchemy.orm import sessionmaker, declarative_base
4
+
5
+ # SQLite DB path. In Docker / HF we’ll set DB_PATH env, default is local "data/app.db"
6
+ DB_PATH = os.environ.get("DB_PATH", "data/app.db")
7
+
8
+ # Create SQLAlchemy engine
9
+ engine = create_engine(
10
+ f"sqlite:///{DB_PATH}",
11
+ connect_args={"check_same_thread": False},
12
+ )
13
+
14
+ # Session factory
15
+ SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
16
+
17
+ # Base model class
18
+ Base = declarative_base()