mirror of
https://github.com/Pythagora-io/gpt-pilot.git
synced 2026-01-08 12:53:50 -05:00
5b474ccc1f
This is a complete rewrite of the GPT Pilot core, from the ground up, making the agentic architecture front and center, and also fixing some long-standing problems with the database architecture that weren't feasible to solve without breaking compatibility. As the database structure and config file syntax have changed, we have automatic imports for projects and current configs, see the README.md file for details. This also relicenses the project to FSL-1.1-MIT license.
50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
from os.path import dirname, join
|
|
|
|
from alembic import command
|
|
from alembic.config import Config
|
|
|
|
from core.config import DBConfig
|
|
from core.log import get_logger
|
|
|
|
log = get_logger(__name__)
|
|
|
|
|
|
def _async_to_sync_db_scheme(url: str) -> str:
|
|
"""
|
|
Convert an async database URL to a synchronous one.
|
|
|
|
This is needed because Alembic does not support async database
|
|
connections.
|
|
|
|
:param url: Asynchronouse database URL.
|
|
:return: Synchronous database URL.
|
|
"""
|
|
if url.startswith("postgresql+asyncpg://"):
|
|
return url.replace("postgresql+asyncpg://", "postgresql://")
|
|
elif url.startswith("sqlite+aiosqlite://"):
|
|
return url.replace("sqlite+aiosqlite://", "sqlite://")
|
|
return url
|
|
|
|
|
|
def run_migrations(config: DBConfig):
|
|
"""
|
|
Run database migrations using Alembic.
|
|
|
|
This needs to happen synchronously, before the asyncio
|
|
mainloop is started, and before any database access.
|
|
|
|
:param config: Database configuration.
|
|
"""
|
|
url = _async_to_sync_db_scheme(config.url)
|
|
ini_location = join(dirname(__file__), "alembic.ini")
|
|
|
|
log.debug(f"Running database migrations for {url} (config: {ini_location})")
|
|
|
|
alembic_cfg = Config(ini_location)
|
|
alembic_cfg.set_main_option("sqlalchemy.url", url)
|
|
alembic_cfg.set_main_option("pythagora_runtime", "true")
|
|
command.upgrade(alembic_cfg, "head")
|
|
|
|
|
|
__all__ = ["run_migrations"]
|