duckrun 0.2.10__py3-none-any.whl → 0.2.10.dev0__py3-none-any.whl

duckrun/core.py

@@ -456,6 +456,154 @@ class Duckrun:
         except Exception as e:
             print(f"❌ Error attaching lakehouse: {e}")
             print("Continuing without pre-attached tables.")
+        
+        # Register lookup functions as DuckDB UDFs
+        self._register_lookup_functions()
+
+    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 f"[Error: No auth token]"
+                
+                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 f"[Workspace {workspace_id} not found]"
+            except Exception as e:
+                return f"[Error: {str(e)}]"
+        
+        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 f"[Error: No auth token]"
+                
+                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 f"[Lakehouse {lakehouse_id} not found]"
+            except Exception as e:
+                return f"[Error: {str(e)}]"
+        
+        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 f"[Error: No auth token]"
+                
+                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 f"[Workspace '{workspace_name}' not found]"
+            except Exception as e:
+                return f"[Error: {str(e)}]"
+        
+        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 f"[Error: No auth token]"
+                
+                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 f"[Lakehouse '{lakehouse_name}' not found]"
+            except Exception as e:
+                return f"[Error: {str(e)}]"
+        
+        # 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:
         """

{duckrun-0.2.10.dist-info → duckrun-0.2.10.dev0.dist-info}/METADATA

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

{duckrun-0.2.10.dist-info → duckrun-0.2.10.dev0.dist-info}/RECORD

@@ -1,14 +1,14 @@
 duckrun/__init__.py,sha256=cTj6KQ6hKmgu1z7k9nhDcO5lct049luxjx1V0QnymCo,235
 duckrun/auth.py,sha256=qPaLQ7InlV9leA9r6E6VEeYavFFoBi0zSN8m_l1aoQs,9545
-duckrun/core.py,sha256=g9WtvhROxFSo2Idb979fY5HhxbMm_x-tajc_zWMtqCU,46853
+duckrun/core.py,sha256=imA1Xt3_Qz1PZFzrcv6jbvsfiyxJiYax2iZ14czgrVo,54637
 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.10.dist-info/licenses/LICENSE,sha256=-DeQQwdbCbkB4507ZF3QbocysB-EIjDtaLexvqRkGZc,1083
-duckrun-0.2.10.dist-info/METADATA,sha256=CwDyjJqyfBoISxZ1bfdojVVsP0HcrLylgqCTpMsC6e8,20624
-duckrun-0.2.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-duckrun-0.2.10.dist-info/top_level.txt,sha256=BknMEwebbUHrVAp3SC92ps8MPhK7XSYsaogTvi_DmEU,8
-duckrun-0.2.10.dist-info/RECORD,,
+duckrun-0.2.10.dev0.dist-info/licenses/LICENSE,sha256=-DeQQwdbCbkB4507ZF3QbocysB-EIjDtaLexvqRkGZc,1083
+duckrun-0.2.10.dev0.dist-info/METADATA,sha256=O6fyWc9A94TlphbmbMEpyrow4iA6ZUrBbs0KbFgmfNE,20629
+duckrun-0.2.10.dev0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+duckrun-0.2.10.dev0.dist-info/top_level.txt,sha256=BknMEwebbUHrVAp3SC92ps8MPhK7XSYsaogTvi_DmEU,8
+duckrun-0.2.10.dev0.dist-info/RECORD,,