q792602257andClaude Opus 4.6
52e442790a
Tests / Test passed: 819
Adds cloud/skills.py (domain + service), SQLAlchemy models and Alembic migration 0010_skill_management (cloud_skills, cloud_skill_entitlements, cloud_skill_sync_state, a per-host changelog, and host_skill_inventory), and repository methods with a monotonic per-host entitlement_version that drives correct incremental fetch_host_delta. cloud-api suite green (41 passed); HEAD_REVISION bumped to 0010. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
from alembic import command
|
|
from alembic.config import Config
|
|
from alembic.runtime.migration import MigrationContext
|
|
|
|
from cloud.database import create_database_engine, normalize_database_url
|
|
|
|
|
|
HEAD_REVISION = "0010_skill_management"
|
|
|
|
|
|
class SchemaVersionError(RuntimeError):
|
|
"""Raised when the database schema is not at the required revision."""
|
|
|
|
|
|
def upgrade_database(database_url: str, revision: str = "head") -> None:
|
|
command.upgrade(_alembic_config(database_url), revision)
|
|
|
|
|
|
def downgrade_database(database_url: str, revision: str = "base") -> None:
|
|
command.downgrade(_alembic_config(database_url), revision)
|
|
|
|
|
|
def current_revision(database_url: str) -> str | None:
|
|
engine = create_database_engine(database_url)
|
|
try:
|
|
with engine.connect() as connection:
|
|
context = MigrationContext.configure(connection)
|
|
return context.get_current_revision()
|
|
finally:
|
|
engine.dispose()
|
|
|
|
|
|
def is_schema_current(database_url: str) -> bool:
|
|
return current_revision(database_url) == HEAD_REVISION
|
|
|
|
|
|
def require_current_schema(database_url: str) -> None:
|
|
revision = current_revision(database_url)
|
|
if revision != HEAD_REVISION:
|
|
raise SchemaVersionError(
|
|
f"cloud database schema is {revision or 'unversioned'}; "
|
|
f"required revision is {HEAD_REVISION}"
|
|
)
|
|
|
|
|
|
def _alembic_config(database_url: str) -> Config:
|
|
migrations_dir = Path(__file__).resolve().parent / "migrations"
|
|
config = Config(str(migrations_dir / "alembic.ini"))
|
|
config.set_main_option("script_location", str(migrations_dir))
|
|
config.set_main_option(
|
|
"sqlalchemy.url",
|
|
normalize_database_url(database_url).replace("%", "%%"),
|
|
)
|
|
return config
|