dirigent-core 0.9.0__py3-none-any.whl

dirigent_core/__init__.py

@@ -0,0 +1,26 @@
+"""Dirigent core: configuration, the baseline schema, and the migrations that create it."""
+
+from importlib.metadata import PackageNotFoundError, version
+
+from dirigent_core.config import Settings, get_settings, reset_settings_cache
+from dirigent_core.database import create_engine, create_session_factory, ping, session_scope
+from dirigent_core.ids import uuid7
+from dirigent_core.models import Base
+
+try:
+    __version__ = version("dirigent-core")
+except PackageNotFoundError:  # pragma: no cover - only when running from a source tree
+    __version__ = "0.0.0"
+
+__all__ = [
+    "Base",
+    "Settings",
+    "__version__",
+    "create_engine",
+    "create_session_factory",
+    "get_settings",
+    "ping",
+    "reset_settings_cache",
+    "session_scope",
+    "uuid7",
+]

dirigent_core/alembic/env.py

@@ -0,0 +1,71 @@
+"""Alembic environment: one migration history that runs on both PostgreSQL and SQLite."""
+
+import asyncio
+from typing import Any
+
+from alembic import context
+from sqlalchemy import Connection
+from sqlalchemy.ext.asyncio import async_engine_from_config
+from sqlalchemy.pool import NullPool
+
+from dirigent_core.config import get_settings
+from dirigent_core.models import Base
+
+config = context.config
+target_metadata = Base.metadata
+
+
+def database_url() -> str:
+    """Resolve the URL to migrate: the Alembic config wins, then the instance settings."""
+    return config.get_main_option("sqlalchemy.url") or get_settings().database_url
+
+
+def run_migrations_offline() -> None:
+    """Emit SQL for the configured URL without connecting to a database."""
+    context.configure(
+        url=database_url(),
+        target_metadata=target_metadata,
+        literal_binds=True,
+        dialect_opts={"paramstyle": "named"},
+        render_as_batch=True,
+        compare_type=True,
+    )
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+def do_run_migrations(connection: Connection) -> None:
+    """Run the migrations on an established synchronous connection."""
+    context.configure(
+        connection=connection,
+        target_metadata=target_metadata,
+        render_as_batch=connection.dialect.name == "sqlite",
+        compare_type=True,
+    )
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+async def run_async_migrations() -> None:
+    """Open the async engine and drive the migrations through it."""
+    section: dict[str, Any] = dict(config.get_section(config.config_ini_section) or {})
+    section["sqlalchemy.url"] = database_url()
+    engine = async_engine_from_config(section, prefix="sqlalchemy.", poolclass=NullPool)
+    async with engine.connect() as connection:
+        await connection.run_sync(do_run_migrations)
+    await engine.dispose()
+
+
+def run_migrations_online() -> None:
+    """Run the migrations against a live database."""
+    connectable = config.attributes.get("connection", None)
+    if isinstance(connectable, Connection):
+        do_run_migrations(connectable)
+        return
+    asyncio.run(run_async_migrations())
+
+
+if context.is_offline_mode():
+    run_migrations_offline()
+else:
+    run_migrations_online()

dirigent_core/alembic/script.py.mako

@@ -0,0 +1,26 @@
+"""${message}.
+
+Revision ID: ${up_revision}
+Revises: ${down_revision | comma,n}
+Create Date: ${create_date}
+"""
+
+from collections.abc import Sequence
+
+import sqlalchemy as sa
+from alembic import op
+${imports if imports else ""}
+revision: str = ${repr(up_revision)}
+down_revision: str | None = ${repr(down_revision)}
+branch_labels: str | Sequence[str] | None = ${repr(branch_labels)}
+depends_on: str | Sequence[str] | None = ${repr(depends_on)}
+
+
+def upgrade() -> None:
+    """Apply the migration."""
+    ${upgrades if upgrades else "pass"}
+
+
+def downgrade() -> None:
+    """Revert the migration."""
+    ${downgrades if downgrades else "pass"}