laipeng/reflex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

model: only pass "check_same_thread" arg for sqlite database (#1662)

Browse Source
This commit is contained in:
Masen Furer 2023-08-22 14:07:36 -07:00 committed by GitHub
parent 42e9924273
commit 196bdc5101
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
reflex/model.py
View File

@ -43,12 +43,9 @@ def get_engine(url: Optional[str] = None):
)
# Print the SQL queries if the log level is INFO or lower.
echo_db_query = os.environ.get("SQLALCHEMY_ECHO") == "True"
return sqlmodel.create_engine(
url,
echo=echo_db_query,
# Needed for the admin dash.
connect_args={"check_same_thread": False},
)
# Needed for the admin dash on sqlite.
connect_args = {"check_same_thread": False} if url.startswith("sqlite") else {}
return sqlmodel.create_engine(url, echo=echo_db_query, connect_args=connect_args)
class Model(Base, sqlmodel.SQLModel):