mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
Paul Hernandez
e6496df595
feat: Add multiple projects support feat: enhanced read_note for when initial result is not found fix: merge frontmatter when updating note fix: handle directory removed on sync watch
25 lines
718 B
Python
25 lines
718 B
Python
"""Functions for managing database migrations."""
|
|
|
|
from pathlib import Path
|
|
from loguru import logger
|
|
from alembic.config import Config
|
|
from alembic import command
|
|
|
|
|
|
def get_alembic_config() -> Config: # pragma: no cover
|
|
"""Get alembic config with correct paths."""
|
|
migrations_path = Path(__file__).parent
|
|
alembic_ini = migrations_path / "alembic.ini"
|
|
|
|
config = Config(alembic_ini)
|
|
config.set_main_option("script_location", str(migrations_path))
|
|
return config
|
|
|
|
|
|
def reset_database(): # pragma: no cover
|
|
"""Drop and recreate all tables."""
|
|
logger.info("Resetting database...")
|
|
config = get_alembic_config()
|
|
command.downgrade(config, "base")
|
|
command.upgrade(config, "head")
|