duckrun 0.2.9.dev4__tar.gz → 0.2.9.dev5__tar.gz

{duckrun-0.2.9.dev4 → duckrun-0.2.9.dev5}/PKG-INFO

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

{duckrun-0.2.9.dev4 → duckrun-0.2.9.dev5}/duckrun/__init__.py

@@ -2,7 +2,7 @@
 
 from duckrun.core import Duckrun
 
-__version__ = "0.2.9.dev4"
+__version__ = "0.2.9.dev5"
 
 # Expose unified connect method at module level
 connect = Duckrun.connect

{duckrun-0.2.9.dev4 → duckrun-0.2.9.dev5}/duckrun/semantic_model.py

@@ -130,7 +130,7 @@ def check_dataset_exists(dataset_name, workspace_id, client):
 
 
 def refresh_dataset(dataset_name, workspace_id, client, dataset_id=None):
-    """Refresh a dataset and monitor progress"""
+    """Refresh a dataset and monitor progress using Power BI API"""
     
     # If dataset_id not provided, look it up by name
     if not dataset_id:
@@ -144,39 +144,46 @@ def refresh_dataset(dataset_name, workspace_id, client, dataset_id=None):
         "objects": []
     }
     
-    response = client.post(
-        f"/v1/workspaces/{workspace_id}/semanticModels/{dataset_id}/refreshes",
-        json=payload
-    )
+    # Use Power BI API for refresh (not Fabric API)
+    powerbi_url = f"https://api.powerbi.com/v1.0/myorg/datasets/{dataset_id}/refreshes"
+    headers = client._get_headers()
+    
+    response = requests.post(powerbi_url, headers=headers, json=payload)
     
     if response.status_code in [200, 202]:
         print(f"✓ Refresh initiated")
         
-        refresh_id = response.json().get('id')
-        if refresh_id:
-            print("   Monitoring refresh progress...")
-            max_attempts = 60
-            for attempt in range(max_attempts):
-                time.sleep(5)
-                
-                status_response = client.get(
-                    f"/v1/workspaces/{workspace_id}/semanticModels/{dataset_id}/refreshes/{refresh_id}"
-                )
-                status = status_response.json().get('status')
+        # For 202, get the refresh_id from the Location header
+        if response.status_code == 202:
+            location = response.headers.get('Location')
+            if location:
+                refresh_id = location.split('/')[-1]
+                print("   Monitoring refresh progress...")
+                max_attempts = 60
+                for attempt in range(max_attempts):
+                    time.sleep(5)
+                    
+                    # Check refresh status using Power BI API
+                    status_url = f"https://api.powerbi.com/v1.0/myorg/datasets/{dataset_id}/refreshes/{refresh_id}"
+                    status_response = requests.get(status_url, headers=headers)
+                    status_response.raise_for_status()
+                    status = status_response.json().get('status')
+                    
+                    if status == 'Completed':
+                        print(f"✓ Refresh completed successfully")
+                        return
+                    elif status == 'Failed':
+                        error = status_response.json().get('serviceExceptionJson', '')
+                        raise Exception(f"Refresh failed: {error}")
+                    elif status == 'Cancelled':
+                        raise Exception("Refresh was cancelled")
+                    
+                    if attempt % 6 == 0:
+                        print(f"   Status: {status}...")
                 
-                if status == 'Completed':
-                    print(f"✓ Refresh completed successfully")
-                    return
-                elif status == 'Failed':
-                    error = status_response.json().get('error', {})
-                    raise Exception(f"Refresh failed: {error.get('message', 'Unknown error')}")
-                elif status == 'Cancelled':
-                    raise Exception("Refresh was cancelled")
-                
-                if attempt % 6 == 0:
-                    print(f"   Status: {status}...")
-            
-            raise Exception(f"Refresh timed out")
+                raise Exception(f"Refresh timed out")
+    else:
+        response.raise_for_status()
 
 
 def download_bim_from_github(url):

{duckrun-0.2.9.dev4 → duckrun-0.2.9.dev5}/duckrun.egg-info/PKG-INFO

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

{duckrun-0.2.9.dev4 → duckrun-0.2.9.dev5}/pyproject.toml

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