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

use async_sessionmaker

Browse Source
This commit is contained in:
Masen Furer 2024-12-09 17:03:07 -08:00
parent 721283a496
commit e04dff8154
No known key found for this signature in database
GPG Key ID: B0008AD22B3B3A95
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
reflex/model.py
View File

@ -25,7 +25,7 @@ from reflex.utils.compat import sqlmodel, sqlmodel_field_has_primary_key
_ENGINE: dict[str, sqlalchemy.engine.Engine] = {} _ENGINE: dict[str, sqlalchemy.engine.Engine] = {}
_ASYNC_ENGINE: dict[str, sqlalchemy.ext.asyncio.AsyncEngine] = {} _ASYNC_ENGINE: dict[str, sqlalchemy.ext.asyncio.AsyncEngine] = {}
_AsyncSessionLocal: dict[str | None, sqlalchemy.orm.sessionmaker] = {} _AsyncSessionLocal: dict[str | None, sqlalchemy.ext.asyncio.async_sessionmaker] = {}
# Import AsyncSession _after_ reflex.utils.compat # Import AsyncSession _after_ reflex.utils.compat
from sqlmodel.ext.asyncio.session import AsyncSession # noqa: E402 from sqlmodel.ext.asyncio.session import AsyncSession # noqa: E402
@ -532,11 +532,11 @@ def asession(url: str | None = None) -> AsyncSession:
""" """
global _AsyncSessionLocal global _AsyncSessionLocal
if url not in _AsyncSessionLocal: if url not in _AsyncSessionLocal:
_AsyncSessionLocal[url] = sqlalchemy.orm.sessionmaker( _AsyncSessionLocal[url] = sqlalchemy.ext.asyncio.async_sessionmaker(
bind=get_async_engine(url),
class_=AsyncSession,
autocommit=False, autocommit=False,
autoflush=False, autoflush=False,
bind=get_async_engine(url), # type: ignore
class_=AsyncSession,
) )
return _AsyncSessionLocal[url]() return _AsyncSessionLocal[url]()