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

resolve the url first, so the key into _ENGINE is correct

Browse Source
This commit is contained in:
Masen Furer 2024-12-05 22:13:09 -08:00
parent f3abe475e9
commit b63110e2cb
No known key found for this signature in database
GPG Key ID: B0008AD22B3B3A95
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
reflex/model.py
View File

@ -39,14 +39,15 @@ def get_engine(url: str | None = None) -> sqlalchemy.engine.Engine:
Raises: Raises:
ValueError: If the database url is None. ValueError: If the database url is None.
""" """
global _ENGINE
if url in _ENGINE:
return _ENGINE[url]
conf = get_config() conf = get_config()
url = url or conf.db_url url = url or conf.db_url
if url is None: if url is None:
raise ValueError("No database url configured") raise ValueError("No database url configured")
global _ENGINE
if url in _ENGINE:
return _ENGINE[url]
if not environment.ALEMBIC_CONFIG.get().exists(): if not environment.ALEMBIC_CONFIG.get().exists():
console.warn( console.warn(
"Database is not initialized, run [bold]reflex db init[/bold] first." "Database is not initialized, run [bold]reflex db init[/bold] first."
@ -74,14 +75,15 @@ def get_async_engine(url: str | None) -> sqlalchemy.engine.Engine:
Raises: Raises:
ValueError: If the async database url is None. ValueError: If the async database url is None.
""" """
global _ASYNC_ENGINE
if url in _ASYNC_ENGINE:
return _ASYNC_ENGINE[url]
conf = get_config() conf = get_config()
url = url or conf.async_db_url url = url or conf.async_db_url
if url is None: if url is None:
raise ValueError("No async database url configured") raise ValueError("No async database url configured")
global _ASYNC_ENGINE
if url in _ASYNC_ENGINE:
return _ASYNC_ENGINE[url]
if not environment.ALEMBIC_CONFIG.get().exists(): if not environment.ALEMBIC_CONFIG.get().exists():
console.warn( console.warn(
"Database is not initialized, run [bold]reflex db init[/bold] first." "Database is not initialized, run [bold]reflex db init[/bold] first."