satisfactoscript 0.5.9__tar.gz → 0.5.10__tar.gz

{satisfactoscript-0.5.9 → satisfactoscript-0.5.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: satisfactoscript
-Version: 0.5.9
+Version: 0.5.10
 Summary: An Enterprise-Ready, Declarative Data Engineering Framework for Databricks Lakehouse.
 Author: julhouba
 Classifier: Programming Language :: Python :: 3

{satisfactoscript-0.5.9 → satisfactoscript-0.5.10}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "satisfactoscript"
-version = "0.5.9"
+version = "0.5.10"
 description = "An Enterprise-Ready, Declarative Data Engineering Framework for Databricks Lakehouse."
 readme = "README.md"
 authors = [

{satisfactoscript-0.5.9 → satisfactoscript-0.5.10}/src/satisfactoscript/__init__.py

@@ -1,5 +1,6 @@
 from .core.core import SatisfactoEngine
 from .core.registry import RuleRegistry
 from .core.config import ConfigurationManager
+from .utils import safe_columns
 
-__all__ = ["SatisfactoEngine", "RuleRegistry", "ConfigurationManager"]
+__all__ = ["SatisfactoEngine", "RuleRegistry", "ConfigurationManager", "safe_columns"]

satisfactoscript-0.5.10/src/satisfactoscript/registry.py

@@ -0,0 +1,7 @@
+"""
+Compatibility shim — keeps `from satisfactoscript.registry import RuleRegistry` working.
+The canonical location is satisfactoscript.core.registry.
+"""
+from satisfactoscript.core.registry import RuleRegistry
+
+__all__ = ["RuleRegistry"]

satisfactoscript-0.5.10/src/satisfactoscript/utils.py

@@ -0,0 +1,29 @@
+"""
+Utility helpers for rule authoring — compatible with Databricks Connect v2.
+"""
+
+
+def safe_columns(df):
+    """
+    Returns the column names of a DataFrame without triggering a failing AnalyzePlan RPC.
+
+    On native Databricks (cluster / notebook), this delegates to ``df.columns`` as normal.
+    On Databricks Connect v2 from local environments (PyCharm / VS Code on Windows), the
+    ``AnalyzePlan`` gRPC call that backs ``df.columns`` can fail with
+    ``Missing required field 'UserContext'``.  In that case this function returns an empty
+    list so that optional-column branches in business rules are silently skipped rather than
+    crashing.  Core transformations (withColumn, filter, write …) are unaffected because they
+    go through ``ExecutePlan``, not ``AnalyzePlan``.
+
+    Args:
+        df: A PySpark DataFrame.
+
+    Returns:
+        list[str]: Column names, or ``[]`` when schema inspection fails locally.
+    """
+    try:
+        return df.columns
+    except Exception as e:
+        if "UserContext" in str(e) or "INVALID_ARGUMENT" in str(e):
+            return []
+        raise

{satisfactoscript-0.5.9 → satisfactoscript-0.5.10}/src/satisfactoscript.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: satisfactoscript
-Version: 0.5.9
+Version: 0.5.10
 Summary: An Enterprise-Ready, Declarative Data Engineering Framework for Databricks Lakehouse.
 Author: julhouba
 Classifier: Programming Language :: Python :: 3

{satisfactoscript-0.5.9 → satisfactoscript-0.5.10}/src/satisfactoscript.egg-info/SOURCES.txt

@@ -1,6 +1,8 @@
 README.md
 pyproject.toml
 src/satisfactoscript/__init__.py
+src/satisfactoscript/registry.py
+src/satisfactoscript/utils.py
 src/satisfactoscript.egg-info/PKG-INFO
 src/satisfactoscript.egg-info/SOURCES.txt
 src/satisfactoscript.egg-info/dependency_links.txt
@@ -23,4 +25,6 @@ tests/test_core_join.py
 tests/test_core_username.py
 tests/test_dummy.py
 tests/test_loaders.py
-tests/test_registry.py
+tests/test_registry.py
+tests/test_registry_import_paths.py
+tests/test_utils_safe_columns.py

satisfactoscript-0.5.10/tests/test_registry_import_paths.py

@@ -0,0 +1,26 @@
+"""
+Tests for RuleRegistry import path compatibility.
+All import paths must resolve to the same singleton class.
+"""
+from satisfactoscript.core.registry import RuleRegistry as RR_core
+from satisfactoscript.registry import RuleRegistry as RR_shim
+from satisfactoscript import RuleRegistry as RR_top
+
+
+def test_all_import_paths_resolve_to_same_class():
+    """All three import paths must return the exact same class (same singleton)."""
+    assert RR_core is RR_shim
+    assert RR_core is RR_top
+
+
+def test_rule_registered_via_shim_is_visible_everywhere():
+    """A rule registered using the shim import must be retrievable from all paths."""
+    @RR_shim.register_rule(name="__test_shim_rule__")
+    def _my_rule(df):
+        return df
+
+    assert RR_core.get_rule("__test_shim_rule__") is _my_rule
+    assert RR_top.get_rule("__test_shim_rule__") is _my_rule
+
+    # Cleanup
+    RR_core._rules.pop("__test_shim_rule__", None)

satisfactoscript-0.5.10/tests/test_utils_safe_columns.py

@@ -0,0 +1,46 @@
+"""
+Tests for satisfactoscript.utils.safe_columns.
+"""
+import pytest
+from unittest.mock import MagicMock, PropertyMock
+from satisfactoscript import safe_columns
+
+
+class TestSafeColumns:
+
+    def test_returns_df_columns_normally(self):
+        df = MagicMock()
+        type(df).columns = PropertyMock(return_value=["id", "name", "value"])
+
+        assert safe_columns(df) == ["id", "name", "value"]
+
+    def test_returns_empty_list_on_usercontext_grpc_error(self):
+        df = MagicMock()
+        type(df).columns = PropertyMock(
+            side_effect=Exception("gRPC: Missing required field 'UserContext' in the request.")
+        )
+
+        result = safe_columns(df)
+
+        assert result == []
+
+    def test_returns_empty_list_on_invalid_argument_grpc_error(self):
+        df = MagicMock()
+        type(df).columns = PropertyMock(
+            side_effect=Exception("StatusCode.INVALID_ARGUMENT details: something")
+        )
+
+        result = safe_columns(df)
+
+        assert result == []
+
+    def test_reraises_unrelated_exceptions(self):
+        df = MagicMock()
+        type(df).columns = PropertyMock(side_effect=RuntimeError("disk full"))
+
+        with pytest.raises(RuntimeError, match="disk full"):
+            safe_columns(df)
+
+    def test_importable_from_satisfactoscript_top_level(self):
+        from satisfactoscript import safe_columns as sc
+        assert callable(sc)