mainsequence 4.3.26__tar.gz → 4.3.30__tar.gz

{mainsequence-4.3.26/mainsequence.egg-info → mainsequence-4.3.30}/PKG-INFO

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

{mainsequence-4.3.26 → mainsequence-4.3.30}/mainsequence/client/metatables/core.py

@@ -44,7 +44,7 @@ from ..dtype_codec import (
     serialize_remote_parameters,
     token_to_pandas_series,
 )
-from ..exceptions import raise_for_response
+from ..exceptions import AuthenticationError, PermissionDeniedError, raise_for_response
 from ..utils import (
     TDAG_CONSTANTS,
     DateInfo,
@@ -488,7 +488,8 @@ class MetaTableOperationScope(BasePydanticModel):
         except Exception as exc:
             raise ValueError(
                 "MetaTable compiled SQL scope requires data_source_uid or a "
-                "configured project/session default data source."
+                "configured project/session default data source. "
+                f"Default data source resolution failed: {exc}"
             ) from exc
 
         uid = getattr(data_source, "uid", None) or getattr(data_source, "data_source_uid", None)
@@ -4597,6 +4598,14 @@ def _build_local_pod_project_resolution() -> _PodProjectResolution:
         )
     try:
         project = Project.get(pk=running_project_uid)
+    except (AuthenticationError, PermissionDeniedError) as exc:
+        raise RuntimeError(
+            "Could not resolve the local project default data source because SDK "
+            "authentication/authorization failed while loading project "
+            f"{running_project_uid!r}. "
+            "Run `mainsequence login` or export MAINSEQUENCE_ACCESS_TOKEN / "
+            f"MAINSEQUENCE_REFRESH_TOKEN. Backend response: {exc}"
+        ) from exc
     except DoesNotExist:
         return _PodProjectResolution(
             project=None,
@@ -4778,7 +4787,15 @@ def get_session_data_source() -> Any:
         SessionDataSource.set_remote_db()
         data_source = getattr(SessionDataSource, "data_source", None)
     if getattr(data_source, "related_resource", None) is None:
-        raise Exception("This Pod does not have a default data source")
+        resolution = _resolve_local_pod_project()
+        detail = (resolution.detail or "No local pod project attached.").strip()
+        if detail and not detail.endswith("."):
+            detail += "."
+        raise RuntimeError(
+            "Could not resolve a session default data source. "
+            f"{detail} Run inside a configured Main Sequence project/session, "
+            "or pass data_source_uid explicitly."
+        )
     return data_source
 
 

{mainsequence-4.3.26 → mainsequence-4.3.30}/mainsequence/client/models_foundry.py

@@ -1071,8 +1071,3 @@ class Constant(ShareableObjectMixin, BasePydanticModel, BaseObjectOrm):
             new_constant = cls.create(name=k, value=v)
             created_constants.append(new_constant)
         return created_constants
-
-
-from . import metatables as _models_metatables  # noqa: E402
-
-_models_metatables.SessionDataSource.set_remote_db()

{mainsequence-4.3.26 → mainsequence-4.3.30/mainsequence.egg-info}/PKG-INFO

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

{mainsequence-4.3.26 → mainsequence-4.3.30}/pyproject.toml

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

{mainsequence-4.3.26 → mainsequence-4.3.30}/tests/test_dependency_extras.py

@@ -6,6 +6,16 @@ import pathlib
 import sys
 import types
 
+import pytest
+
+
+def _snapshot_mainsequence_modules() -> dict[str, types.ModuleType]:
+    return {
+        name: module
+        for name, module in sys.modules.items()
+        if name == "mainsequence" or name.startswith("mainsequence.")
+    }
+
 
 def _reset_mainsequence_modules() -> None:
     for name in tuple(sys.modules):
@@ -13,6 +23,14 @@ def _reset_mainsequence_modules() -> None:
             sys.modules.pop(name, None)
 
 
+@pytest.fixture(autouse=True)
+def _restore_mainsequence_modules_after_import_test():
+    snapshot = _snapshot_mainsequence_modules()
+    yield
+    _reset_mainsequence_modules()
+    sys.modules.update(snapshot)
+
+
 def _seed_mainsequence_packages() -> None:
     repo_root = pathlib.Path(__file__).resolve().parents[1]
     pkg_root = repo_root / "mainsequence"
@@ -52,6 +70,26 @@ def test_core_client_import_does_not_require_duckdb(monkeypatch):
     assert not hasattr(foundry_module, "DynamicTableDataSource")
 
 
+def test_models_foundry_import_does_not_resolve_session_data_source(monkeypatch):
+    _reset_mainsequence_modules()
+    _seed_mainsequence_packages()
+
+    metatables_module = importlib.import_module("mainsequence.client.metatables")
+
+    def _fail_import_time_resolution():
+        raise AssertionError("models_foundry import must not resolve SessionDataSource")
+
+    monkeypatch.setattr(
+        metatables_module.SessionDataSource,
+        "set_remote_db",
+        _fail_import_time_resolution,
+    )
+
+    foundry_module = importlib.import_module("mainsequence.client.models_foundry")
+
+    assert hasattr(foundry_module, "Project")
+
+
 def test_duckdb_helper_points_to_local_data_extra(monkeypatch):
     _reset_mainsequence_modules()
     _seed_mainsequence_packages()

{mainsequence-4.3.26 → mainsequence-4.3.30}/tests/test_meta_tables_client_models.py

@@ -956,6 +956,34 @@ def test_compiled_sql_v1_scope_resolves_session_data_source(monkeypatch):
     )
 
 
+def test_compiled_sql_v1_scope_preserves_session_data_source_auth_failure(monkeypatch):
+    def _raise_auth_failure():
+        raise RuntimeError(
+            "Could not resolve the local project default data source because SDK "
+            "authentication/authorization failed."
+        )
+
+    monkeypatch.setattr(meta_table_models, "get_session_data_source", _raise_auth_failure)
+
+    with pytest.raises(ValidationError) as exc_info:
+        build_operation(
+            operation="select",
+            sql="SELECT asset.symbol FROM public.asset AS asset",
+            scope={
+                "tables": [
+                    {
+                        "metaTableUid": "aaaaaaaa-aaaa-4aaa-8aaa-aaaaaaaaaaaa",
+                        "alias": "asset",
+                    }
+                ]
+            },
+        )
+
+    message = str(exc_info.value)
+    assert "Default data source resolution failed" in message
+    assert "SDK authentication/authorization failed" in message
+
+
 def test_compiled_sql_v1_serializes_typed_temporal_parameters():
     operation = build_operation(
         operation="select",

{mainsequence-4.3.26 → mainsequence-4.3.30}/tests/test_pod_project_resolution.py

@@ -4,6 +4,7 @@ import pytest
 
 import mainsequence.client.metatables as models_metatables
 import mainsequence.client.models_foundry as models_foundry
+from mainsequence.client.exceptions import AuthenticationError
 from mainsequence.meta_tables.data_nodes import build_operations
 
 PROJECT_UID = "1d0530c0-65d1-4db0-856b-dc29d8260a09"
@@ -154,6 +155,41 @@ def test_set_remote_db_warns_once_for_lookup_failure(monkeypatch):
     assert debugs == []
 
 
+def test_resolve_local_pod_project_raises_auth_failure(monkeypatch):
+    monkeypatch.setenv("MAIN_SEQUENCE_PROJECT_UID", PROJECT_UID)
+    monkeypatch.delenv("MAIN_SEQUENCE_PROJECT_ID", raising=False)
+
+    def _project_get(*args, **kwargs):
+        raise AuthenticationError("No auth configured.")
+
+    monkeypatch.setattr(models_foundry.Project, "get", _project_get)
+
+    with pytest.raises(RuntimeError) as exc_info:
+        models_metatables._resolve_local_pod_project()
+
+    message = str(exc_info.value)
+    assert "SDK authentication/authorization failed" in message
+    assert f"project {PROJECT_UID!r}" in message
+    assert "mainsequence login" in message
+    assert "No auth configured." in message
+
+
+def test_get_session_data_source_reports_missing_project_uid(monkeypatch):
+    monkeypatch.delenv("MAIN_SEQUENCE_PROJECT_UID", raising=False)
+    monkeypatch.delenv("MAIN_SEQUENCE_PROJECT_ID", raising=False)
+    monkeypatch.setattr(models_metatables, "SessionDataSource", models_metatables.PodDataSource())
+    monkeypatch.setattr(models_metatables.logger, "debug", lambda message: None)
+
+    with pytest.raises(RuntimeError) as exc_info:
+        models_metatables.get_session_data_source()
+
+    message = str(exc_info.value)
+    assert "Could not resolve a session default data source." in message
+    assert "MAIN_SEQUENCE_PROJECT_UID is not configured." in message
+    assert "pass data_source_uid explicitly" in message
+    assert "This Pod does not have a default data source" not in message
+
+
 def test_data_source_create_duckdb_makes_creation_explicit(monkeypatch):
     captured = {}
     monkeypatch.setattr(models_metatables, "bios_uuid", lambda: "host-123")