mainsequence 4.3.7__tar.gz → 4.3.8__tar.gz

{mainsequence-4.3.7/mainsequence.egg-info → mainsequence-4.3.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mainsequence
-Version: 4.3.7
+Version: 4.3.8
 Summary: Main Sequence SDK 
 Author-email: Main Sequence GmbH <dev@main-sequence.io>
 License: MainSequence GmbH SDK License Agreement

{mainsequence-4.3.7 → mainsequence-4.3.8}/mainsequence/meta_tables/migrations.py

@@ -930,7 +930,9 @@ def _collection_create_row_from_registration_request(
         "alembic_revision": alembic_revision,
         "physical_table_name": str(physical_table_name),
         "protect_from_deletion": bool(getattr(request, "protect_from_deletion", False)),
-        "table_contract": request.table_contract,
+        "table_contract": _collection_create_table_contract_payload(
+            request.table_contract
+        ),
     }
     time_index_name = getattr(request, "time_index_name", None)
     if time_index_name not in (None, ""):
@@ -944,6 +946,30 @@ def _collection_create_row_from_registration_request(
     return row
 
 
+def _collection_create_table_contract_payload(table_contract: Any) -> dict[str, Any]:
+    if isinstance(table_contract, MetaTableContract):
+        payload = table_contract.model_dump(mode="json", exclude_none=True)
+    elif hasattr(table_contract, "model_dump"):
+        payload = table_contract.model_dump(mode="json", exclude_none=True)
+    elif isinstance(table_contract, Mapping):
+        payload = dict(table_contract)
+    else:
+        raise TypeError("Alembic MetaTable collection-create requires table_contract.")
+
+    physical = payload.get("physical")
+    if hasattr(physical, "model_dump"):
+        physical_payload = physical.model_dump(mode="json", exclude_none=True)
+    elif isinstance(physical, Mapping):
+        physical_payload = dict(physical)
+    else:
+        physical_payload = None
+    if physical_payload is not None:
+        for schema_key in ("schema", "schema_", "table_schema", "schema_name"):
+            physical_payload.pop(schema_key, None)
+        payload["physical"] = physical_payload
+    return payload
+
+
 def _request_contract_physical_table_name(request: Any) -> str | None:
     table_contract = getattr(request, "table_contract", None)
     if isinstance(table_contract, MetaTableContract):

{mainsequence-4.3.7 → mainsequence-4.3.8/mainsequence.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mainsequence
-Version: 4.3.7
+Version: 4.3.8
 Summary: Main Sequence SDK 
 Author-email: Main Sequence GmbH <dev@main-sequence.io>
 License: MainSequence GmbH SDK License Agreement

{mainsequence-4.3.7 → mainsequence-4.3.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "mainsequence"
-version = "4.3.7"
+version = "4.3.8"
 description = "Main Sequence SDK "
 readme = "README.md"
 requires-python = ">=3.11"

{mainsequence-4.3.7 → mainsequence-4.3.8}/tests/test_meta_table_migrations.py

@@ -853,10 +853,12 @@ def test_prepare_for_alembic_preserves_authored_table_names(monkeypatch):
         assert all(table["management_mode"] == "platform_managed" for table in rows)
         assert all("schema_management" not in table for table in rows)
         assert all(table["protect_from_deletion"] is False for table in rows)
-        assert not hasattr(rows[0]["table_contract"], "indexes")
-        assert not hasattr(rows[1]["table_contract"], "indexes")
-        assert not hasattr(rows[0]["table_contract"], "foreign_keys")
-        assert not hasattr(rows[1]["table_contract"], "foreign_keys")
+        assert "indexes" not in rows[0]["table_contract"]
+        assert "indexes" not in rows[1]["table_contract"]
+        assert "foreign_keys" not in rows[0]["table_contract"]
+        assert "foreign_keys" not in rows[1]["table_contract"]
+        assert "schema" not in rows[0]["table_contract"]["physical"]
+        assert "schema" not in rows[1]["table_contract"]["physical"]
 
         response_tables = []
         for table in rows:
@@ -909,12 +911,20 @@ def test_prepare_for_alembic_preserves_authored_table_names(monkeypatch):
     ]
     assert Account.__table__.name == "example_assets__account"
     assert Asset.__table__.name == "example_assets__asset"
-    assert reserved_payloads[0]["table_contract"].physical.table_name == "example_assets__account"
-    assert reserved_payloads[1]["table_contract"].physical.table_name == "example_assets__asset"
+    assert (
+        reserved_payloads[0]["table_contract"]["physical"]["table_name"]
+        == "example_assets__account"
+    )
+    assert (
+        reserved_payloads[1]["table_contract"]["physical"]["table_name"]
+        == "example_assets__asset"
+    )
     assert "backend" not in str(next(iter(Asset.__table__.indexes)).name)
     assert next(iter(Asset.__table__.foreign_key_constraints)).name == "asset_account_uid_fkey"
-    assert not hasattr(reserved_payloads[1]["table_contract"], "indexes")
-    assert not hasattr(reserved_payloads[1]["table_contract"], "foreign_keys")
+    assert "indexes" not in reserved_payloads[1]["table_contract"]
+    assert "foreign_keys" not in reserved_payloads[1]["table_contract"]
+    assert "schema" not in reserved_payloads[0]["table_contract"]["physical"]
+    assert "schema" not in reserved_payloads[1]["table_contract"]["physical"]
     assert "schema_management" not in reserved_payloads[0]
     assert reserved_payloads[0]["protect_from_deletion"] is False
 
@@ -958,8 +968,9 @@ def test_prepare_for_alembic_does_not_resolve_foreign_key_targets(monkeypatch):
         assert rows[0]["migration_namespace"] == "markets"
         assert rows[0]["migration_provider_key"] == "sample:markets"
         assert rows[0]["is_alembic_managed"] is True
-        assert not hasattr(rows[0]["table_contract"], "indexes")
-        assert not hasattr(rows[0]["table_contract"], "foreign_keys")
+        assert "indexes" not in rows[0]["table_contract"]
+        assert "foreign_keys" not in rows[0]["table_contract"]
+        assert "schema" not in rows[0]["table_contract"]["physical"]
         assert "schema_management" not in rows[0]
         return [
             _reserved_metatable(
@@ -1194,8 +1205,9 @@ def test_prepare_for_alembic_reserves_existing_table_name_with_provider_identity
         assert all(table["migration_namespace"] == "markets" for table in rows)
         assert all(table["migration_provider_key"] == "sample:markets" for table in rows)
         assert all(table["is_alembic_managed"] is True for table in rows)
-        assert not hasattr(rows[0]["table_contract"], "indexes")
-        assert not hasattr(rows[0]["table_contract"], "foreign_keys")
+        assert "indexes" not in rows[0]["table_contract"]
+        assert "foreign_keys" not in rows[0]["table_contract"]
+        assert "schema" not in rows[0]["table_contract"]["physical"]
         assert all("schema_management" not in table for table in rows)
         return [
             _reserved_metatable(