duckrun 0.2.14.dev1__py3-none-any.whl → 0.2.14.dev2__py3-none-any.whl

duckrun/__init__.py

@@ -2,7 +2,7 @@
 
 from duckrun.core import Duckrun
 
-__version__ = "0.2.14.dev1"
+__version__ = "0.2.14.dev2"
 
 # Expose unified connect method at module level
 connect = Duckrun.connect

duckrun/core.py

@@ -587,24 +587,97 @@ class Duckrun:
         except Exception as e:
             print(f"⚠️  Warning: Could not register lookup functions: {e}")
 
-    def get_workspace_id(self) -> str:
+    def get_workspace_id(self, force: bool = False) -> str:
         """
         Get the workspace ID (GUID or name without spaces).
         Use this when passing workspace parameter to Python functions.
         
+        Args:
+            force: If True, always resolve to actual GUID via API. If False, returns stored value (default: False)
+        
         Returns:
             Workspace ID - either a GUID or workspace name without spaces
         """
+        if not force:
+            return self.workspace_id
+        
+        # Force resolution to GUID
+        import re
+        guid_pattern = re.compile(r'^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$', re.IGNORECASE)
+        
+        # If already a GUID, return it
+        if guid_pattern.match(self.workspace_id):
+            return self.workspace_id
+        
+        # Try to get from notebook context first (fastest)
+        try:
+            import notebookutils  # type: ignore
+            workspace_guid = notebookutils.runtime.context.get("workspaceId")
+            if workspace_guid:
+                return workspace_guid
+        except ImportError:
+            pass
+        
+        # Resolve via API
+        try:
+            from .auth import get_fabric_api_token
+            token = get_fabric_api_token()
+            if token:
+                resolved_id = self._resolve_workspace_id_by_name(token, self.workspace_id)
+                if resolved_id:
+                    return resolved_id
+        except Exception:
+            pass
+        
+        # Fallback to original value
         return self.workspace_id
     
-    def get_lakehouse_id(self) -> str:
+    def get_lakehouse_id(self, force: bool = False) -> str:
         """
         Get the lakehouse ID (GUID or name).
         Use this when passing lakehouse parameter to Python functions.
         
+        Args:
+            force: If True, always resolve to actual GUID via API. If False, returns stored value (default: False)
+        
         Returns:
             Lakehouse ID - either a GUID or lakehouse name
         """
+        if not force:
+            return self.lakehouse_id
+        
+        # Force resolution to GUID
+        import re
+        guid_pattern = re.compile(r'^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$', re.IGNORECASE)
+        
+        # If already a GUID, return it
+        if guid_pattern.match(self.lakehouse_id):
+            return self.lakehouse_id
+        
+        # Try to get from notebook context first (fastest)
+        try:
+            import notebookutils  # type: ignore
+            lakehouse_guid = notebookutils.lakehouse.get("id")
+            if lakehouse_guid:
+                return lakehouse_guid
+        except (ImportError, Exception):
+            pass
+        
+        # Resolve via API
+        try:
+            from .auth import get_fabric_api_token
+            token = get_fabric_api_token()
+            if token:
+                # First get workspace GUID
+                workspace_guid = self.get_workspace_id(force=True)
+                # Then resolve lakehouse name to ID
+                resolved_id = self._resolve_lakehouse_id_by_name(token, workspace_guid, self.lakehouse_id)
+                if resolved_id:
+                    return resolved_id
+        except Exception:
+            pass
+        
+        # Fallback to original value
         return self.lakehouse_id
 
     def run(self, pipeline: List[Tuple]) -> bool:

{duckrun-0.2.14.dev1.dist-info → duckrun-0.2.14.dev2.dist-info}/METADATA

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

{duckrun-0.2.14.dev1.dist-info → duckrun-0.2.14.dev2.dist-info}/RECORD

@@ -1,14 +1,14 @@
-duckrun/__init__.py,sha256=OxPnNrxYqv_2XfiQPO27JiZDIxze4jnUE3VaqDdioAg,236
+duckrun/__init__.py,sha256=oPQXpJEgHpX_KgMrx_TWax9awIbr2B9z32cFuuG_p30,236
 duckrun/auth.py,sha256=dMqIzozgEQ5v7Uc3Mb_OoFZGmsAq0m-VOoYCVL7rehc,9281
-duckrun/core.py,sha256=LvxplwziTLb_18n064waoN3oWMuhpVJe_-y6GYfoBOc,53127
+duckrun/core.py,sha256=Rbfm0VtN3o-tMg1_iH5yCtgBFOAXbp6GeExgtxg4MX0,55912
 duckrun/files.py,sha256=Fvdjg3DyHJzIVzKo8M_j-eGz4zU61lOB38Y_onbQJkI,10137
 duckrun/lakehouse.py,sha256=j--Z3zo8AOWt1GF9VzRosmmTAy6ey2D0LVubti58twU,14109
 duckrun/runner.py,sha256=yrDxfy1RVkb8iK9GKGmIFZHzCvcO_0GVQlbng7Vw_iM,14171
 duckrun/semantic_model.py,sha256=obzlN2-dbEW3JmDop-vrZGGGLi9u3ThhTbgtDjou7uY,29509
 duckrun/stats.py,sha256=oKIjZ7u5cFVT63FuOl5UqoDsOG3098woSCn-uI6i_sQ,11084
 duckrun/writer.py,sha256=svUuPCYOhrz299NgnpTKhARKjfej0PxnoND2iPDSypk,8098
-duckrun-0.2.14.dev1.dist-info/licenses/LICENSE,sha256=-DeQQwdbCbkB4507ZF3QbocysB-EIjDtaLexvqRkGZc,1083
-duckrun-0.2.14.dev1.dist-info/METADATA,sha256=MhpAtTMLpzOwOINN7Dgs6ih_JhjhbzxX73W_E6N30pA,20771
-duckrun-0.2.14.dev1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-duckrun-0.2.14.dev1.dist-info/top_level.txt,sha256=BknMEwebbUHrVAp3SC92ps8MPhK7XSYsaogTvi_DmEU,8
-duckrun-0.2.14.dev1.dist-info/RECORD,,
+duckrun-0.2.14.dev2.dist-info/licenses/LICENSE,sha256=-DeQQwdbCbkB4507ZF3QbocysB-EIjDtaLexvqRkGZc,1083
+duckrun-0.2.14.dev2.dist-info/METADATA,sha256=94ZCUDXEIK9BQRYhBoc1H1n_bZx5giXEulpazkK9FU4,20771
+duckrun-0.2.14.dev2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+duckrun-0.2.14.dev2.dist-info/top_level.txt,sha256=BknMEwebbUHrVAp3SC92ps8MPhK7XSYsaogTvi_DmEU,8
+duckrun-0.2.14.dev2.dist-info/RECORD,,