semantic-link-labs 0.5.0__py3-none-any.whl → 0.6.0__py3-none-any.whl

sempy_labs/_list_functions.py

@@ -1,17 +1,22 @@
-import sempy
 import sempy.fabric as fabric
 from sempy_labs._helper_functions import (
-    resolve_workspace_name_and_id, 
-    resolve_lakehouse_name, 
-    create_relationship_name, 
-    resolve_lakehouse_id)
+    resolve_workspace_name_and_id,
+    resolve_lakehouse_name,
+    create_relationship_name,
+    resolve_lakehouse_id,
+)
 import pandas as pd
-import json, time
+import json
+import time
 from pyspark.sql import SparkSession
 from typing import Optional
 import sempy_labs._icons as icons
+from sempy.fabric.exceptions import FabricHTTPException
 
-def get_object_level_security(dataset: str, workspace: Optional[str] = None) -> pd.DataFrame:
+
+def get_object_level_security(
+    dataset: str, workspace: Optional[str] = None
+) -> pd.DataFrame:
     """
     Shows the object level security for the semantic model.
 
@@ -32,12 +37,14 @@ def get_object_level_security(dataset: str, workspace: Optional[str] = None) ->
 
     from sempy_labs.tom import connect_semantic_model
 
-    if workspace is None:        
+    if workspace is None:
         workspace = fabric.resolve_workspace_name()
-    
+
     df = pd.DataFrame(columns=["Role Name", "Object Type", "Table Name", "Object Name"])
 
-    with connect_semantic_model(dataset=dataset, readonly=True, workspace=workspace) as tom:
+    with connect_semantic_model(
+        dataset=dataset, readonly=True, workspace=workspace
+    ) as tom:
 
         for r in tom.model.Roles:
             for tp in r.TablePermissions:
@@ -45,7 +52,7 @@ def get_object_level_security(dataset: str, workspace: Optional[str] = None) ->
                     columnCount = 0
                     try:
                         columnCount = len(tp.ColumnPermissions)
-                    except:
+                    except Exception:
                         pass
                     objectType = "Table"
                     if columnCount == 0:
@@ -68,7 +75,8 @@ def get_object_level_security(dataset: str, workspace: Optional[str] = None) ->
                                 "Object Name": cp.Name,
                             }
                             df = pd.concat(
-                                [df, pd.DataFrame(new_data, index=[0])], ignore_index=True
+                                [df, pd.DataFrame(new_data, index=[0])],
+                                ignore_index=True,
                             )
 
         return df
@@ -93,53 +101,20 @@ def list_tables(dataset: str, workspace: Optional[str] = None) -> pd.DataFrame:
         A pandas dataframe showing the semantic model's tables and their properties.
     """
 
-    from sempy_labs.tom import connect_semantic_model
-
-    if workspace is None:
-        workspace = fabric.resolve_workspace_name()
+    workspace = fabric.resolve_workspace_name()
 
-    df = pd.DataFrame(
-        columns=[
-            "Name",
-            "Type",
-            "Hidden",
-            "Data Category",
-            "Description",
-            "Refresh Policy",
-            "Source Expression",
-        ]
+    df = fabric.list_tables(
+        dataset=dataset,
+        workspace=workspace,
+        additional_xmla_properties=["RefreshPolicy", "RefreshPolicy.SourceExpression"],
     )
 
-    with connect_semantic_model(dataset=dataset, readonly=True, workspace=workspace) as tom:
-
-        import Microsoft.AnalysisServices.Tabular as TOM
-
-        for t in tom.model.Tables:
-            tableType = "Table"
-            rPolicy = bool(t.RefreshPolicy)
-            sourceExpression = None
-            if str(t.CalculationGroup) != "None":
-                tableType = "Calculation Group"
-            else:
-                for p in t.Partitions:
-                    if p.SourceType == TOM.PartitionSourceType.Calculated:
-                        tableType = "Calculated Table"
-
-            if rPolicy:
-                sourceExpression = t.RefreshPolicy.SourceExpression
-
-            new_data = {
-                "Name": t.Name,
-                "Type": tableType,
-                "Hidden": t.IsHidden,
-                "Data Category": t.DataCategory,
-                "Description": t.Description,
-                "Refresh Policy": rPolicy,
-                "Source Expression": sourceExpression,
-            }
-            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+    df["Refresh Policy"] = df["Refresh Policy"].notna()
+    df.rename(
+        columns={"Refresh Policy Source Expression": "Source Expression"}, inplace=True
+    )
 
-        return df
+    return df
 
 
 def list_annotations(dataset: str, workspace: Optional[str] = None) -> pd.DataFrame:
@@ -175,7 +150,9 @@ def list_annotations(dataset: str, workspace: Optional[str] = None) -> pd.DataFr
         ]
     )
 
-    with connect_semantic_model(dataset=dataset, readonly=True, workspace=workspace) as tom:
+    with connect_semantic_model(
+        dataset=dataset, readonly=True, workspace=workspace
+    ) as tom:
 
         mName = tom.model.Name
         for a in tom.model.Annotations:
@@ -203,7 +180,9 @@ def list_annotations(dataset: str, workspace: Optional[str] = None) -> pd.DataFr
                     "Annotation Name": taName,
                     "Annotation Value": taValue,
                 }
-                df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+                df = pd.concat(
+                    [df, pd.DataFrame(new_data, index=[0])], ignore_index=True
+                )
             for p in t.Partitions:
                 pName = p.Name
                 objectType = "Partition"
@@ -281,7 +260,9 @@ def list_annotations(dataset: str, workspace: Optional[str] = None) -> pd.DataFr
                     "Annotation Name": daName,
                     "Annotation Value": daValue,
                 }
-                df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+                df = pd.concat(
+                    [df, pd.DataFrame(new_data, index=[0])], ignore_index=True
+                )
         for r in tom.model.Relationships:
             rName = r.Name
             objectType = "Relationship"
@@ -295,7 +276,9 @@ def list_annotations(dataset: str, workspace: Optional[str] = None) -> pd.DataFr
                     "Annotation Name": raName,
                     "Annotation Value": raValue,
                 }
-                df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+                df = pd.concat(
+                    [df, pd.DataFrame(new_data, index=[0])], ignore_index=True
+                )
         for cul in tom.model.Cultures:
             culName = cul.Name
             objectType = "Translation"
@@ -309,7 +292,9 @@ def list_annotations(dataset: str, workspace: Optional[str] = None) -> pd.DataFr
                     "Annotation Name": culaName,
                     "Annotation Value": culaValue,
                 }
-                df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+                df = pd.concat(
+                    [df, pd.DataFrame(new_data, index=[0])], ignore_index=True
+                )
         for e in tom.model.Expressions:
             eName = e.Name
             objectType = "Expression"
@@ -323,7 +308,9 @@ def list_annotations(dataset: str, workspace: Optional[str] = None) -> pd.DataFr
                     "Annotation Name": eaName,
                     "Annotation Value": eaValue,
                 }
-                df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+                df = pd.concat(
+                    [df, pd.DataFrame(new_data, index=[0])], ignore_index=True
+                )
         for per in tom.model.Perspectives:
             perName = per.Name
             objectType = "Perspective"
@@ -337,7 +324,9 @@ def list_annotations(dataset: str, workspace: Optional[str] = None) -> pd.DataFr
                     "Annotation Name": peraName,
                     "Annotation Value": peraValue,
                 }
-                df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+                df = pd.concat(
+                    [df, pd.DataFrame(new_data, index=[0])], ignore_index=True
+                )
         for rol in tom.model.Roles:
             rolName = rol.Name
             objectType = "Role"
@@ -351,7 +340,9 @@ def list_annotations(dataset: str, workspace: Optional[str] = None) -> pd.DataFr
                     "Annotation Name": rolaName,
                     "Annotation Value": rolaValue,
                 }
-                df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+                df = pd.concat(
+                    [df, pd.DataFrame(new_data, index=[0])], ignore_index=True
+                )
 
         return df
 
@@ -550,9 +541,9 @@ def list_lakehouses(workspace: Optional[str] = None) -> pd.DataFrame:
     client = fabric.FabricRestClient()
     response = client.get(f"/v1/workspaces/{workspace_id}/lakehouses/")
 
-    for v in response.json()["value"]:        
-        prop = v.get("properties",{})
-        sqlEPProp = prop.get("sqlEndpointProperties",{})
+    for v in response.json()["value"]:
+        prop = v.get("properties", {})
+        sqlEPProp = prop.get("sqlEndpointProperties", {})
 
         new_data = {
             "Lakehouse Name": v.get("displayName"),
@@ -602,8 +593,8 @@ def list_warehouses(workspace: Optional[str] = None) -> pd.DataFrame:
     client = fabric.FabricRestClient()
     response = client.get(f"/v1/workspaces/{workspace_id}/warehouses/")
 
-    for v in response.json()["value"]:        
-        prop = v.get("properties",{})
+    for v in response.json()["value"]:
+        prop = v.get("properties", {})
 
         new_data = {
             "Warehouse Name": v.get("displayName"),
@@ -680,7 +671,7 @@ def list_mirroredwarehouses(workspace: Optional[str] = None) -> pd.DataFrame:
     client = fabric.FabricRestClient()
     response = client.get(f"/v1/workspaces/{workspace_id}/mirroredWarehouses/")
 
-    for v in response.json()["value"]:        
+    for v in response.json()["value"]:
 
         new_data = {
             "Mirrored Warehouse": v.get("displayName"),
@@ -726,8 +717,8 @@ def list_kqldatabases(workspace: Optional[str] = None) -> pd.DataFrame:
     client = fabric.FabricRestClient()
     response = client.get(f"/v1/workspaces/{workspace_id}/kqlDatabases/")
 
-    for v in response.json()["value"]:        
-        prop = v.get("properties",{})
+    for v in response.json()["value"]:
+        prop = v.get("properties", {})
 
         new_data = {
             "KQL Database Name": v.get("displayName"),
@@ -1019,7 +1010,10 @@ def create_warehouse(
             f"{icons.green_dot} The '{warehouse}' warehouse has been created within the '{workspace}' workspace."
         )
     else:
-        raise ValueError(f"{icons.red_dot} Failed to create the '{warehouse}' warehouse within the '{workspace}' workspace.")
+        raise ValueError(
+            f"{icons.red_dot} Failed to create the '{warehouse}' warehouse within the '{workspace}' workspace."
+        )
+
 
 def update_item(
     item_type: str,
@@ -1064,15 +1058,19 @@ def update_item(
     item_type = item_type.replace(" ", "").capitalize()
 
     if item_type not in itemTypes.keys():
-        raise ValueError(f"{icons.red_dot} The '{item_type}' is not a valid item type. ")
-        
+        raise ValueError(
+            f"{icons.red_dot} The '{item_type}' is not a valid item type. "
+        )
+
     itemType = itemTypes[item_type]
 
     dfI = fabric.list_items(workspace=workspace, type=item_type)
     dfI_filt = dfI[(dfI["Display Name"] == current_name)]
 
     if len(dfI_filt) == 0:
-        raise ValueError(f"{icons.red_dot} The '{current_name}' {item_type} does not exist within the '{workspace}' workspace.")
+        raise ValueError(
+            f"{icons.red_dot} The '{current_name}' {item_type} does not exist within the '{workspace}' workspace."
+        )
 
     itemId = dfI_filt["Id"].iloc[0]
 
@@ -1085,17 +1083,17 @@ def update_item(
         f"/v1/workspaces/{workspace_id}/{itemType}/{itemId}", json=request_body
     )
 
-    if response.status_code == 200:
-        if description is None:
-            print(
-                f"{icons.green_dot} The '{current_name}' {item_type} within the '{workspace}' workspace has been updated to be named '{new_name}'"
-            )
-        else:
-            print(
-                f"{icons.green_dot} The '{current_name}' {item_type} within the '{workspace}' workspace has been updated to be named '{new_name}' and have a description of '{description}'"
-            )
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+    if description is None:
+        print(
+            f"{icons.green_dot} The '{current_name}' {item_type} within the '{workspace}' workspace has been updated to be named '{new_name}'"
+        )
     else:
-        raise ValueError(f"{icons.red_dot}: The '{current_name}' {item_type} within the '{workspace}' workspace was not updateds.")
+        print(
+            f"{icons.green_dot} The '{current_name}' {item_type} within the '{workspace}' workspace has been updated to be named '{new_name}' and have a description of '{description}'"
+        )
+
 
 def list_relationships(
     dataset: str, workspace: Optional[str] = None, extended: Optional[bool] = False
@@ -1133,7 +1131,7 @@ def list_relationships(
             dax_string="""
                 SELECT
                 [ID] AS [RelationshipID]
-                ,[Name] 
+                ,[Name]
                 FROM $SYSTEM.TMSCHEMA_RELATIONSHIPS
                 """,
         )
@@ -1143,7 +1141,7 @@ def list_relationships(
             dataset=dataset,
             workspace=workspace,
             dax_string="""
-                SELECT 
+                SELECT
                 [TABLE_ID]
                 ,[USED_SIZE]
                 FROM $SYSTEM.DISCOVER_STORAGE_TABLE_COLUMN_SEGMENTS
@@ -1200,7 +1198,7 @@ def list_dataflow_storage_accounts() -> pd.DataFrame:
         ]
     )
     client = fabric.PowerBIRestClient()
-    response = client.get(f"/v1.0/myorg/dataflowStorageAccounts")
+    response = client.get("/v1.0/myorg/dataflowStorageAccounts")
 
     for v in response.json()["value"]:
 
@@ -1305,10 +1303,12 @@ def list_workspace_role_assignments(workspace: Optional[str] = None) -> pd.DataF
     response = client.get(f"/v1/workspaces/{workspace_id}/roleAssignments")
 
     for i in response.json()["value"]:
-        user_name = i.get("principal",{}).get("displayName")
+        user_name = i.get("principal", {}).get("displayName")
         role_name = i.get("role")
-        user_email = i.get("principal",{}).get("userDetails",{}).get("userPrincipalName")
-        user_type = i.get("principal",{}).get("type")
+        user_email = (
+            i.get("principal", {}).get("userDetails", {}).get("userPrincipalName")
+        )
+        user_type = i.get("principal", {}).get("type")
 
         new_data = {
             "User Name": user_name,
@@ -1320,7 +1320,10 @@ def list_workspace_role_assignments(workspace: Optional[str] = None) -> pd.DataF
 
     return df
 
-def list_semantic_model_objects(dataset: str, workspace: Optional[str] = None) -> pd.DataFrame:
+
+def list_semantic_model_objects(
+    dataset: str, workspace: Optional[str] = None
+) -> pd.DataFrame:
     """
     Shows a list of semantic model objects.
 
@@ -1420,11 +1423,11 @@ def list_semantic_model_objects(dataset: str, workspace: Optional[str] = None) -
                 df = pd.concat(
                     [df, pd.DataFrame(new_data, index=[0])], ignore_index=True
                 )
-                for l in h.Levels:
+                for lev in h.Levels:
                     new_data = {
-                        "Parent Name": l.Parent.Name,
-                        "Object Name": l.Name,
-                        "Object Type": str(l.ObjectType),
+                        "Parent Name": lev.Parent.Name,
+                        "Object Name": lev.Name,
+                        "Object Type": str(lev.ObjectType),
                     }
                     df = pd.concat(
                         [df, pd.DataFrame(new_data, index=[0])], ignore_index=True
@@ -1481,6 +1484,7 @@ def list_semantic_model_objects(dataset: str, workspace: Optional[str] = None) -
 
     return df
 
+
 def list_shortcuts(
     lakehouse: Optional[str] = None, workspace: Optional[str] = None
 ) -> pd.DataFrame:
@@ -1529,52 +1533,51 @@ def list_shortcuts(
     response = client.get(
         f"/v1/workspaces/{workspace_id}/items/{lakehouse_id}/shortcuts"
     )
-    if response.status_code == 200:
-        for s in response.json()["value"]:
-            shortcutName = s.get("name")
-            shortcutPath = s.get("path")
-            source = list(s["target"].keys())[0]
-            (
-                sourceLakehouseName,
-                sourceWorkspaceName,
-                sourcePath,
-                connectionId,
-                location,
-                subpath,
-            ) = (None, None, None, None, None, None)
-            if source == "oneLake":
-                sourceLakehouseId = s.get("target",{}).get(source,{}).get("itemId")
-                sourcePath = s.get("target",{}).get(source,{}).get("path")
-                sourceWorkspaceId = s.get("target",{}).get(source,{}).get("workspaceId")
-                sourceWorkspaceName = fabric.resolve_workspace_name(sourceWorkspaceId)
-                sourceLakehouseName = resolve_lakehouse_name(
-                    sourceLakehouseId, sourceWorkspaceName
-                )
-            else:
-                connectionId = s.get("target",{}).get(source,{}).get("connectionId")
-                location = s.get("target",{}).get(source,{}).get("location")
-                subpath = s.get("target",{}).get(source,{}).get("subpath")
 
-            new_data = {
-                "Shortcut Name": shortcutName,
-                "Shortcut Path": shortcutPath,
-                "Source": source,
-                "Source Lakehouse Name": sourceLakehouseName,
-                "Source Workspace Name": sourceWorkspaceName,
-                "Source Path": sourcePath,
-                "Source Connection ID": connectionId,
-                "Source Location": location,
-                "Source SubPath": subpath,
-            }
-            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+    for s in response.json()["value"]:
+        shortcutName = s.get("name")
+        shortcutPath = s.get("path")
+        source = list(s["target"].keys())[0]
+        (
+            sourceLakehouseName,
+            sourceWorkspaceName,
+            sourcePath,
+            connectionId,
+            location,
+            subpath,
+        ) = (None, None, None, None, None, None)
+        if source == "oneLake":
+            sourceLakehouseId = s.get("target", {}).get(source, {}).get("itemId")
+            sourcePath = s.get("target", {}).get(source, {}).get("path")
+            sourceWorkspaceId = s.get("target", {}).get(source, {}).get("workspaceId")
+            sourceWorkspaceName = fabric.resolve_workspace_name(sourceWorkspaceId)
+            sourceLakehouseName = resolve_lakehouse_name(
+                sourceLakehouseId, sourceWorkspaceName
+            )
+        else:
+            connectionId = s.get("target", {}).get(source, {}).get("connectionId")
+            location = s.get("target", {}).get(source, {}).get("location")
+            subpath = s.get("target", {}).get(source, {}).get("subpath")
+
+        new_data = {
+            "Shortcut Name": shortcutName,
+            "Shortcut Path": shortcutPath,
+            "Source": source,
+            "Source Lakehouse Name": sourceLakehouseName,
+            "Source Workspace Name": sourceWorkspaceName,
+            "Source Path": sourcePath,
+            "Source Connection ID": connectionId,
+            "Source Location": location,
+            "Source SubPath": subpath,
+        }
+        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
-    print(
-        f"{icons.warning} This function relies on an API which is not yet official as of May 21, 2024. Once the API becomes official this function will work as expected."
-    )
     return df
 
+
 def list_custom_pools(workspace: Optional[str] = None) -> pd.DataFrame:
-  
     """
     Lists all `custom pools <https://learn.microsoft.com/fabric/data-engineering/create-custom-spark-pools>`_ within a workspace.
 
@@ -1590,42 +1593,83 @@ def list_custom_pools(workspace: Optional[str] = None) -> pd.DataFrame:
     pandas.DataFrame
         A pandas dataframe showing all the custom pools within the Fabric workspace.
     """
-  
-    #https://learn.microsoft.com/rest/api/fabric/spark/custom-pools/list-workspace-custom-pools
+
+    # https://learn.microsoft.com/rest/api/fabric/spark/custom-pools/list-workspace-custom-pools
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
-    df = pd.DataFrame(columns=['Custom Pool ID', 'Custom Pool Name', 'Type', 'Node Family', 'Node Size', 'Auto Scale Enabled', 'Auto Scale Min Node Count', 'Auto Scale Max Node Count', 'Dynamic Executor Allocation Enabled', 'Dynamic Executor Allocation Min Executors', 'Dynamic Executor Allocation Max Executors'])
+    df = pd.DataFrame(
+        columns=[
+            "Custom Pool ID",
+            "Custom Pool Name",
+            "Type",
+            "Node Family",
+            "Node Size",
+            "Auto Scale Enabled",
+            "Auto Scale Min Node Count",
+            "Auto Scale Max Node Count",
+            "Dynamic Executor Allocation Enabled",
+            "Dynamic Executor Allocation Min Executors",
+            "Dynamic Executor Allocation Max Executors",
+        ]
+    )
 
     client = fabric.FabricRestClient()
     response = client.get(f"/v1/workspaces/{workspace_id}/spark/pools")
 
-    for i in response.json()['value']:
+    for i in response.json()["value"]:
 
-        aScale = i.get('autoScale',{})
-        d = i.get('dynamicExecutorAllocation',{})
+        aScale = i.get("autoScale", {})
+        d = i.get("dynamicExecutorAllocation", {})
 
-        new_data = {'Custom Pool ID': i.get('id'), 'Custom Pool Name': i.get('name'), 'Type': i.get('type'), 'Node Family': i.get('nodeFamily'), 'Node Size': i.get('nodeSize'), \
-        'Auto Scale Enabled': aScale.get('enabled'), 'Auto Scale Min Node Count': aScale.get('minNodeCount'), 'Auto Scale Max Node Count': aScale.get('maxNodeCount'), \
-        'Dynamic Executor Allocation Enabled': d.get('enabled'), 'Dynamic Executor Allocation Min Executors': d.get('minExecutors'), 'Dynamic Executor Allocation Max Executors': d.get('maxExecutors')}
+        new_data = {
+            "Custom Pool ID": i.get("id"),
+            "Custom Pool Name": i.get("name"),
+            "Type": i.get("type"),
+            "Node Family": i.get("nodeFamily"),
+            "Node Size": i.get("nodeSize"),
+            "Auto Scale Enabled": aScale.get("enabled"),
+            "Auto Scale Min Node Count": aScale.get("minNodeCount"),
+            "Auto Scale Max Node Count": aScale.get("maxNodeCount"),
+            "Dynamic Executor Allocation Enabled": d.get("enabled"),
+            "Dynamic Executor Allocation Min Executors": d.get("minExecutors"),
+            "Dynamic Executor Allocation Max Executors": d.get("maxExecutors"),
+        }
         df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
-    bool_cols = ['Auto Scale Enabled', 'Dynamic Executor Allocation Enabled']
-    int_cols = ['Auto Scale Min Node Count', 'Auto Scale Max Node Count', 'Dynamic Executor Allocation Enabled', 'Dynamic Executor Allocation Min Executors', 'Dynamic Executor Allocation Max Executors']
+    bool_cols = ["Auto Scale Enabled", "Dynamic Executor Allocation Enabled"]
+    int_cols = [
+        "Auto Scale Min Node Count",
+        "Auto Scale Max Node Count",
+        "Dynamic Executor Allocation Enabled",
+        "Dynamic Executor Allocation Min Executors",
+        "Dynamic Executor Allocation Max Executors",
+    ]
 
     df[bool_cols] = df[bool_cols].astype(bool)
     df[int_cols] = df[int_cols].astype(int)
 
     return df
 
-def create_custom_pool(pool_name: str, node_size: str, min_node_count: int, max_node_count: int, min_executors: int, max_executors: int, node_family: Optional[str] = 'MemoryOptimized', auto_scale_enabled: Optional[bool] = True, dynamic_executor_allocation_enabled: Optional[bool] = True, workspace: Optional[str] = None):
-  
+
+def create_custom_pool(
+    pool_name: str,
+    node_size: str,
+    min_node_count: int,
+    max_node_count: int,
+    min_executors: int,
+    max_executors: int,
+    node_family: Optional[str] = "MemoryOptimized",
+    auto_scale_enabled: Optional[bool] = True,
+    dynamic_executor_allocation_enabled: Optional[bool] = True,
+    workspace: Optional[str] = None,
+):
     """
     Creates a `custom pool <https://learn.microsoft.com/fabric/data-engineering/create-custom-spark-pools>`_ within a workspace.
 
     Parameters
     ----------
     pool_name : str
-        The custom pool name.    
+        The custom pool name.
     node_size : str
         The `node size <https://learn.microsoft.com/rest/api/fabric/spark/custom-pools/create-workspace-custom-pool?tabs=HTTP#nodesize>`_.
     min_node_count : int
@@ -1648,10 +1692,10 @@ def create_custom_pool(pool_name: str, node_size: str, min_node_count: int, max_
         or if no lakehouse attached, resolves to the workspace of the notebook.
 
     Returns
-    -------    
+    -------
     """
 
-    #https://learn.microsoft.com/en-us/rest/api/fabric/spark/custom-pools/create-workspace-custom-pool
+    # https://learn.microsoft.com/en-us/rest/api/fabric/spark/custom-pools/create-workspace-custom-pool
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     request_body = {
@@ -1659,34 +1703,49 @@ def create_custom_pool(pool_name: str, node_size: str, min_node_count: int, max_
         "nodeFamily": node_family,
         "nodeSize": node_size,
         "autoScale": {
-        "enabled": auto_scale_enabled,
-        "minNodeCount": min_node_count,
-        "maxNodeCount": max_node_count
+            "enabled": auto_scale_enabled,
+            "minNodeCount": min_node_count,
+            "maxNodeCount": max_node_count,
         },
         "dynamicExecutorAllocation": {
-        "enabled": dynamic_executor_allocation_enabled,
-        "minExecutors": min_executors,
-        "maxExecutors": max_executors
-        }
+            "enabled": dynamic_executor_allocation_enabled,
+            "minExecutors": min_executors,
+            "maxExecutors": max_executors,
+        },
     }
 
     client = fabric.FabricRestClient()
-    response = client.post(f"/v1/workspaces/{workspace_id}/spark/pools", json = request_body)
+    response = client.post(
+        f"/v1/workspaces/{workspace_id}/spark/pools", json=request_body
+    )
 
     if response.status_code == 201:
-        print(f"{icons.green_dot} The '{pool_name}' spark pool has been created within the '{workspace}' workspace.")
+        print(
+            f"{icons.green_dot} The '{pool_name}' spark pool has been created within the '{workspace}' workspace."
+        )
     else:
         raise ValueError(f"{icons.red_dot} {response.status_code}")
-  
-def update_custom_pool(pool_name: str, node_size: Optional[str] = None, min_node_count: Optional[int] = None, max_node_count: Optional[int] = None, min_executors: Optional[int] = None, max_executors: Optional[int] = None, node_family: Optional[str] = None, auto_scale_enabled: Optional[bool] = None, dynamic_executor_allocation_enabled: Optional[bool] = None, workspace: Optional[str] = None):
 
+
+def update_custom_pool(
+    pool_name: str,
+    node_size: Optional[str] = None,
+    min_node_count: Optional[int] = None,
+    max_node_count: Optional[int] = None,
+    min_executors: Optional[int] = None,
+    max_executors: Optional[int] = None,
+    node_family: Optional[str] = None,
+    auto_scale_enabled: Optional[bool] = None,
+    dynamic_executor_allocation_enabled: Optional[bool] = None,
+    workspace: Optional[str] = None,
+):
     """
     Updates the properties of a `custom pool <https://learn.microsoft.com/fabric/data-engineering/create-custom-spark-pools>`_ within a workspace.
 
     Parameters
     ----------
     pool_name : str
-        The custom pool name.    
+        The custom pool name.
     node_size : str, default=None
         The `node size <https://learn.microsoft.com/rest/api/fabric/spark/custom-pools/create-workspace-custom-pool?tabs=HTTP#nodesize>`_.
         Defaults to None which keeps the existing property setting.
@@ -1717,61 +1776,106 @@ def update_custom_pool(pool_name: str, node_size: Optional[str] = None, min_node
         or if no lakehouse attached, resolves to the workspace of the notebook.
 
     Returns
-    -------    
+    -------
     """
 
-    #https://learn.microsoft.com/en-us/rest/api/fabric/spark/custom-pools/update-workspace-custom-pool?tabs=HTTP
+    # https://learn.microsoft.com/en-us/rest/api/fabric/spark/custom-pools/update-workspace-custom-pool?tabs=HTTP
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
-    df = list_custom_pools(workspace = workspace)
-    df_pool = df[df['Custom Pool Name'] == pool_name]
+    df = list_custom_pools(workspace=workspace)
+    df_pool = df[df["Custom Pool Name"] == pool_name]
 
     if len(df_pool) == 0:
-        raise ValueError(f"{icons.red_dot} The '{pool_name}' custom pool does not exist within the '{workspace}'. Please choose a valid custom pool.")
+        raise ValueError(
+            f"{icons.red_dot} The '{pool_name}' custom pool does not exist within the '{workspace}'. Please choose a valid custom pool."
+        )
 
     if node_family is None:
-        node_family = df_pool['Node Family'].iloc[0]
+        node_family = df_pool["Node Family"].iloc[0]
     if node_size is None:
-        node_size = df_pool['Node Size'].iloc[0]
+        node_size = df_pool["Node Size"].iloc[0]
     if auto_scale_enabled is None:
-        auto_scale_enabled = bool(df_pool['Auto Scale Enabled'].iloc[0])
+        auto_scale_enabled = bool(df_pool["Auto Scale Enabled"].iloc[0])
     if min_node_count is None:
-        min_node_count = int(df_pool['Min Node Count'].iloc[0])
+        min_node_count = int(df_pool["Min Node Count"].iloc[0])
     if max_node_count is None:
-        max_node_count = int(df_pool['Max Node Count'].iloc[0])
+        max_node_count = int(df_pool["Max Node Count"].iloc[0])
     if dynamic_executor_allocation_enabled is None:
-        dynamic_executor_allocation_enabled = bool(df_pool['Dynami Executor Allocation Enabled'].iloc[0])
+        dynamic_executor_allocation_enabled = bool(
+            df_pool["Dynami Executor Allocation Enabled"].iloc[0]
+        )
     if min_executors is None:
-        min_executors = int(df_pool['Min Executors'].iloc[0])
+        min_executors = int(df_pool["Min Executors"].iloc[0])
     if max_executors is None:
-        max_executors = int(df_pool['Max Executors'].iloc[0])
+        max_executors = int(df_pool["Max Executors"].iloc[0])
 
     request_body = {
         "name": pool_name,
         "nodeFamily": node_family,
         "nodeSize": node_size,
         "autoScale": {
-        "enabled": auto_scale_enabled,
-        "minNodeCount": min_node_count,
-        "maxNodeCount": max_node_count
+            "enabled": auto_scale_enabled,
+            "minNodeCount": min_node_count,
+            "maxNodeCount": max_node_count,
         },
         "dynamicExecutorAllocation": {
-        "enabled": dynamic_executor_allocation_enabled,
-        "minExecutors": min_executors,
-        "maxExecutors": max_executors
-        }
+            "enabled": dynamic_executor_allocation_enabled,
+            "minExecutors": min_executors,
+            "maxExecutors": max_executors,
+        },
     }
 
     client = fabric.FabricRestClient()
-    response = client.post(f"/v1/workspaces/{workspace_id}/spark/pools", json = request_body)
+    response = client.post(
+        f"/v1/workspaces/{workspace_id}/spark/pools", json=request_body
+    )
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+    print(
+        f"{icons.green_dot} The '{pool_name}' spark pool within the '{workspace}' workspace has been updated."
+    )
+
+
+def delete_custom_pool(pool_name: str, workspace: Optional[str | None] = None):
+    """
+    Deletes a `custom pool <https://learn.microsoft.com/fabric/data-engineering/create-custom-spark-pools>`_ within a workspace.
+
+    Parameters
+    ----------
+    pool_name : str
+        The custom pool name.
+    workspace : str, default=None
+        The name of the Fabric workspace.
+        Defaults to None which resolves to the workspace of the attached lakehouse
+        or if no lakehouse attached, resolves to the workspace of the notebook.
+
+    Returns
+    -------
+    """
+
+    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+
+    dfL = list_custom_pools(workspace=workspace)
+    dfL_filt = dfL[dfL["Custom Pool Name"] == pool_name]
+
+    if len(dfL_filt) == 0:
+        raise ValueError(
+            f"{icons.red_dot} The '{pool_name}' custom pool does not exist within the '{workspace}' workspace."
+        )
+    poolId = dfL_filt["Custom Pool ID"].iloc[0]
+
+    client = fabric.FabricRestClient()
+    response = client.delete(f"/v1/workspaces/{workspace_id}/spark/pools/{poolId}")
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+    print(
+        f"{icons.green_dot} The '{pool_name}' spark pool has been deleted from the '{workspace}' workspace."
+    )
+
 
-    if response.status_code == 200:
-        print(f"{icons.green_dot} The '{pool_name}' spark pool within the '{workspace}' workspace has been updated.")
-    else:
-        raise ValueError(f"{icons.red_dot} {response.status_code}")
-    
 def assign_workspace_to_capacity(capacity_name: str, workspace: Optional[str] = None):
-  
     """
     Assigns a workspace to a capacity.
 
@@ -1788,27 +1892,28 @@ def assign_workspace_to_capacity(capacity_name: str, workspace: Optional[str] =
     -------
     """
 
-    #https://learn.microsoft.com/en-us/rest/api/fabric/core/workspaces/assign-to-capacity?tabs=HTTP
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     dfC = fabric.list_capacities()
-    dfC_filt = dfC[dfC['Name'] == capacity_name]
-    capacity_id = dfC_filt['Id'].iloc[0]
+    dfC_filt = dfC[dfC["Display Name"] == capacity_name]
+    capacity_id = dfC_filt["Id"].iloc[0]
 
-    request_body = {
-        "capacityId": capacity_id
-    }
+    request_body = {"capacityId": capacity_id}
 
     client = fabric.FabricRestClient()
-    response = client.post(f"/v1/workspaces/{workspace_id}/assignToCapacity", json = request_body)
+    response = client.post(
+        f"/v1/workspaces/{workspace_id}/assignToCapacity", json=request_body
+    )
 
     if response.status_code == 202:
-        print(f"{icons.green_dot} The '{workspace}' workspace has been assigned to the '{capacity_name}' capacity.")
+        print(
+            f"{icons.green_dot} The '{workspace}' workspace has been assigned to the '{capacity_name}' capacity."
+        )
     else:
         raise ValueError(f"{icons.red_dot} {response.status_code}")
 
+
 def unassign_workspace_from_capacity(workspace: Optional[str] = None):
-  
     """
     Unassigns a workspace from its assigned capacity.
 
@@ -1823,19 +1928,21 @@ def unassign_workspace_from_capacity(workspace: Optional[str] = None):
     -------
     """
 
-    #https://learn.microsoft.com/en-us/rest/api/fabric/core/workspaces/unassign-from-capacity?tabs=HTTP
+    # https://learn.microsoft.com/en-us/rest/api/fabric/core/workspaces/unassign-from-capacity?tabs=HTTP
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
-    
+
     client = fabric.FabricRestClient()
     response = client.post(f"/v1/workspaces/{workspace_id}/unassignFromCapacity")
 
     if response.status_code == 202:
-        print(f"{icons.green_dot} The '{workspace}' workspace has been unassigned from its capacity.")
+        print(
+            f"{icons.green_dot} The '{workspace}' workspace has been unassigned from its capacity."
+        )
     else:
         raise ValueError(f"{icons.red_dot} {response.status_code}")
-  
+
+
 def get_spark_settings(workspace: Optional[str] = None) -> pd.DataFrame:
-  
     """
     Shows the spark settings for a workspace.
 
@@ -1852,35 +1959,71 @@ def get_spark_settings(workspace: Optional[str] = None) -> pd.DataFrame:
         A pandas dataframe showing the spark settings for a workspace.
     """
 
-    #https://learn.microsoft.com/en-us/rest/api/fabric/spark/workspace-settings/get-spark-settings?tabs=HTTP
+    # https://learn.microsoft.com/en-us/rest/api/fabric/spark/workspace-settings/get-spark-settings?tabs=HTTP
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
-    df = pd.DataFrame(columns=['Automatic Log Enabled', 'High Concurrency Enabled', 'Customize Compute Enabled', 'Default Pool Name', 'Default Pool Type', 'Max Node Count', 'Max Executors', 'Environment Name', 'Runtime Version'])
+    df = pd.DataFrame(
+        columns=[
+            "Automatic Log Enabled",
+            "High Concurrency Enabled",
+            "Customize Compute Enabled",
+            "Default Pool Name",
+            "Default Pool Type",
+            "Max Node Count",
+            "Max Executors",
+            "Environment Name",
+            "Runtime Version",
+        ]
+    )
 
     client = fabric.FabricRestClient()
     response = client.get(f"/v1/workspaces/{workspace_id}/spark/settings")
 
     i = response.json()
-    p = i.get('pool')
-    dp = i.get('pool',{}).get('defaultPool',{})
-    sp = i.get('pool',{}).get('starterPool',{})
-    e = i.get('environment',{})
-
-    new_data = {'Automatic Log Enabled': i.get('automaticLog').get('enabled'), 'High Concurrency Enabled': i.get('highConcurrency').get('notebookInteractiveRunEnabled'), \
-    'Customize Compute Enabled': p.get('customizeComputeEnabled'), 'Default Pool Name': dp.get('name'), 'Default Pool Type': dp.get('type'), \
-    'Max Node Count': sp.get('maxNodeCount'), 'Max Node Executors': sp.get('maxExecutors'), 'Environment Name': e.get('name') , 'Runtime Version': e.get('runtimeVersion')}
+    p = i.get("pool")
+    dp = i.get("pool", {}).get("defaultPool", {})
+    sp = i.get("pool", {}).get("starterPool", {})
+    e = i.get("environment", {})
+
+    new_data = {
+        "Automatic Log Enabled": i.get("automaticLog").get("enabled"),
+        "High Concurrency Enabled": i.get("highConcurrency").get(
+            "notebookInteractiveRunEnabled"
+        ),
+        "Customize Compute Enabled": p.get("customizeComputeEnabled"),
+        "Default Pool Name": dp.get("name"),
+        "Default Pool Type": dp.get("type"),
+        "Max Node Count": sp.get("maxNodeCount"),
+        "Max Node Executors": sp.get("maxExecutors"),
+        "Environment Name": e.get("name"),
+        "Runtime Version": e.get("runtimeVersion"),
+    }
     df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
-    bool_cols = ['Automatic Log Enabled', 'High Concurrency Enabled', 'Customize Compute Enabled']
-    int_cols = ['Max Node Count', 'Max Executors']
+    bool_cols = [
+        "Automatic Log Enabled",
+        "High Concurrency Enabled",
+        "Customize Compute Enabled",
+    ]
+    int_cols = ["Max Node Count", "Max Executors"]
 
     df[bool_cols] = df[bool_cols].astype(bool)
     df[int_cols] = df[int_cols].astype(int)
 
     return df
 
-def update_spark_settings(automatic_log_enabled: Optional[bool] = None, high_concurrency_enabled: Optional[bool] = None, customize_compute_enabled: Optional[bool] = None, default_pool_name: Optional[str] = None, max_node_count: Optional[int] = None, max_executors: Optional[int] = None, environment_name: Optional[str] = None, runtime_version: Optional[str] = None, workspace: Optional[str] = None):
-  
+
+def update_spark_settings(
+    automatic_log_enabled: Optional[bool] = None,
+    high_concurrency_enabled: Optional[bool] = None,
+    customize_compute_enabled: Optional[bool] = None,
+    default_pool_name: Optional[str] = None,
+    max_node_count: Optional[int] = None,
+    max_executors: Optional[int] = None,
+    environment_name: Optional[str] = None,
+    runtime_version: Optional[str] = None,
+    workspace: Optional[str] = None,
+):
     """
     Updates the spark settings for a workspace.
 
@@ -1919,62 +2062,57 @@ def update_spark_settings(automatic_log_enabled: Optional[bool] = None, high_con
     -------
     """
 
-    #https://learn.microsoft.com/en-us/rest/api/fabric/spark/workspace-settings/update-spark-settings?tabs=HTTP
+    # https://learn.microsoft.com/en-us/rest/api/fabric/spark/workspace-settings/update-spark-settings?tabs=HTTP
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
-    dfS = get_spark_settings(workspace = workspace)
+    dfS = get_spark_settings(workspace=workspace)
 
     if automatic_log_enabled is None:
-        automatic_log_enabled = bool(dfS['Automatic Log Enabled'].iloc[0])
+        automatic_log_enabled = bool(dfS["Automatic Log Enabled"].iloc[0])
     if high_concurrency_enabled is None:
-        high_concurrency_enabled = bool(dfS['High Concurrency Enabled'].iloc[0])
+        high_concurrency_enabled = bool(dfS["High Concurrency Enabled"].iloc[0])
     if customize_compute_enabled is None:
-        customize_compute_enabled = bool(dfS['Customize Compute Enabled'].iloc[0])
+        customize_compute_enabled = bool(dfS["Customize Compute Enabled"].iloc[0])
     if default_pool_name is None:
-        default_pool_name = dfS['Default Pool Name'].iloc[0]
+        default_pool_name = dfS["Default Pool Name"].iloc[0]
     if max_node_count is None:
-        max_node_count = int(dfS['Max Node Count'].iloc[0])
+        max_node_count = int(dfS["Max Node Count"].iloc[0])
     if max_executors is None:
-        max_executors = int(dfS['Max Executors'].iloc[0])
+        max_executors = int(dfS["Max Executors"].iloc[0])
     if environment_name is None:
-        environment_name = dfS['Environment Name'].iloc[0]
+        environment_name = dfS["Environment Name"].iloc[0]
     if runtime_version is None:
-        runtime_version = dfS['Runtime Version'].iloc[0]
+        runtime_version = dfS["Runtime Version"].iloc[0]
 
     request_body = {
-    "automaticLog": {
-        "enabled": automatic_log_enabled
-    },
-    "highConcurrency": {
-        "notebookInteractiveRunEnabled": high_concurrency_enabled
-    },
-    "pool": {
-        "customizeComputeEnabled": customize_compute_enabled,
-        "defaultPool": {
-        "name": default_pool_name,
-        "type": "Workspace"
+        "automaticLog": {"enabled": automatic_log_enabled},
+        "highConcurrency": {"notebookInteractiveRunEnabled": high_concurrency_enabled},
+        "pool": {
+            "customizeComputeEnabled": customize_compute_enabled,
+            "defaultPool": {"name": default_pool_name, "type": "Workspace"},
+            "starterPool": {
+                "maxNodeCount": max_node_count,
+                "maxExecutors": max_executors,
+            },
         },
-        "starterPool": {
-        "maxNodeCount": max_node_count,
-        "maxExecutors": max_executors
-        }
-    },
-    "environment": {
-        "name": environment_name,
-        "runtimeVersion": runtime_version
-    }
+        "environment": {"name": environment_name, "runtimeVersion": runtime_version},
     }
 
     client = fabric.FabricRestClient()
-    response = client.patch(f"/v1/workspaces/{workspace_id}/spark/settings", json = request_body)
+    response = client.patch(
+        f"/v1/workspaces/{workspace_id}/spark/settings", json=request_body
+    )
 
-    if response.status_code == 200:
-        print(f"{icons.green_dot} The spark settings within the '{workspace}' workspace have been updated accordingly.")
-    else:
-        raise ValueError(f"{icons.red_dot} {response.status_code}")
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+    print(
+        f"{icons.green_dot} The spark settings within the '{workspace}' workspace have been updated accordingly."
+    )
 
-def add_user_to_workspace(email_address: str, role_name: str, workspace: Optional[str] = None):
 
+def add_user_to_workspace(
+    email_address: str, role_name: str, workspace: Optional[str] = None
+):
     """
     Adds a user to a workspace.
 
@@ -1995,28 +2133,30 @@ def add_user_to_workspace(email_address: str, role_name: str, workspace: Optiona
 
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
-    role_names = ['Admin', 'Member', 'Viewer', 'Contributor']
+    role_names = ["Admin", "Member", "Viewer", "Contributor"]
     role_name = role_name.capitalize()
     if role_name not in role_names:
-        raise ValueError(f"{icons.red_dot} Invalid role. The 'role_name' parameter must be one of the following: {role_names}.")
-    plural = 'n' if role_name == 'Admin' else ''
+        raise ValueError(
+            f"{icons.red_dot} Invalid role. The 'role_name' parameter must be one of the following: {role_names}."
+        )
+    plural = "n" if role_name == "Admin" else ""
 
     client = fabric.PowerBIRestClient()
 
-    request_body = {
-    "emailAddress": email_address,
-    "groupUserAccessRight": role_name
-    }
+    request_body = {"emailAddress": email_address, "groupUserAccessRight": role_name}
 
-    response = client.post(f"/v1.0/myorg/groups/{workspace_id}/users",json=request_body)
-    
-    if response.status_code == 200:
-        print(f"{icons.green_dot} The '{email_address}' user has been added as a{plural} '{role_name}' within the '{workspace}' workspace.")
-    else:
-        print(f"{icons.red_dot} {response.status_code}")
+    response = client.post(
+        f"/v1.0/myorg/groups/{workspace_id}/users", json=request_body
+    )
 
-def delete_user_from_workspace(email_address : str, workspace : Optional[str] = None):
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+    print(
+        f"{icons.green_dot} The '{email_address}' user has been added as a{plural} '{role_name}' within the '{workspace}' workspace."
+    )
 
+
+def delete_user_from_workspace(email_address: str, workspace: Optional[str] = None):
     """
     Removes a user from a workspace.
 
@@ -2037,14 +2177,17 @@ def delete_user_from_workspace(email_address : str, workspace : Optional[str] =
 
     client = fabric.PowerBIRestClient()
     response = client.delete(f"/v1.0/myorg/groups/{workspace_id}/users/{email_address}")
-    
-    if response.status_code == 200:
-        print(f"{icons.green_dot} The '{email_address}' user has been removed from accessing the '{workspace}' workspace.")
-    else:
-        print(f"{icons.red_dot} {response.status_code}")
 
-def update_workspace_user(email_address: str, role_name: str, workspace: Optional[str] = None):
-  
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+    print(
+        f"{icons.green_dot} The '{email_address}' user has been removed from accessing the '{workspace}' workspace."
+    )
+
+
+def update_workspace_user(
+    email_address: str, role_name: str, workspace: Optional[str] = None
+):
     """
     Updates a user's role within a workspace.
 
@@ -2065,26 +2208,26 @@ def update_workspace_user(email_address: str, role_name: str, workspace: Optiona
 
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
-    role_names = ['Admin', 'Member', 'Viewer', 'Contributor']
+    role_names = ["Admin", "Member", "Viewer", "Contributor"]
     role_name = role_name.capitalize()
     if role_name not in role_names:
-            raise ValueError(f"{icons.red_dot} Invalid role. The 'role_name' parameter must be one of the following: {role_names}.")
+        raise ValueError(
+            f"{icons.red_dot} Invalid role. The 'role_name' parameter must be one of the following: {role_names}."
+        )
 
-    request_body = {
-    "emailAddress": email_address,
-    "groupUserAccessRight": role_name
-    }
+    request_body = {"emailAddress": email_address, "groupUserAccessRight": role_name}
 
     client = fabric.PowerBIRestClient()
-    response = client.put(f"/v1.0/myorg/groups/{workspace_id}/users", json = request_body)
+    response = client.put(f"/v1.0/myorg/groups/{workspace_id}/users", json=request_body)
 
-    if response.status_code == 200:
-        print(f"{icons.green_dot} The '{email_address}' user has been updated to a '{role_name}' within the '{workspace}' workspace.")
-    else:
-        print(f"{icons.red_dot} {response.status_code}")
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+    print(
+        f"{icons.green_dot} The '{email_address}' user has been updated to a '{role_name}' within the '{workspace}' workspace."
+    )
 
-def list_workspace_users(workspace: Optional[str] = None) -> pd.DataFrame:
 
+def list_workspace_users(workspace: Optional[str] = None) -> pd.DataFrame:
     """
     A list of all the users of a workspace and their roles.
 
@@ -2103,20 +2246,28 @@ def list_workspace_users(workspace: Optional[str] = None) -> pd.DataFrame:
 
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
-    df = pd.DataFrame(columns=['User Name', 'Email Address', 'Role', 'Type', 'User ID'])
+    df = pd.DataFrame(columns=["User Name", "Email Address", "Role", "Type", "User ID"])
     client = fabric.FabricRestClient()
     response = client.get(f"/v1/workspaces/{workspace_id}/roleAssignments")
 
-    for v in response.json()['value']:
-        p = v.get('principal',{})
+    for v in response.json()["value"]:
+        p = v.get("principal", {})
 
-        new_data = {'User Name': p.get('displayName'), 'User ID': p.get('id'), 'Type': p.get('type'), 'Role': v.get('role'), 'Email Address': p.get('userDetails',{}).get('userPrincipalName')}
+        new_data = {
+            "User Name": p.get("displayName"),
+            "User ID": p.get("id"),
+            "Type": p.get("type"),
+            "Role": v.get("role"),
+            "Email Address": p.get("userDetails", {}).get("userPrincipalName"),
+        }
         df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
     return df
 
-def assign_workspace_to_dataflow_storage(dataflow_storage_account: str, workspace: Optional[str] = None):
 
+def assign_workspace_to_dataflow_storage(
+    dataflow_storage_account: str, workspace: Optional[str] = None
+):
     """
     Assigns a dataflow storage account to a workspace.
 
@@ -2136,17 +2287,53 @@ def assign_workspace_to_dataflow_storage(dataflow_storage_account: str, workspac
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     df = list_dataflow_storage_accounts()
-    df_filt = df[df['Dataflow Storage Account Name'] == dataflow_storage_account]
-    dataflow_storage_id = df_filt['Dataflow Storage Account ID'].iloc[0]
+    df_filt = df[df["Dataflow Storage Account Name"] == dataflow_storage_account]
+    dataflow_storage_id = df_filt["Dataflow Storage Account ID"].iloc[0]
 
     client = fabric.PowerBIRestClient()
 
-    request_body = {
-    "dataflowStorageId": dataflow_storage_id
-    }
+    request_body = {"dataflowStorageId": dataflow_storage_id}
 
-    response = client.post(f"/v1.0/myorg/groups/{workspace_id}/AssignToDataflowStorage",json=request_body)
-    if response.status_code == 200:
-        print(f"{icons.green_dot} The '{dataflow_storage_account}' dataflow storage account has been assigned to the '{workspace}' workspacce.")
-    else:
-        print(f"{icons.red_dot} {response.status_code}")
+    response = client.post(
+        f"/v1.0/myorg/groups/{workspace_id}/AssignToDataflowStorage", json=request_body
+    )
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+    print(
+        f"{icons.green_dot} The '{dataflow_storage_account}' dataflow storage account has been assigned to the '{workspace}' workspacce."
+    )
+
+
+def list_capacities() -> pd.DataFrame:
+    """
+    Shows the capacities and their properties.
+
+    Parameters
+    ----------
+
+    Returns
+    -------
+    pandas.DataFrame
+        A pandas dataframe showing the capacities and their properties
+    """
+
+    df = pd.DataFrame(
+        columns=["Id", "Display Name", "Sku", "Region", "State", "Admins"]
+    )
+
+    client = fabric.PowerBIRestClient()
+    response = client.get("/v1.0/myorg/capacities")
+
+    for i in response.json()["value"]:
+        new_data = {
+            "Id": i.get("id", {}).lower(),
+            "Display Name": i.get("displayName", {}),
+            "Sku": i.get("sku", {}),
+            "Region": i.get("region", {}),
+            "State": i.get("state", {}),
+            "Admins": [i.get("admins", [])],
+        }
+        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+
+    return df