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 sqlalchemy.orm import sessionmaker, declarative_base
|
||||||
from config import settings
|
from config import settings
|
||||||
|
|
||||||
|
connect_args = {"check_same_thread": False} if settings.database_url.startswith("sqlite") else {}
|
||||||
|
|
||||||
engine = create_engine(
|
engine = create_engine(
|
||||||
settings.database_url,
|
settings.database_url,
|
||||||
connect_args={"check_same_thread": False},
|
connect_args=connect_args,
|
||||||
poolclass=NullPool
|
poolclass=NullPool
|
||||||
)
|
)
|
||||||
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
||||||
|
|||||||
Reference in New Issue
Block a user