fix: support both SQLite and PostgreSQL without check_same_thread crash
This commit is contained in:
+3
-1
@@ -3,9 +3,11 @@ from sqlalchemy.pool import NullPool
|
||||
from sqlalchemy.orm import sessionmaker, declarative_base
|
||||
from config import settings
|
||||
|
||||
connect_args = {"check_same_thread": False} if settings.database_url.startswith("sqlite") else {}
|
||||
|
||||
engine = create_engine(
|
||||
settings.database_url,
|
||||
connect_args={"check_same_thread": False},
|
||||
connect_args=connect_args,
|
||||
poolclass=NullPool
|
||||
)
|
||||
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
||||
|
||||
Reference in New Issue
Block a user