msfabricpysdkcore 0.0.9__py3-none-any.whl → 0.0.11__py3-none-any.whl

msfabricpysdkcore/admin_item.py

@@ -83,6 +83,13 @@ class AdminItem:
     def get_item_access_details(self, type=None):
         """Get the access details of the item
         
+        Returns:
+            dict: The access details of the item"""
+        return self.list_item_access_details(type)
+        
+    def list_item_access_details(self, type=None):
+        """Get the access details of the item
+        
         Returns:
             dict: The access details of the item"""
         

msfabricpysdkcore/admin_workspace.py

@@ -70,6 +70,14 @@ class AdminWorkspace:
     def get_workspace_access_details(self):
         """Get the access details of the workspace
 
+        Returns:
+            dict: The access details of the workspace
+        """
+        return self.list_workspace_access_details()
+    
+    def list_workspace_access_details(self):
+        """Get the access details of the workspace
+
         Returns:
             dict: The access details of the workspace
         """
@@ -115,6 +123,17 @@ class AdminWorkspace:
         item_dict = json.loads(response.text) 
         return AdminItem.from_dict(item_dict, self.auth)
     
+    def list_item_access_details(self, item_id, type=None):
+        """Get the access details of the item
+        
+        Args:
+            item_id (str): The ID of the item
+            type (str): The type of the item
+        Returns:
+            dict: The access details of the item
+        """
+        return self.get_item(item_id, type).list_item_access_details()
+
     def get_item_access_details(self, item_id, type=None):
         """Get the access details of the item
         
@@ -124,4 +143,4 @@ class AdminWorkspace:
         Returns:
             dict: The access details of the item
         """
-        return self.get_item(item_id, type).get_item_access_details()
+        return self.list_item_access_details(item_id, type)

msfabricpysdkcore/adminapi.py

@@ -284,13 +284,24 @@ class FabricClientAdmin(FabricClient):
     def get_workspace_access_details(self, workspace_id):
         """Get the access details of the workspace
         
+        Args:
+            workspace_id (str): The ID of the workspace
+        Returns:
+            dict: The access details of the workspace
+        """
+        print("DEPRECATED: Use list_workspace_access_details instead")
+        return self.list_workspace_access_details(workspace_id)
+
+    def list_workspace_access_details(self, workspace_id):
+        """Get the access details of the workspace
+        
         Args:
             workspace_id (str): The ID of the workspace
         Returns:
             dict: The access details of the workspace
         """
         ws = self.get_workspace(workspace_id)
-        return ws.get_workspace_access_details()
+        return ws.list_workspace_access_details()
     
     def list_workspaces(self, capacity_id = None, name=None, state=None, type=None, continuationToken = None):
         """List all workspaces
@@ -426,6 +437,15 @@ class FabricClientAdmin(FabricClient):
     def get_tenant_settings(self):
         """Get the tenant settings
         
+        Returns:
+            dict: The tenant settings
+        """
+        print("DEPRECATED: Use list_tenant_settings instead")
+        return self.list_tenant_settings()
+    
+    def list_tenant_settings(self):
+        """Get the tenant settings
+        
         Returns:
             dict: The tenant settings
         """
@@ -445,7 +465,7 @@ class FabricClientAdmin(FabricClient):
         return json.loads(response.text)
     
 
-    def get_capacities_tenant_settings_overrides(self, continuationToken = None):
+    def list_capacities_tenant_settings_overrides(self, continuationToken = None):
         """Returns list of tenant setting overrides that override at the capacities
         
         Returns:
@@ -471,12 +491,35 @@ class FabricClientAdmin(FabricClient):
         overrides = resp_dict["Overrides"]
 
         if "continuationToken" in resp_dict and resp_dict["continuationToken"] is not None:
-            overrides_next = self.get_capacities_tenant_settings_overrides(continuationToken=resp_dict["continuationToken"])
+            overrides_next = self.list_capacities_tenant_settings_overrides(continuationToken=resp_dict["continuationToken"])
             overrides.extend(overrides_next)
 
         return overrides
+
+    def get_capacities_tenant_settings_overrides(self):
+        """Returns list of tenant setting overrides that override at the capacities
+        
+        Returns:
+            list: The capacities tenant settings overrides
+        """
+        print("DEPRECATED: Use list_capacities_tenant_settings_overrides instead")
+        return self.list_capacities_tenant_settings_overrides()
     
-    def get_access_entities(self, user_id, type = None, continuationToken = None):
+
+    def get_access_entities(self, user_id, type = None):
+        """Get the access entities for a user
+        
+        Args:
+            user_id (str): The ID of the user
+            type (str): The type of the access entity
+            continuationToken (str): The continuation token
+        Returns:
+            list: The list of access entities
+        """
+        print("DEPRECATED: Use list_access_entities instead")
+        return self.list_access_entities(user_id, type)
+
+    def list_access_entities(self, user_id, type = None, continuationToken = None):
         """Get the access entities for a user
         
         Args:
@@ -515,12 +558,12 @@ class FabricClientAdmin(FabricClient):
         access_entities = resp_dict["accessEntities"]
 
         if "continuationToken" in resp_dict and resp_dict["continuationToken"] is not None:
-            access_entities_next = self.get_access_entities(user_id, type, continuationToken=resp_dict["continuationToken"])
+            access_entities_next = self.list_access_entities(user_id, type, continuationToken=resp_dict["continuationToken"])
             resp_dict["accessEntities"].extend(access_entities_next)
         
         return access_entities
     
-    def get_item_access_details(self, workspace_id, item_id, type=None):
+    def list_item_access_details(self, workspace_id, item_id, type=None):
         """Get the access details of the item
         
         Args:
@@ -532,4 +575,87 @@ class FabricClientAdmin(FabricClient):
         """
         ws = self.get_workspace(workspace_id)
         item = ws.get_item(item_id, type)
-        return item.get_item_access_details(type)
+        return item.list_item_access_details(type)
+    
+    def get_item_access_details(self, workspace_id, item_id, type=None):
+        """Get the access details of the item
+        
+        Args:
+            workspace_id (str): The ID of the workspace
+            item_id (str): The ID of the item
+            type (str): The type of the item
+        Returns:
+            dict: The access details of the item
+        """
+        print("DEPRECATED: Use list_item_access_details instead")
+        return self.list_item_access_details(workspace_id, item_id, type)
+    
+    def bulk_set_labels(self, items, label_id, assignment_method = None, delegated_principal = None):
+        """Set labels in bulk"""
+        # POST https://api.fabric.microsoft.com/v1/admin/items/bulkSetLabels
+
+        url = "https://api.fabric.microsoft.com/v1/admin/items/bulkSetLabels"
+
+        if len(items) > 2000:
+            self.bulk_set_labels(items[2000:], label_id, assignment_method, delegated_principal)
+            items = items[:2000]
+
+        body = {
+            "items": items,
+            "labelId": label_id
+        }
+        if assignment_method:
+            body["assignmentMethod"] = assignment_method
+
+        if delegated_principal:
+            body["delegatedPrincipal"] = delegated_principal
+
+        for _ in range(10):
+            response = requests.post(url=url, headers=self.auth.get_headers(), json=body)
+            if response.status_code == 429:
+                print("Too many requests, waiting 10 seconds")
+                sleep(10)
+                continue
+            if response.status_code not in (200, 429):
+                print(response.status_code)
+                print(response.text)
+                raise Exception(f"Error setting labels: {response.text}")
+            break
+
+        response = json.loads(response.text)
+        return response
+    
+
+    def bulk_remove_labels(self, items):
+        """Remove labels in bulk
+        Args:
+            items (list): The list of item IDs
+            
+        Returns:
+            dict: The response from the API"""
+        # POST https://api.fabric.microsoft.com/v1/admin/items/bulkRemoveLabels
+
+        url = "https://api.fabric.microsoft.com/v1/admin/items/bulkRemoveLabels"
+
+        if len(items) > 2000:
+            self.bulk_remove_labels(items[2000:])
+            items = items[:2000]
+        
+        body = {
+            "items": items
+        }
+
+        for _ in range(10):
+            response = requests.post(url=url, headers=self.auth.get_headers(), json=body)
+            if response.status_code == 429:
+                print("Too many requests, waiting 10 seconds")
+                sleep(10)
+                continue
+            if response.status_code not in (200, 429):
+                print(response.status_code)
+                print(response.text)
+                raise Exception(f"Error removing labels: {response.text}")
+            break
+
+        response = json.loads(response.text)
+        return response

msfabricpysdkcore/auth.py

@@ -26,8 +26,9 @@ class FabricAuth():
 class FabricAuthClient(FabricAuth):
     """FabricAuthClient class to interact with Entra ID"""
 
-    def __init__(self):
-        print("Using Azure CLI for authentication")
+    def __init__(self, silent = False):
+        if not silent:
+            print("Using Azure CLI for authentication")
         self.auth = AzureCliCredential()
 
     def get_token(self):
@@ -38,8 +39,9 @@ class FabricAuthClient(FabricAuth):
 class FabricServicePrincipal(FabricAuth):
     """FabricServicePrincipal class to interact with Entra ID"""
 
-    def __init__(self, tenant_id, client_id, client_secret):
-        print("Using Service Principal for authentication")
+    def __init__(self, tenant_id, client_id, client_secret, silent = False):
+        if not silent:
+            print("Using Service Principal for authentication")
 
         self.tenant_id = tenant_id
         self.client_id = client_id
@@ -65,9 +67,10 @@ class FabricServicePrincipal(FabricAuth):
 class FabricSparkUtilsAuthentication(FabricAuth):
     """FabricSparkUtilsAuthentication class to interact with Entra ID"""
 
-    def __init__(self):
+    def __init__(self, silent = False):
         mssparkutils.credentials.getToken("pbi")
-        print("Using Synapse Spark Utils for authentication")
+        if not silent:
+            print("Using Synapse Spark Utils for authentication")
 
     def get_token(self):
         """Get token from Azure AD"""

msfabricpysdkcore/client.py

@@ -6,7 +6,7 @@ from msfabricpysdkcore.auth import FabricAuthClient, FabricServicePrincipal, Fab
 class FabricClient():
     """FabricClient class to interact with Fabric API"""
 
-    def __init__(self, tenant_id = None, client_id = None, client_secret = None) -> None:
+    def __init__(self, tenant_id = None, client_id = None, client_secret = None, silent=False) -> None:
         """Initialize FabricClient object"""
         self.tenant_id = tenant_id if tenant_id else os.getenv("FABRIC_TENANT_ID")
         self.client_id = client_id if client_id else os.getenv("FABRIC_CLIENT_ID")
@@ -16,10 +16,11 @@ class FabricClient():
 
         if self.client_id is None or self.client_secret is None or self.tenant_id is None:
             try:
-                self.auth = FabricSparkUtilsAuthentication()
+                self.auth = FabricSparkUtilsAuthentication(silent=silent)
             except:
-                self.auth = FabricAuthClient()
+                self.auth = FabricAuthClient(silent=silent)
         else:
             self.auth = FabricServicePrincipal(tenant_id = self.tenant_id,
                                                client_id = self.client_id, 
-                                               client_secret = self.client_secret)
+                                               client_secret = self.client_secret,
+                                               silent=silent)