duckrun 0.2.5.dev2__py3-none-any.whl → 0.2.5.dev4__py3-none-any.whl

duckrun/core.py

@@ -441,6 +441,26 @@ class Duckrun:
             print(f"❌ Error attaching lakehouse: {e}")
             print("Continuing without pre-attached tables.")
 
+    def get_workspace_id(self) -> str:
+        """
+        Get the workspace ID (GUID or name without spaces).
+        Use this when passing workspace parameter to Python functions.
+        
+        Returns:
+            Workspace ID - either a GUID or workspace name without spaces
+        """
+        return self.workspace_id
+    
+    def get_lakehouse_id(self) -> str:
+        """
+        Get the lakehouse ID (GUID or name).
+        Use this when passing lakehouse parameter to Python functions.
+        
+        Returns:
+            Lakehouse ID - either a GUID or lakehouse name
+        """
+        return self.lakehouse_id
+
     def run(self, pipeline: List[Tuple]) -> bool:
         """
         Execute pipeline of tasks.
@@ -702,7 +722,11 @@ class WorkspaceConnection:
             try:
                 import notebookutils  # type: ignore
                 token = notebookutils.credentials.getToken("pbi")
-                workspace_id = notebookutils.runtime.context.get("workspaceId")
+                # Always resolve workspace name to ID, even in notebook environment
+                workspace_id = self._get_workspace_id_by_name(token, self.workspace_name)
+                if not workspace_id:
+                    print(f"Workspace '{self.workspace_name}' not found")
+                    return []
             except ImportError:
                 # Fallback to azure-identity
                 print("Getting authentication token...")
@@ -749,7 +773,11 @@ class WorkspaceConnection:
             try:
                 import notebookutils  # type: ignore
                 token = notebookutils.credentials.getToken("pbi")
-                workspace_id = notebookutils.runtime.context.get("workspaceId")
+                # Always resolve workspace name to ID, even in notebook environment
+                workspace_id = self._get_workspace_id_by_name(token, self.workspace_name)
+                if not workspace_id:
+                    print(f"Workspace '{self.workspace_name}' not found")
+                    return False
             except ImportError:
                 # Fallback to azure-identity
                 print("Getting authentication token...")

duckrun/runner.py

@@ -110,12 +110,40 @@ def run(duckrun_instance, pipeline: List[Tuple]) -> bool:
 
 
 def _run_python(duckrun_instance, name: str, args: tuple) -> Any:
-    """Execute Python task, return result"""
+    """
+    Execute Python task, return result.
+    
+    Automatically substitutes workspace/lakehouse names in args with their resolved IDs
+    to prevent URL encoding issues with names containing spaces.
+    """
     duckrun_instance._create_onelake_secret()
     func = _load_py_function(duckrun_instance, name)
     if not func:
         raise RuntimeError(f"Python function '{name}' not found")
     
+    # Get original and resolved names
+    original_workspace = duckrun_instance.workspace
+    original_lakehouse = duckrun_instance.lakehouse_name
+    resolved_workspace = duckrun_instance.workspace_id
+    resolved_lakehouse = duckrun_instance.lakehouse_id
+    
+    # Substitute workspace/lakehouse names in args if they differ
+    # This prevents URL encoding issues when names contain spaces
+    substituted_args = []
+    needs_substitution = (original_workspace != resolved_workspace or 
+                         original_lakehouse != resolved_lakehouse)
+    
+    if needs_substitution:
+        for arg in args:
+            if arg == original_workspace:
+                substituted_args.append(resolved_workspace)
+            elif arg == original_lakehouse:
+                substituted_args.append(resolved_lakehouse)
+            else:
+                substituted_args.append(arg)
+        args = tuple(substituted_args)
+        print(f"📝 Auto-substituted workspace/lakehouse names in args for URL compatibility")
+    
     print(f"Running Python: {name}{args}")
     result = func(*args)
     print(f"✅ Python '{name}' completed")

{duckrun-0.2.5.dev2.dist-info → duckrun-0.2.5.dev4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: duckrun
-Version: 0.2.5.dev2
+Version: 0.2.5.dev4
 Summary: Lakehouse task runner powered by DuckDB for Microsoft Fabric
 Author: mim
 License: MIT

duckrun-0.2.5.dev4.dist-info/RECORD

@@ -0,0 +1,12 @@
+duckrun/__init__.py,sha256=XA85pL2vK1AkmBic8e7WxeqNvcd6SjFX4zsQpImDO6E,230
+duckrun/core.py,sha256=6Us3dRuQFCziK12r0j2CuwgcDQeV78iEeLcA40IwIiA,39476
+duckrun/files.py,sha256=piWRU5w9jHrW-wuV4Gf-SKY_jhFv9eflxgWO8AZCQTI,10495
+duckrun/lakehouse.py,sha256=j--Z3zo8AOWt1GF9VzRosmmTAy6ey2D0LVubti58twU,14109
+duckrun/runner.py,sha256=XsQqWlesFD2cuhH2gsQj3Astg0XN7xhW15WPmr8D65I,13797
+duckrun/stats.py,sha256=2FTqoQNVjD84-H1HjStHxZkOpAGKXS79M55B00pOlok,9804
+duckrun/writer.py,sha256=eWrGtDQTbXi8H3sSt2WucYTdEQUjK97KmQxzCbqAuMs,6221
+duckrun-0.2.5.dev4.dist-info/licenses/LICENSE,sha256=-DeQQwdbCbkB4507ZF3QbocysB-EIjDtaLexvqRkGZc,1083
+duckrun-0.2.5.dev4.dist-info/METADATA,sha256=HE5rFI5A227bw-prSq3cHv9wR9-etSYJD31lIrKjpzs,18344
+duckrun-0.2.5.dev4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+duckrun-0.2.5.dev4.dist-info/top_level.txt,sha256=BknMEwebbUHrVAp3SC92ps8MPhK7XSYsaogTvi_DmEU,8
+duckrun-0.2.5.dev4.dist-info/RECORD,,

duckrun-0.2.5.dev2.dist-info/RECORD

@@ -1,12 +0,0 @@
-duckrun/__init__.py,sha256=XA85pL2vK1AkmBic8e7WxeqNvcd6SjFX4zsQpImDO6E,230
-duckrun/core.py,sha256=UgBE90zTFvnieTrUEb4tDA2cWSwFfh24M_e46FTmFvg,38345
-duckrun/files.py,sha256=piWRU5w9jHrW-wuV4Gf-SKY_jhFv9eflxgWO8AZCQTI,10495
-duckrun/lakehouse.py,sha256=j--Z3zo8AOWt1GF9VzRosmmTAy6ey2D0LVubti58twU,14109
-duckrun/runner.py,sha256=lfwNoU1CZXh6bPTHvGWVaUWjzG5crvT7Pzq4onMEVjw,12576
-duckrun/stats.py,sha256=2FTqoQNVjD84-H1HjStHxZkOpAGKXS79M55B00pOlok,9804
-duckrun/writer.py,sha256=eWrGtDQTbXi8H3sSt2WucYTdEQUjK97KmQxzCbqAuMs,6221
-duckrun-0.2.5.dev2.dist-info/licenses/LICENSE,sha256=-DeQQwdbCbkB4507ZF3QbocysB-EIjDtaLexvqRkGZc,1083
-duckrun-0.2.5.dev2.dist-info/METADATA,sha256=wiB12-pG_jlyUbghVNMyE3KHcwt8GFJd04VNZ2C4VwE,18344
-duckrun-0.2.5.dev2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-duckrun-0.2.5.dev2.dist-info/top_level.txt,sha256=BknMEwebbUHrVAp3SC92ps8MPhK7XSYsaogTvi_DmEU,8
-duckrun-0.2.5.dev2.dist-info/RECORD,,