duckrun 0.2.10__tar.gz → 0.2.10.dev1__tar.gz

{duckrun-0.2.10 → duckrun-0.2.10.dev1}/PKG-INFO

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

{duckrun-0.2.10 → duckrun-0.2.10.dev1}/duckrun/core.py

@@ -94,6 +94,7 @@ class Duckrun:
             pass  # Not in Colab, use default transport
         
         self._attach_lakehouse()
+        self._register_lookup_functions()
 
     @classmethod
     def connect(cls, connection_string: str, sql_folder: Optional[str] = None,
@@ -457,6 +458,151 @@ class Duckrun:
             print(f"❌ Error attaching lakehouse: {e}")
             print("Continuing without pre-attached tables.")
 
+    def _register_lookup_functions(self):
+        """
+        Register Fabric API lookup functions as DuckDB UDFs.
+        Allows SQL queries to resolve workspace/lakehouse names from IDs and vice versa.
+        
+        Functions registered:
+        - get_workspace_name(workspace_id) -> str
+        - get_lakehouse_name(workspace_id, lakehouse_id) -> str
+        - get_workspace_id_from_name(workspace_name) -> str
+        - get_lakehouse_id_from_name(workspace_id, lakehouse_name) -> str
+        """
+        # Cache to avoid repeated API calls
+        self._name_cache = {
+            'workspace_id_to_name': {},
+            'workspace_name_to_id': {},
+            'lakehouse_id_to_name': {},
+            'lakehouse_name_to_id': {}
+        }
+        
+        def get_workspace_name(workspace_id: str) -> str:
+            """Get workspace display name from workspace ID (GUID)"""
+            if workspace_id in self._name_cache['workspace_id_to_name']:
+                return self._name_cache['workspace_id_to_name'][workspace_id]
+            
+            try:
+                from .auth import get_fabric_api_token
+                token = get_fabric_api_token()
+                if not token:
+                    return None
+                
+                url = "https://api.fabric.microsoft.com/v1/workspaces"
+                headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
+                response = requests.get(url, headers=headers)
+                response.raise_for_status()
+                
+                workspaces = response.json().get("value", [])
+                for workspace in workspaces:
+                    if workspace.get("id") == workspace_id:
+                        name = workspace.get("displayName", "")
+                        self._name_cache['workspace_id_to_name'][workspace_id] = name
+                        self._name_cache['workspace_name_to_id'][name] = workspace_id
+                        return name
+                
+                return None
+            except Exception as e:
+                return None
+        
+        def get_lakehouse_name(workspace_id: str, lakehouse_id: str) -> str:
+            """Get lakehouse display name from workspace ID and lakehouse ID (GUIDs)"""
+            cache_key = f"{workspace_id}/{lakehouse_id}"
+            if cache_key in self._name_cache['lakehouse_id_to_name']:
+                return self._name_cache['lakehouse_id_to_name'][cache_key]
+            
+            try:
+                from .auth import get_fabric_api_token
+                token = get_fabric_api_token()
+                if not token:
+                    return None
+                
+                url = f"https://api.fabric.microsoft.com/v1/workspaces/{workspace_id}/lakehouses"
+                headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
+                response = requests.get(url, headers=headers)
+                response.raise_for_status()
+                
+                lakehouses = response.json().get("value", [])
+                for lakehouse in lakehouses:
+                    if lakehouse.get("id") == lakehouse_id:
+                        name = lakehouse.get("displayName", "")
+                        self._name_cache['lakehouse_id_to_name'][cache_key] = name
+                        lh_cache_key = f"{workspace_id}/{name}"
+                        self._name_cache['lakehouse_name_to_id'][lh_cache_key] = lakehouse_id
+                        return name
+                
+                return None
+            except Exception as e:
+                return None
+        
+        def get_workspace_id_from_name(workspace_name: str) -> str:
+            """Get workspace ID (GUID) from workspace display name"""
+            if workspace_name in self._name_cache['workspace_name_to_id']:
+                return self._name_cache['workspace_name_to_id'][workspace_name]
+            
+            try:
+                from .auth import get_fabric_api_token
+                token = get_fabric_api_token()
+                if not token:
+                    return None
+                
+                url = "https://api.fabric.microsoft.com/v1/workspaces"
+                headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
+                response = requests.get(url, headers=headers)
+                response.raise_for_status()
+                
+                workspaces = response.json().get("value", [])
+                for workspace in workspaces:
+                    if workspace.get("displayName") == workspace_name:
+                        workspace_id = workspace.get("id", "")
+                        self._name_cache['workspace_name_to_id'][workspace_name] = workspace_id
+                        self._name_cache['workspace_id_to_name'][workspace_id] = workspace_name
+                        return workspace_id
+                
+                return None
+            except Exception as e:
+                return None
+        
+        def get_lakehouse_id_from_name(workspace_id: str, lakehouse_name: str) -> str:
+            """Get lakehouse ID (GUID) from workspace ID and lakehouse display name"""
+            cache_key = f"{workspace_id}/{lakehouse_name}"
+            if cache_key in self._name_cache['lakehouse_name_to_id']:
+                return self._name_cache['lakehouse_name_to_id'][cache_key]
+            
+            try:
+                from .auth import get_fabric_api_token
+                token = get_fabric_api_token()
+                if not token:
+                    return None
+                
+                url = f"https://api.fabric.microsoft.com/v1/workspaces/{workspace_id}/lakehouses"
+                headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
+                response = requests.get(url, headers=headers)
+                response.raise_for_status()
+                
+                lakehouses = response.json().get("value", [])
+                for lakehouse in lakehouses:
+                    if lakehouse.get("displayName") == lakehouse_name:
+                        lakehouse_id = lakehouse.get("id", "")
+                        self._name_cache['lakehouse_name_to_id'][cache_key] = lakehouse_id
+                        id_cache_key = f"{workspace_id}/{lakehouse_id}"
+                        self._name_cache['lakehouse_id_to_name'][id_cache_key] = lakehouse_name
+                        return lakehouse_id
+                
+                return None
+            except Exception as e:
+                return None
+        
+        # Register functions in DuckDB
+        try:
+            self.con.create_function("get_workspace_name", get_workspace_name)
+            self.con.create_function("get_lakehouse_name", get_lakehouse_name)
+            self.con.create_function("get_workspace_id_from_name", get_workspace_id_from_name)
+            self.con.create_function("get_lakehouse_id_from_name", get_lakehouse_id_from_name)
+            print("✅ Registered lookup functions: get_workspace_name, get_lakehouse_name, get_workspace_id_from_name, get_lakehouse_id_from_name")
+        except Exception as e:
+            print(f"⚠️  Warning: Could not register lookup functions: {e}")
+
     def get_workspace_id(self) -> str:
         """
         Get the workspace ID (GUID or name without spaces).

{duckrun-0.2.10 → duckrun-0.2.10.dev1}/duckrun.egg-info/PKG-INFO

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

{duckrun-0.2.10 → duckrun-0.2.10.dev1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "duckrun"
-version = "0.2.10"
+version = "0.2.10.dev1"
 description = "Lakehouse task runner powered by DuckDB for Microsoft Fabric"
 readme = "README.md"
 license = {text = "MIT"}