etlantic-sqlmodel 0.45.0__tar.gz → 0.47.0__tar.gz

{etlantic_sqlmodel-0.45.0 → etlantic_sqlmodel-0.47.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: etlantic-sqlmodel
-Version: 0.45.0
+Version: 0.47.0
 Summary: ContractModel ↔ SQLModel bridge for ETLantic.
 Project-URL: Homepage, https://github.com/eddiethedean/etlantic
 Project-URL: Documentation, https://github.com/eddiethedean/etlantic/tree/main/docs
@@ -18,7 +18,7 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Typing :: Typed
 Requires-Python: >=3.11
-Requires-Dist: etlantic<0.46,>=0.45.0
+Requires-Dist: etlantic<0.48,>=0.47.0
 Requires-Dist: sqlmodel<1,>=0.0.22
 Description-Content-Type: text/markdown
 
@@ -28,13 +28,13 @@ Optional bridge between ETLantic `Data` contracts and
 [SQLModel](https://sqlmodel.tiangolo.com/) table models, plus optional CP1
 control-plane reference stores. Install when you need `contract_to_sqlmodel`
 helpers or SQLModel-backed definition/submission stores for local CP1 demos.
-Package version is **0.45.0** — pin with core.
+Package version is **0.47.0** — pin with core.
 
 ## Install
 
 ```bash
-pip install 'etlantic-sqlmodel==0.45.0'
-# pip install 'etlantic==0.45.0'
+pip install 'etlantic-sqlmodel==0.47.0'
+# pip install 'etlantic==0.47.0'
 ```
 
 ## Schema bridge
@@ -96,6 +96,6 @@ for isolation queries. Use separate engines/schemas per tenant for
 
 ## Links
 
-[Optional packages](https://etlantic.readthedocs.io/en/v0.45.0/10_REFERENCE/OPTIONAL_PACKAGES/) ·
+[Optional packages](https://etlantic.readthedocs.io/en/v0.47.0/10_REFERENCE/OPTIONAL_PACKAGES/) ·
 [Source](https://github.com/eddiethedean/etlantic/tree/main/packages/etlantic-sqlmodel) ·
 [Issues](https://github.com/eddiethedean/etlantic/issues)

{etlantic_sqlmodel-0.45.0 → etlantic_sqlmodel-0.47.0}/README.md

@@ -4,13 +4,13 @@ Optional bridge between ETLantic `Data` contracts and
 [SQLModel](https://sqlmodel.tiangolo.com/) table models, plus optional CP1
 control-plane reference stores. Install when you need `contract_to_sqlmodel`
 helpers or SQLModel-backed definition/submission stores for local CP1 demos.
-Package version is **0.45.0** — pin with core.
+Package version is **0.47.0** — pin with core.
 
 ## Install
 
 ```bash
-pip install 'etlantic-sqlmodel==0.45.0'
-# pip install 'etlantic==0.45.0'
+pip install 'etlantic-sqlmodel==0.47.0'
+# pip install 'etlantic==0.47.0'
 ```
 
 ## Schema bridge
@@ -72,6 +72,6 @@ for isolation queries. Use separate engines/schemas per tenant for
 
 ## Links
 
-[Optional packages](https://etlantic.readthedocs.io/en/v0.45.0/10_REFERENCE/OPTIONAL_PACKAGES/) ·
+[Optional packages](https://etlantic.readthedocs.io/en/v0.47.0/10_REFERENCE/OPTIONAL_PACKAGES/) ·
 [Source](https://github.com/eddiethedean/etlantic/tree/main/packages/etlantic-sqlmodel) ·
 [Issues](https://github.com/eddiethedean/etlantic/issues)

{etlantic_sqlmodel-0.45.0 → etlantic_sqlmodel-0.47.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "etlantic-sqlmodel"
-version = "0.45.0"
+version = "0.47.0"
 description = "ContractModel ↔ SQLModel bridge for ETLantic."
 readme = "README.md"
 license = "MIT"
@@ -17,7 +17,7 @@ classifiers = [
     "Typing :: Typed",
 ]
 dependencies = [
-    "etlantic>=0.45.0,<0.46",
+    "etlantic>=0.47.0,<0.48",
     "sqlmodel>=0.0.22,<1",
 ]
 

{etlantic_sqlmodel-0.45.0 → etlantic_sqlmodel-0.47.0}/src/etlantic_sqlmodel/__init__.py

@@ -52,7 +52,7 @@ from etlantic_sqlmodel.migrations import (
 )
 from sqlmodel import Field, SQLModel
 
-__version__ = "0.45.0"
+__version__ = "0.47.0"
 
 __all__ = [
     "AliasRow",

{etlantic_sqlmodel-0.45.0 → etlantic_sqlmodel-0.47.0}/src/etlantic_sqlmodel/control_plane/__init__.py

@@ -37,6 +37,7 @@ from etlantic_sqlmodel.control_plane.models import (
     LogicalIdentityRow,
     PromotionRow,
     RevisionRow,
+    ScheduleSnapshotRow,
     SecurityDomainRow,
     SubmissionRow,
     TenantRow,
@@ -60,6 +61,11 @@ from etlantic_sqlmodel.control_plane.registry_stores import (
     SqlModelWorkspaceDirectory,
     create_registry_tables,
 )
+from etlantic_sqlmodel.control_plane.schedule_stores import (
+    SCHEDULE_TABLES,
+    SQLModelScheduleStore,
+    create_schedule_tables,
+)
 from etlantic_sqlmodel.control_plane.session import (
     create_sqlite_engine,
     make_session_factory,
@@ -78,6 +84,7 @@ __all__ = [
     "CP4_TABLES",
     "DURABLE_TABLES",
     "REGISTRY_TABLES",
+    "SCHEDULE_TABLES",
     "AliasRow",
     "BackupTranscript",
     "Cp4GovernanceSnapshotRow",
@@ -99,7 +106,9 @@ __all__ = [
     "SQLModelObjectiveStore",
     "SQLModelPolicyProvider",
     "SQLModelQuotaProvider",
+    "SQLModelScheduleStore",
     "SQLModelSubmissionStore",
+    "ScheduleSnapshotRow",
     "SecurityDomainRow",
     "SqlModelEventStore",
     "SqlModelRegistryProvider",
@@ -115,6 +124,7 @@ __all__ = [
     "create_cp4_tables",
     "create_durable_tables",
     "create_registry_tables",
+    "create_schedule_tables",
     "create_sqlite_engine",
     "dump_registry_sqlite",
     "load_registry_sqlite",

{etlantic_sqlmodel-0.45.0 → etlantic_sqlmodel-0.47.0}/src/etlantic_sqlmodel/control_plane/models.py

@@ -353,6 +353,21 @@ class Cp4GovernanceSnapshotRow(SQLModel, table=True):
     updated_at: str | None = None
 
 
+class ScheduleSnapshotRow(SQLModel, table=True):
+    """Transactional 0.47 schedule-store snapshot (reference provider)."""
+
+    __tablename__ = "cp_schedule_snapshot"
+    __table_args__ = (UniqueConstraint("store_id", name="uq_cp_schedule_snapshot"),)
+
+    id: int | None = Field(default=None, primary_key=True)
+    store_id: str = Field(
+        sa_column=Column(String(), index=True, nullable=False), default="default"
+    )
+    payload_json: str = "{}"
+    payload_version: int = Field(default=0, sa_column=Column(Integer(), nullable=False))
+    updated_at: str | None = None
+
+
 __all__ = [
     "AliasRow",
     "Cp4GovernanceSnapshotRow",
@@ -365,6 +380,7 @@ __all__ = [
     "LogicalIdentityRow",
     "PromotionRow",
     "RevisionRow",
+    "ScheduleSnapshotRow",
     "SecurityDomainRow",
     "SubmissionRow",
     "TenantRow",

etlantic_sqlmodel-0.47.0/src/etlantic_sqlmodel/control_plane/schedule_stores.py

@@ -0,0 +1,252 @@
+"""SQLModel-backed ScheduleStore (0.47).
+
+Reference provider: each mutating call loads the schedule snapshot inside a
+database transaction, applies MemoryScheduleStore semantics, and writes the
+snapshot back. When the paired DurableWorkStore is the SQLModel provider on
+the same engine, firing claim and durable accept share one commit.
+"""
+
+from __future__ import annotations
+
+import json
+from collections.abc import Callable, Sequence
+from datetime import UTC, datetime
+from typing import Any, TypeVar
+
+from sqlalchemy.engine import Engine
+
+from etlantic.control_plane.durable_protocols import DurableWorkStore
+from etlantic.control_plane.errors import ControlPlaneError
+from etlantic.control_plane.models import ControlPlaneContext
+from etlantic.control_plane.schedule_memory import MemoryScheduleStore
+from etlantic.control_plane.schedule_models import (
+    FiringRecord,
+    ScheduleRecord,
+    ScheduleSpec,
+)
+from etlantic.control_plane.schedule_protocols import SchedulerLeaderLease
+from etlantic_sqlmodel.control_plane.durable_stores import SQLModelDurableWorkStore
+from etlantic_sqlmodel.control_plane.models import ScheduleSnapshotRow
+from etlantic_sqlmodel.control_plane.session import session_scope
+from sqlmodel import Session, SQLModel, select
+
+T = TypeVar("T")
+
+SCHEDULE_TABLES = (ScheduleSnapshotRow,)
+
+
+def create_schedule_tables(engine: Engine) -> None:
+    """Create schedule snapshot tables (tests/demos only)."""
+    SQLModel.metadata.create_all(
+        engine,
+        tables=[cls.__table__ for cls in SCHEDULE_TABLES],  # type: ignore[list-item]
+    )
+
+
+def _utcnow_iso() -> str:
+    return datetime.now(UTC).isoformat().replace("+00:00", "Z")
+
+
+class SQLModelScheduleStore:
+    """Transactional ScheduleStore backed by a SQL snapshot row."""
+
+    def __init__(self, engine: Engine, *, store_id: str = "default") -> None:
+        self.engine = engine
+        self.store_id = store_id
+
+    def _txn(self, fn: Callable[[MemoryScheduleStore], T]) -> T:
+        with session_scope(self.engine) as session:
+            mem, version = self._read(session, for_update=True)
+            result = fn(mem)
+            self._write(session, mem, expected_version=version)
+            return result
+
+    def _read_only(self, fn: Callable[[MemoryScheduleStore], T]) -> T:
+        with session_scope(self.engine) as session:
+            mem, _version = self._read(session, for_update=False)
+            return fn(mem)
+
+    def _read(
+        self, session: Session, *, for_update: bool
+    ) -> tuple[MemoryScheduleStore, int]:
+        stmt = select(ScheduleSnapshotRow).where(
+            ScheduleSnapshotRow.store_id == self.store_id
+        )
+        if for_update:
+            stmt = stmt.with_for_update()
+        row = session.exec(stmt).first()
+        store = MemoryScheduleStore()
+        if row is None:
+            return store, 0
+        store.load(json.loads(row.payload_json or "{}"))
+        return store, int(row.payload_version or 0)
+
+    def _write(
+        self,
+        session: Session,
+        store: MemoryScheduleStore,
+        *,
+        expected_version: int,
+    ) -> None:
+        payload = json.dumps(store.dump(), sort_keys=True)
+        row = session.exec(
+            select(ScheduleSnapshotRow)
+            .where(ScheduleSnapshotRow.store_id == self.store_id)
+            .with_for_update()
+        ).first()
+        if row is None:
+            if expected_version != 0:
+                raise ControlPlaneError.conflict(
+                    "Schedule snapshot version conflict (missing row)"
+                )
+            session.add(
+                ScheduleSnapshotRow(
+                    store_id=self.store_id,
+                    payload_json=payload,
+                    payload_version=1,
+                    updated_at=_utcnow_iso(),
+                )
+            )
+            return
+        current = int(row.payload_version or 0)
+        if current != expected_version:
+            raise ControlPlaneError.conflict(
+                "Schedule snapshot version conflict (stale write)"
+            )
+        row.payload_json = payload
+        row.payload_version = current + 1
+        row.updated_at = _utcnow_iso()
+        session.add(row)
+
+    def create(
+        self,
+        ctx: ControlPlaneContext,
+        *,
+        definition_id: str,
+        profile_name: str,
+        spec: ScheduleSpec,
+        schedule_id: str | None = None,
+        policy_fingerprint: str = "",
+        parameter_refs: dict[str, str] | None = None,
+        secret_refs: dict[str, str] | None = None,
+        next_fire_at: str | None = None,
+    ) -> ScheduleRecord:
+        return self._txn(
+            lambda m: m.create(
+                ctx,
+                definition_id=definition_id,
+                profile_name=profile_name,
+                spec=spec,
+                schedule_id=schedule_id,
+                policy_fingerprint=policy_fingerprint,
+                parameter_refs=parameter_refs,
+                secret_refs=secret_refs,
+                next_fire_at=next_fire_at,
+            )
+        )
+
+    def get(self, ctx: ControlPlaneContext, schedule_id: str) -> ScheduleRecord:
+        return self._read_only(lambda m: m.get(ctx, schedule_id))
+
+    def list_schedules(self, ctx: ControlPlaneContext) -> Sequence[ScheduleRecord]:
+        return self._read_only(lambda m: m.list_schedules(ctx))
+
+    def pause(self, ctx: ControlPlaneContext, schedule_id: str) -> ScheduleRecord:
+        return self._txn(lambda m: m.pause(ctx, schedule_id))
+
+    def resume(self, ctx: ControlPlaneContext, schedule_id: str) -> ScheduleRecord:
+        return self._txn(lambda m: m.resume(ctx, schedule_id))
+
+    def delete(self, ctx: ControlPlaneContext, schedule_id: str) -> ScheduleRecord:
+        return self._txn(lambda m: m.delete(ctx, schedule_id))
+
+    def acquire_leader_lease(
+        self, ctx: ControlPlaneContext, *, owner_id: str, ttl_seconds: int
+    ) -> SchedulerLeaderLease:
+        return self._txn(
+            lambda m: m.acquire_leader_lease(
+                ctx, owner_id=owner_id, ttl_seconds=ttl_seconds
+            )
+        )
+
+    def heartbeat_leader(
+        self,
+        ctx: ControlPlaneContext,
+        *,
+        owner_id: str,
+        fencing_token: int,
+        ttl_seconds: int,
+    ) -> SchedulerLeaderLease:
+        return self._txn(
+            lambda m: m.heartbeat_leader(
+                ctx,
+                owner_id=owner_id,
+                fencing_token=fencing_token,
+                ttl_seconds=ttl_seconds,
+            )
+        )
+
+    def release_leader(
+        self, ctx: ControlPlaneContext, *, owner_id: str, fencing_token: int
+    ) -> None:
+        self._txn(
+            lambda m: m.release_leader(
+                ctx, owner_id=owner_id, fencing_token=fencing_token
+            )
+        )
+
+    def due_schedules(
+        self, ctx: ControlPlaneContext, *, now: str
+    ) -> Sequence[ScheduleRecord]:
+        return self._read_only(lambda m: m.due_schedules(ctx, now=now))
+
+    def list_firings(
+        self, ctx: ControlPlaneContext, schedule_id: str
+    ) -> Sequence[FiringRecord]:
+        return self._read_only(lambda m: m.list_firings(ctx, schedule_id))
+
+    def claim_firing(
+        self,
+        ctx: ControlPlaneContext,
+        *,
+        schedule_id: str,
+        revision_id: str,
+        nominal_fire_time: str,
+        owner_id: str,
+        fencing_token: int,
+        plan_fingerprint: str,
+        durable: DurableWorkStore | None = None,
+        next_fire_at: str | None = None,
+        require_leader_lease: bool = True,
+        skip_status: str | None = None,
+    ) -> tuple[FiringRecord, bool]:
+        kwargs: dict[str, Any] = {
+            "schedule_id": schedule_id,
+            "revision_id": revision_id,
+            "nominal_fire_time": nominal_fire_time,
+            "owner_id": owner_id,
+            "fencing_token": fencing_token,
+            "plan_fingerprint": plan_fingerprint,
+            "next_fire_at": next_fire_at,
+            "require_leader_lease": require_leader_lease,
+            "skip_status": skip_status,
+        }
+        if (
+            isinstance(durable, SQLModelDurableWorkStore)
+            and durable.engine is self.engine
+        ):
+            with session_scope(self.engine) as session:
+                sched, sv = self._read(session, for_update=True)
+                dur_mem, dv = durable._read(session, for_update=True)
+                result = sched.claim_firing(ctx, durable=dur_mem, **kwargs)
+                self._write(session, sched, expected_version=sv)
+                durable._write(session, dur_mem, expected_version=dv)
+                return result
+        return self._txn(lambda m: m.claim_firing(ctx, durable=durable, **kwargs))
+
+
+__all__ = [
+    "SCHEDULE_TABLES",
+    "SQLModelScheduleStore",
+    "create_schedule_tables",
+]

{etlantic_sqlmodel-0.45.0 → etlantic_sqlmodel-0.47.0}/src/etlantic_sqlmodel/migrations/__init__.py

@@ -18,6 +18,7 @@ VERSIONS: Sequence[str] = (
     "001_registry_cp2",
     "002_durable_cp3",
     "003_cp4_governance",
+    "004_schedules_0_47",
 )
 
 

etlantic_sqlmodel-0.47.0/src/etlantic_sqlmodel/migrations/versions/004_schedules_0_47.py

@@ -0,0 +1,29 @@
+"""004 — schedule snapshot storage for 0.47 ScheduleStore."""
+
+from __future__ import annotations
+
+from sqlalchemy.engine import Engine
+
+from etlantic_sqlmodel.control_plane.models import ScheduleSnapshotRow
+from sqlmodel import SQLModel
+
+TABLES = (ScheduleSnapshotRow,)
+
+
+def upgrade(engine: Engine) -> None:
+    """Create schedule snapshot tables."""
+    SQLModel.metadata.create_all(
+        engine,
+        tables=[cls.__table__ for cls in TABLES],  # type: ignore[list-item]
+    )
+
+
+def downgrade(engine: Engine) -> None:
+    """Drop schedule snapshot tables."""
+    SQLModel.metadata.drop_all(
+        engine,
+        tables=[cls.__table__ for cls in reversed(TABLES)],  # type: ignore[list-item]
+    )
+
+
+__all__ = ["downgrade", "upgrade"]