mainsequence 4.4.10__tar.gz → 4.4.11__tar.gz

{mainsequence-4.4.10/mainsequence.egg-info → mainsequence-4.4.11}/PKG-INFO

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

{mainsequence-4.4.10 → mainsequence-4.4.11}/mainsequence/client/base.py

@@ -89,7 +89,7 @@ class BaseObjectOrm:
         "DataNodeUpdateDetails": "ts_manager/local_time_serie_update_details",
         "LocalTimeSerieHistoricalUpdate": "ts_manager/lts_historical_update",
         "DynamicTableDataSource": "ts_manager/dynamic_table_data_source",
-        "DataSource": "pods/data_source",
+        "DataSource": "connections/data_source",
         "Project": "pods/projects",
         "Artifact": "pods/artifact",
         "Job": "pods/job",

{mainsequence-4.4.10 → mainsequence-4.4.11}/mainsequence/client/metatables/core.py

@@ -142,6 +142,7 @@ MetaTableSchemaManagementMode = Literal[
     "external_registered",
 ]
 MetaTableOperation = Literal["select", "insert", "update", "delete", "upsert"]
+StorageAccessMode = Literal["read_write", "read_only", "disabled"]
 COMPILED_SQL_V1 = "compiled-sql.v1"
 MetaTableCompiledSQLVersion = Literal["compiled-sql.v1"]
 MetaTableCompiledSQLDialect = Literal["postgresql"]
@@ -792,6 +793,8 @@ class ManagedMetaTableFinalizeResponse(BasePydanticModel):
 
 
 class DataSource(BasePydanticModel, BaseObjectOrm):
+    model_config = ConfigDict(extra="allow")
+
     uid: str | None = Field(
         None,
         description="Public uid of the data source.",
@@ -801,7 +804,7 @@ class DataSource(BasePydanticModel, BaseObjectOrm):
         description="Compatibility alias for the public data source uid.",
     )
     id: int | None = Field(None, description="The unique identifier of the Local Disk Source Lake")
-    display_name: str
+    display_name: str | None = None
     organization: int | None = Field(
         None, description="The unique identifier of the Local Disk Source Lake"
     )
@@ -809,11 +812,26 @@ class DataSource(BasePydanticModel, BaseObjectOrm):
         None,
         description="Public uid of the owning organization.",
     )
-    class_type: str
-    status: str
+    class_type: str | None = None
+    status: str | None = None
+    storage_access_mode: StorageAccessMode | None = Field(
+        default="read_write",
+        description="Runtime storage access gate: read_write, read_only, or disabled.",
+    )
     extra_arguments: dict | None = None
 
     STATUS_AVAILABLE: ClassVar[str] = "AVAILABLE"
+    STORAGE_ACCESS_MODE_READ_WRITE: ClassVar[str] = "read_write"
+    STORAGE_ACCESS_MODE_READ_ONLY: ClassVar[str] = "read_only"
+    STORAGE_ACCESS_MODE_DISABLED: ClassVar[str] = "disabled"
+
+    @property
+    def allows_runtime_reads(self) -> bool:
+        return self.storage_access_mode in {"read_write", "read_only"}
+
+    @property
+    def allows_runtime_writes(self) -> bool:
+        return self.storage_access_mode == "read_write"
 
     @classmethod
     def get_or_create_duck_db(cls, time_out=None, *args, **kwargs):
@@ -1019,6 +1037,8 @@ class DataSource(BasePydanticModel, BaseObjectOrm):
 
 
 class DynamicTableDataSource(BasePydanticModel, BaseObjectOrm):
+    model_config = ConfigDict(extra="allow", use_enum_values=True)
+
     uid: str | None = Field(
         None,
         description="Public uid of the dynamic table data source.",
@@ -1027,15 +1047,14 @@ class DynamicTableDataSource(BasePydanticModel, BaseObjectOrm):
         None,
         description="Backend numeric row identifier of the time-indexed data source.",
     )
-    related_resource: DataSource
-    related_resource_class_type: str
-
-    class Config:
-        use_enum_values = True
+    related_resource: DataSource | None = None
+    related_resource_class_type: str | None = None
 
     def model_dump_json(self, **json_dumps_kwargs) -> str:
         dump = self.model_dump()
-        dump["related_resource"] = self.related_resource.model_dump()
+        dump["related_resource"] = (
+            self.related_resource.model_dump() if self.related_resource is not None else None
+        )
         return json.dumps(dump, **json_dumps_kwargs)
 
     def _public_uid(self) -> str:

{mainsequence-4.4.10 → mainsequence-4.4.11}/mainsequence/client/models_foundry.py

@@ -10,7 +10,7 @@ from uuid import UUID
 import pandas as pd
 import yaml
 from cachetools import TTLCache, cachedmethod
-from pydantic import Field, SecretStr
+from pydantic import ConfigDict, Field, SecretStr
 
 from mainsequence.logconf import logger
 
@@ -143,6 +143,8 @@ class ProjectQuickSearchResult(BasePydanticModel):
 
 
 class Project(LabelableObjectMixin, ShareableObjectMixin, BasePydanticModel, BaseObjectOrm):
+    model_config = ConfigDict(extra="allow")
+
     FILTERSET_FIELDS: ClassVar[dict[str, list[str]]] = {
         "project_name": ["in", "exact", "contains"],
         "uid": ["in", "exact"],

{mainsequence-4.4.10 → mainsequence-4.4.11/mainsequence.egg-info}/PKG-INFO

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

{mainsequence-4.4.10 → mainsequence-4.4.11}/pyproject.toml

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

{mainsequence-4.4.10 → mainsequence-4.4.11}/tests/test_pod_project_resolution.py

@@ -25,6 +25,7 @@ def _project_payload_public() -> dict:
         "project_name": "Markets Repository",
         "data_source": {
             "uid": DATA_SOURCE_UID,
+            "backend_can_add_fields": "ignored",
             "related_resource": {
                 "uid": DATA_SOURCE_UID,
                 "data_source_uid": DATA_SOURCE_UID,
@@ -32,6 +33,8 @@ def _project_payload_public() -> dict:
                 "organization_uid": ORGANIZATION_UID,
                 "class_type": "timescale_db_remote",
                 "status": "AVAILABLE",
+                "storage_access_mode": "read_write",
+                "backend_can_add_nested_fields": "ignored",
             },
             "related_resource_class_type": "timescale_db_remote",
         },
@@ -39,6 +42,7 @@ def _project_payload_public() -> dict:
         "is_initialized": True,
         "created_by": "user-4",
         "labels": ["markets"],
+        "backend_can_add_project_fields": "ignored",
     }
 
 
@@ -53,6 +57,40 @@ def test_project_deserializes_public_uid_serializer_payload():
     assert project.data_source.related_resource.data_source_uid == DATA_SOURCE_UID
     assert project.data_source.related_resource.organization_uid == ORGANIZATION_UID
     assert project.data_source.related_resource.class_type == "timescale_db_remote"
+    assert project.data_source.related_resource.storage_access_mode == "read_write"
+    assert project.data_source.related_resource.allows_runtime_reads is True
+    assert project.data_source.related_resource.allows_runtime_writes is True
+
+
+def test_project_deserializes_nullable_datasource_related_resource_edge_case():
+    payload = _project_payload_public()
+    payload["data_source"]["related_resource"] = {
+        "uid": None,
+        "data_source_uid": None,
+        "display_name": None,
+        "organization_uid": None,
+        "class_type": None,
+        "status": None,
+        "storage_access_mode": None,
+    }
+
+    project = models_foundry.Project(**payload)
+
+    assert project.data_source is not None
+    assert project.data_source.related_resource is not None
+    assert project.data_source.related_resource.storage_access_mode is None
+    assert project.data_source.related_resource.allows_runtime_reads is False
+    assert project.data_source.related_resource.allows_runtime_writes is False
+
+
+def test_project_deserializes_null_datasource_related_resource_edge_case():
+    payload = _project_payload_public()
+    payload["data_source"]["related_resource"] = None
+
+    project = models_foundry.Project(**payload)
+
+    assert project.data_source is not None
+    assert project.data_source.related_resource is None
 
 
 def test_data_node_update_get_or_create_uses_current_project_uid(monkeypatch):