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

sempy_labs/_list_functions.py

@@ -1,17 +1,28 @@
-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,
+    resolve_dataset_id,
+    _decode_b64,
+    pagination,
+    lro,
+)
 import pandas as pd
-import json, time
+import base64
+import requests
+import time
+import json
 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 +43,13 @@ def get_object_level_security(dataset: str, workspace: Optional[str] = None) ->
 
     from sempy_labs.tom import connect_semantic_model
 
-    if workspace is None:        
-        workspace = fabric.resolve_workspace_name()
-    
+    workspace = fabric.resolve_workspace_name(workspace)
+
     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 +57,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 +80,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 +106,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(workspace)
 
-    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:
@@ -163,7 +143,7 @@ def list_annotations(dataset: str, workspace: Optional[str] = None) -> pd.DataFr
 
     from sempy_labs.tom import connect_semantic_model
 
-    workspace = fabric.resolve_workspace_name()
+    workspace = fabric.resolve_workspace_name(workspace)
 
     df = pd.DataFrame(
         columns=[
@@ -175,7 +155,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 +185,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 +265,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 +281,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 +297,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 +313,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 +329,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 +345,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
 
@@ -390,8 +386,7 @@ def list_columns(
         get_direct_lake_lakehouse,
     )
 
-    if workspace is None:
-        workspace = fabric.resolve_workspace_name()
+    workspace = fabric.resolve_workspace_name(workspace)
 
     dfP = fabric.list_partitions(dataset=dataset, workspace=workspace)
 
@@ -431,10 +426,10 @@ def list_columns(
                 ].iloc[0]
 
                 # Build the query to be executed dynamically
-                query = query + f"COUNT(DISTINCT({scName})) AS {scName}, "
+                query = f"{query}COUNT(DISTINCT({scName})) AS {scName}, "
 
             query = query[:-2]
-            query = query + f" FROM {lakehouse}.{lakeTName}"
+            query = f"{query} FROM {lakehouse}.{lakeTName}"
             sql_statements.append((table_name, query))
 
             spark = SparkSession.builder.getOrCreate()
@@ -496,8 +491,10 @@ def list_dashboards(workspace: Optional[str] = None) -> pd.DataFrame:
 
     client = fabric.PowerBIRestClient()
     response = client.get(f"/v1.0/myorg/groups/{workspace_id}/dashboards")
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
 
-    for v in response.json()["value"]:
+    for v in response.json().get("value", []):
         new_data = {
             "Dashboard ID": v.get("id"),
             "Dashboard Name": v.get("displayName"),
@@ -548,23 +545,29 @@ def list_lakehouses(workspace: Optional[str] = None) -> pd.DataFrame:
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     client = fabric.FabricRestClient()
-    response = client.get(f"/v1/workspaces/{workspace_id}/lakehouses/")
+    response = client.get(f"/v1/workspaces/{workspace_id}/lakehouses")
 
-    for v in response.json()["value"]:        
-        prop = v.get("properties",{})
-        sqlEPProp = prop.get("sqlEndpointProperties",{})
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
 
-        new_data = {
-            "Lakehouse Name": v.get("displayName"),
-            "Lakehouse ID": v.get("id"),
-            "Description": v.get("description"),
-            "OneLake Tables Path": prop.get("oneLakeTablesPath"),
-            "OneLake Files Path": prop.get("oneLakeFilesPath"),
-            "SQL Endpoint Connection String": sqlEPProp.get("connectionString"),
-            "SQL Endpoint ID": sqlEPProp.get("id"),
-            "SQL Endpoint Provisioning Status": sqlEPProp.get("provisioningStatus"),
-        }
-        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+    responses = pagination(client, response)
+
+    for r in responses:
+        for v in r.get("value", []):
+            prop = v.get("properties", {})
+            sqlEPProp = prop.get("sqlEndpointProperties", {})
+
+            new_data = {
+                "Lakehouse Name": v.get("displayName"),
+                "Lakehouse ID": v.get("id"),
+                "Description": v.get("description"),
+                "OneLake Tables Path": prop.get("oneLakeTablesPath"),
+                "OneLake Files Path": prop.get("oneLakeFilesPath"),
+                "SQL Endpoint Connection String": sqlEPProp.get("connectionString"),
+                "SQL Endpoint ID": sqlEPProp.get("id"),
+                "SQL Endpoint Provisioning Status": sqlEPProp.get("provisioningStatus"),
+            }
+            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
     return df
 
@@ -600,20 +603,25 @@ def list_warehouses(workspace: Optional[str] = None) -> pd.DataFrame:
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     client = fabric.FabricRestClient()
-    response = client.get(f"/v1/workspaces/{workspace_id}/warehouses/")
+    response = client.get(f"/v1/workspaces/{workspace_id}/warehouses")
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
 
-    for v in response.json()["value"]:        
-        prop = v.get("properties",{})
+    responses = pagination(client, response)
 
-        new_data = {
-            "Warehouse Name": v.get("displayName"),
-            "Warehouse ID": v.get("id"),
-            "Description": v.get("description"),
-            "Connection Info": prop.get("connectionInfo"),
-            "Created Date": prop.get("createdDate"),
-            "Last Updated Time": prop.get("lastUpdatedTime"),
-        }
-        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+    for r in responses:
+        for v in r.get("value", []):
+            prop = v.get("properties", {})
+
+            new_data = {
+                "Warehouse Name": v.get("displayName"),
+                "Warehouse ID": v.get("id"),
+                "Description": v.get("description"),
+                "Connection Info": prop.get("connectionInfo"),
+                "Created Date": prop.get("createdDate"),
+                "Last Updated Time": prop.get("lastUpdatedTime"),
+            }
+            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
     return df
 
@@ -640,16 +648,21 @@ def list_sqlendpoints(workspace: Optional[str] = None) -> pd.DataFrame:
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     client = fabric.FabricRestClient()
-    response = client.get(f"/v1/workspaces/{workspace_id}/sqlEndpoints/")
+    response = client.get(f"/v1/workspaces/{workspace_id}/sqlEndpoints")
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
 
-    for v in response.json()["value"]:
+    responses = pagination(client, response)
 
-        new_data = {
-            "SQL Endpoint ID": v.get("id"),
-            "SQL Endpoint Name": v.get("displayName"),
-            "Description": v.get("description"),
-        }
-        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+    for r in responses:
+        for v in r.get("value", []):
+
+            new_data = {
+                "SQL Endpoint ID": v.get("id"),
+                "SQL Endpoint Name": v.get("displayName"),
+                "Description": v.get("description"),
+            }
+            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
     return df
 
@@ -678,16 +691,21 @@ def list_mirroredwarehouses(workspace: Optional[str] = None) -> pd.DataFrame:
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     client = fabric.FabricRestClient()
-    response = client.get(f"/v1/workspaces/{workspace_id}/mirroredWarehouses/")
+    response = client.get(f"/v1/workspaces/{workspace_id}/mirroredWarehouses")
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
 
-    for v in response.json()["value"]:        
+    responses = pagination(client, response)
 
-        new_data = {
-            "Mirrored Warehouse": v.get("displayName"),
-            "Mirrored Warehouse ID": v.get("id"),
-            "Description": v.get("description"),
-        }
-        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+    for r in responses:
+        for v in r.get("value", []):
+
+            new_data = {
+                "Mirrored Warehouse": v.get("displayName"),
+                "Mirrored Warehouse ID": v.get("id"),
+                "Description": v.get("description"),
+            }
+            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
     return df
 
@@ -724,21 +742,26 @@ def list_kqldatabases(workspace: Optional[str] = None) -> pd.DataFrame:
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     client = fabric.FabricRestClient()
-    response = client.get(f"/v1/workspaces/{workspace_id}/kqlDatabases/")
+    response = client.get(f"/v1/workspaces/{workspace_id}/kqlDatabases")
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
 
-    for v in response.json()["value"]:        
-        prop = v.get("properties",{})
+    responses = pagination(client, response)
 
-        new_data = {
-            "KQL Database Name": v.get("displayName"),
-            "KQL Database ID": v.get("id"),
-            "Description": v.get("description"),
-            "Parent Eventhouse Item ID": prop.get("parentEventhouseItemId"),
-            "Query Service URI": prop.get("queryServiceUri"),
-            "Ingestion Service URI": prop.get("ingestionServiceUri"),
-            "Kusto Database Type": prop.get("kustoDatabaseType"),
-        }
-        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+    for r in responses:
+        for v in r.get("value", []):
+            prop = v.get("properties", {})
+
+            new_data = {
+                "KQL Database Name": v.get("displayName"),
+                "KQL Database ID": v.get("id"),
+                "Description": v.get("description"),
+                "Parent Eventhouse Item ID": prop.get("parentEventhouseItemId"),
+                "Query Service URI": prop.get("queryServiceUri"),
+                "Ingestion Service URI": prop.get("ingestionServiceUri"),
+                "Kusto Database Type": prop.get("kustoDatabaseType"),
+            }
+            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
     return df
 
@@ -765,16 +788,21 @@ def list_kqlquerysets(workspace: Optional[str] = None) -> pd.DataFrame:
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     client = fabric.FabricRestClient()
-    response = client.get(f"/v1/workspaces/{workspace_id}/kqlQuerysets/")
+    response = client.get(f"/v1/workspaces/{workspace_id}/kqlQuerysets")
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
 
-    for v in response.json()["value"]:
+    responses = pagination(client, response)
 
-        new_data = {
-            "KQL Queryset Name": v.get("displayName"),
-            "KQL Queryset ID": v.get("id"),
-            "Description": v.get("description"),
-        }
-        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+    for r in responses:
+        for v in r.get("value", []):
+
+            new_data = {
+                "KQL Queryset Name": v.get("displayName"),
+                "KQL Queryset ID": v.get("id"),
+                "Description": v.get("description"),
+            }
+            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
     return df
 
@@ -801,19 +829,24 @@ def list_mlmodels(workspace: Optional[str] = None) -> pd.DataFrame:
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     client = fabric.FabricRestClient()
-    response = client.get(f"/v1/workspaces/{workspace_id}/mlModels/")
+    response = client.get(f"/v1/workspaces/{workspace_id}/mlModels")
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
 
-    for v in response.json()["value"]:
-        model_id = v.get("id")
-        modelName = v.get("displayName")
-        desc = v.get("description")
+    responses = pagination(client, response)
 
-        new_data = {
-            "ML Model Name": modelName,
-            "ML Model ID": model_id,
-            "Description": desc,
-        }
-        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+    for r in responses:
+        for v in r.get("value", []):
+            model_id = v.get("id")
+            modelName = v.get("displayName")
+            desc = v.get("description")
+
+            new_data = {
+                "ML Model Name": modelName,
+                "ML Model ID": model_id,
+                "Description": desc,
+            }
+            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
     return df
 
@@ -840,19 +873,24 @@ def list_eventstreams(workspace: Optional[str] = None) -> pd.DataFrame:
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     client = fabric.FabricRestClient()
-    response = client.get(f"/v1/workspaces/{workspace_id}/eventstreams/")
+    response = client.get(f"/v1/workspaces/{workspace_id}/eventstreams")
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
 
-    for v in response.json()["value"]:
-        model_id = v.get("id")
-        modelName = v.get("displayName")
-        desc = v.get("description")
+    responses = pagination(client, response)
 
-        new_data = {
-            "Eventstream Name": modelName,
-            "Eventstream ID": model_id,
-            "Description": desc,
-        }
-        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+    for r in responses:
+        for v in r.get("value", []):
+            model_id = v.get("id")
+            modelName = v.get("displayName")
+            desc = v.get("description")
+
+            new_data = {
+                "Eventstream Name": modelName,
+                "Eventstream ID": model_id,
+                "Description": desc,
+            }
+            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
     return df
 
@@ -879,19 +917,20 @@ def list_datapipelines(workspace: Optional[str] = None) -> pd.DataFrame:
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     client = fabric.FabricRestClient()
-    response = client.get(f"/v1/workspaces/{workspace_id}/dataPipelines/")
+    response = client.get(f"/v1/workspaces/{workspace_id}/dataPipelines")
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
 
-    for v in response.json()["value"]:
-        model_id = v.get("id")
-        modelName = v.get("displayName")
-        desc = v.get("description")
+    responses = pagination(client, response)
 
-        new_data = {
-            "Data Pipeline Name": modelName,
-            "Data Pipeline ID": model_id,
-            "Description": desc,
-        }
-        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+    for r in responses:
+        for v in r.get("value", []):
+            new_data = {
+                "Data Pipeline Name": v.get("displayName"),
+                "Data Pipeline ID": v.get("id"),
+                "Description": v.get("description"),
+            }
+            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
     return df
 
@@ -918,16 +957,20 @@ def list_mlexperiments(workspace: Optional[str] = None) -> pd.DataFrame:
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     client = fabric.FabricRestClient()
-    response = client.get(f"/v1/workspaces/{workspace_id}/mlExperiments/")
+    response = client.get(f"/v1/workspaces/{workspace_id}/mlExperiments")
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
 
-    for v in response.json()["value"]:
+    responses = pagination(client, response)
 
-        new_data = {
-            "ML Experiment Name": v.get("displayName"),
-            "ML Experiment ID": v.get("id"),
-            "Description": v.get("description"),
-        }
-        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+    for r in responses:
+        for v in r.get("value", []):
+            new_data = {
+                "ML Experiment Name": v.get("displayName"),
+                "ML Experiment ID": v.get("id"),
+                "Description": v.get("description"),
+            }
+            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
     return df
 
@@ -954,16 +997,20 @@ def list_datamarts(workspace: Optional[str] = None) -> pd.DataFrame:
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     client = fabric.FabricRestClient()
-    response = client.get(f"/v1/workspaces/{workspace_id}/datamarts/")
+    response = client.get(f"/v1/workspaces/{workspace_id}/datamarts")
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
 
-    for v in response.json()["value"]:
+    responses = pagination(client, response)
 
-        new_data = {
-            "Datamart Name": v.get("displayName"),
-            "Datamart ID": v.get("id"),
-            "Description": v.get("description"),
-        }
-        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+    for r in responses:
+        for v in response.get("value", []):
+            new_data = {
+                "Datamart Name": v.get("displayName"),
+                "Datamart ID": v.get("id"),
+                "Description": v.get("description"),
+            }
+            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
     return df
 
@@ -999,27 +1046,15 @@ def create_warehouse(
 
     client = fabric.FabricRestClient()
     response = client.post(
-        f"/v1/workspaces/{workspace_id}/warehouses/", json=request_body
+        f"/v1/workspaces/{workspace_id}/warehouses/", json=request_body, lro_wait=True
+    )
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+    print(
+        f"{icons.green_dot} The '{warehouse}' warehouse has been created within the '{workspace}' workspace."
     )
 
-    if response.status_code == 201:
-        print(
-            f"{icons.green_dot} The '{warehouse}' warehouse has been created within the '{workspace}' workspace."
-        )
-    elif response.status_code == 202:
-        operationId = response.headers["x-ms-operation-id"]
-        response = client.get(f"/v1/operations/{operationId}")
-        response_body = json.loads(response.content)
-        while response_body["status"] != "Succeeded":
-            time.sleep(3)
-            response = client.get(f"/v1/operations/{operationId}")
-            response_body = json.loads(response.content)
-        response = client.get(f"/v1/operations/{operationId}/result")
-        print(
-            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.")
 
 def update_item(
     item_type: str,
@@ -1064,15 +1099,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 +1124,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
@@ -1120,8 +1159,7 @@ def list_relationships(
         A pandas dataframe showing the object level security for the semantic model.
     """
 
-    if workspace is None:
-        workspace = fabric.resolve_workspace_name()
+    workspace = fabric.resolve_workspace_name(workspace)
 
     dfR = fabric.list_relationships(dataset=dataset, workspace=workspace)
 
@@ -1133,7 +1171,7 @@ def list_relationships(
             dax_string="""
                 SELECT
                 [ID] AS [RelationshipID]
-                ,[Name] 
+                ,[Name]
                 FROM $SYSTEM.TMSCHEMA_RELATIONSHIPS
                 """,
         )
@@ -1143,7 +1181,7 @@ def list_relationships(
             dataset=dataset,
             workspace=workspace,
             dax_string="""
-                SELECT 
+                SELECT
                 [TABLE_ID]
                 ,[USED_SIZE]
                 FROM $SYSTEM.DISCOVER_STORAGE_TABLE_COLUMN_SEGMENTS
@@ -1200,10 +1238,11 @@ def list_dataflow_storage_accounts() -> pd.DataFrame:
         ]
     )
     client = fabric.PowerBIRestClient()
-    response = client.get(f"/v1.0/myorg/dataflowStorageAccounts")
-
-    for v in response.json()["value"]:
+    response = client.get("/v1.0/myorg/dataflowStorageAccounts")
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
 
+    for v in response.json().get("value", []):
         new_data = {
             "Dataflow Storage Account ID": v.get("id"),
             "Dataflow Storage Account Name": v.get("name"),
@@ -1303,24 +1342,28 @@ def list_workspace_role_assignments(workspace: Optional[str] = None) -> pd.DataF
 
     client = fabric.FabricRestClient()
     response = client.get(f"/v1/workspaces/{workspace_id}/roleAssignments")
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
 
-    for i in response.json()["value"]:
-        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")
+    responses = pagination(client, response)
 
-        new_data = {
-            "User Name": user_name,
-            "Role Name": role_name,
-            "Type": user_type,
-            "User Email": user_email,
-        }
-        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+    for r in responses:
+        for i in r.get("value", []):
+            principal = i.get("principal", {})
+            new_data = {
+                "User Name": principal.get("displayName"),
+                "Role Name": i.get("role"),
+                "Type": principal.get("type"),
+                "User Email": principal.get("userDetails", {}).get("userPrincipalName"),
+            }
+            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
     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 +1463,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 +1524,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,8 +1573,13 @@ 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"]:
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    responses = pagination(client, response)
+
+    for r in responses:
+        for s in r.get("value", []):
             shortcutName = s.get("name")
             shortcutPath = s.get("path")
             source = list(s["target"].keys())[0]
@@ -1543,17 +1592,19 @@ def list_shortcuts(
                 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")
+                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")
+                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,
@@ -1568,13 +1619,10 @@ def list_shortcuts(
             }
             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 +1638,85 @@ 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")
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
 
-    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 +1739,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 +1750,48 @@ 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, lro_wait=True
+    )
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+    print(
+        f"{icons.green_dot} The '{pool_name}' spark pool has been created within the '{workspace}' workspace."
+    )
 
-    if response.status_code == 201:
-        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 +1822,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):
+    """
+    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 +1938,29 @@ 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,
+        lro_wait=True,
+    )
+
+    if response.status_code not in [200, 202]:
+        raise FabricHTTPException(response)
+    print(
+        f"{icons.green_dot} The '{workspace}' workspace has been assigned to the '{capacity_name}' capacity."
+    )
 
-    if response.status_code == 202:
-        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 +1975,22 @@ 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")
+    response = client.post(
+        f"/v1/workspaces/{workspace_id}/unassignFromCapacity", lro_wait=True
+    )
+
+    if response.status_code not in [200, 202]:
+        raise FabricHTTPException(response)
+    print(
+        f"{icons.green_dot} The '{workspace}' workspace has been unassigned from its capacity."
+    )
+
 
-    if response.status_code == 202:
-        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 +2007,73 @@ 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")
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
 
     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 +2112,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 +2183,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
+    )
+
+    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):
 
+def delete_user_from_workspace(email_address: str, workspace: Optional[str] = None):
     """
     Removes a user from a workspace.
 
@@ -2037,14 +2227,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 +2258,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 +2296,32 @@ 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")
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
 
-    for v in response.json()['value']:
-        p = v.get('principal',{})
+    responses = pagination(client, response)
 
-        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)
+    for r in responses:
+        for v in r.get("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"),
+            }
+            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 +2341,240 @@ 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}
+
+    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")
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    for i in response.json().get("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
+
+
+def get_notebook_definition(
+    notebook_name: str, workspace: Optional[str] = None, decode: Optional[bool] = True
+):
+    """
+    Obtains the notebook definition.
+
+    Parameters
+    ----------
+    notebook_name : str
+        The name of the notebook.
+    workspace : str, default=None
+        The name of the 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.
+    decode : bool, default=True
+        If True, decodes the notebook definition file into .ipynb format.
+        If False, obtains the notebook definition file in base64 format.
+
+    Returns
+    -------
+    ipynb
+        The notebook definition.
+    """
+
+    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+
+    dfI = fabric.list_items(workspace=workspace, type="Notebook")
+    dfI_filt = dfI[dfI["Display Name"] == notebook_name]
+
+    if len(dfI_filt) == 0:
+        raise ValueError(
+            f"{icons.red_dot} The '{notebook_name}' notebook does not exist within the '{workspace}' workspace."
+        )
+
+    notebook_id = dfI_filt["Id"].iloc[0]
+    client = fabric.FabricRestClient()
+    response = client.post(
+        f"v1/workspaces/{workspace_id}/notebooks/{notebook_id}/getDefinition",
+    )
+
+    result = lro(client, response).json()
+    df_items = pd.json_normalize(result["definition"]["parts"])
+    df_items_filt = df_items[df_items["path"] == "notebook-content.py"]
+    payload = df_items_filt["payload"].iloc[0]
+
+    if decode:
+        result = _decode_b64(payload)
+    else:
+        result = payload
+
+    return result
+
+
+def import_notebook_from_web(
+    notebook_name: str,
+    url: str,
+    description: Optional[str] = None,
+    workspace: Optional[str] = None,
+):
+    """
+    Creates a new notebook within a workspace based on a Jupyter notebook hosted in the web.
+
+    Parameters
+    ----------
+    notebook_name : str
+        The name of the notebook to be created.
+    url : str
+        The url of the Jupyter Notebook (.ipynb)
+    description : str, default=None
+        The description of the notebook.
+        Defaults to None which does not place a description.
+    workspace : str, default=None
+        The name of the 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)
+    client = fabric.FabricRestClient()
+    dfI = fabric.list_items(workspace=workspace, type="Notebook")
+    dfI_filt = dfI[dfI["Display Name"] == notebook_name]
+    if len(dfI_filt) > 0:
+        raise ValueError(
+            f"{icons.red_dot} The '{notebook_name}' already exists within the '{workspace}' workspace."
+        )
+
+    # Fix links to go to the raw github file
+    starting_text = "https://github.com/"
+    starting_text_len = len(starting_text)
+    if url.startswith(starting_text):
+        url = f"https://raw.githubusercontent.com/{url[starting_text_len:]}".replace(
+            "/blob/", "/"
+        )
+
+    response = requests.get(url)
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+    file_content = response.content
+    notebook_payload = base64.b64encode(file_content)
+
     request_body = {
-    "dataflowStorageId": dataflow_storage_id
+        "displayName": notebook_name,
+        "definition": {
+            "format": "ipynb",
+            "parts": [
+                {
+                    "path": "notebook-content.py",
+                    "payload": notebook_payload,
+                    "payloadType": "InlineBase64",
+                }
+            ],
+        },
     }
+    if description is not None:
+        request_body["description"] = description
 
-    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/workspaces/{workspace_id}/notebooks", json=request_body)
+
+    lro(client, response, status_codes=[201, 202])
+
+    print(
+        f"{icons.green_dot} The '{notebook_name}' notebook was created within the '{workspace}' workspace."
+    )
+
+
+def list_reports_using_semantic_model(
+    dataset: str, workspace: Optional[str] = None
+) -> pd.DataFrame:
+    """
+    Shows a list of all the reports (in all workspaces) which use a given semantic model.
+
+    Parameters
+    ----------
+    dataset : str
+        Name of the semantic model.
+    workspace : str, default=None
+        The Fabric workspace name.
+        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
+    -------
+    pandas.DataFrame
+        A pandas dataframe showing the reports which use a given semantic model.
+    """
+
+    df = pd.DataFrame(
+        columns=[
+            "Report Name",
+            "Report Id",
+            "Report Workspace Name",
+            "Report Workspace Id",
+        ]
+    )
+
+    workspace = fabric.resolve_workspace_name(workspace)
+    dataset_id = resolve_dataset_id(dataset, workspace)
+    client = fabric.PowerBIRestClient()
+    response = client.get(
+        f"metadata/relations/downstream/dataset/{dataset_id}?apiVersion=3"
+    )
+
+    response_json = response.json()
+
+    for i in response_json.get("artifacts", []):
+        object_workspace_id = i.get("workspace", {}).get("objectId")
+        object_type = i.get("typeName")
+
+        if object_type == "Report":
+            new_data = {
+                "Report Name": i.get("displayName"),
+                "Report Id": i.get("objectId"),
+                "Report Workspace Name": fabric.resolve_workspace_name(
+                    object_workspace_id
+                ),
+                "Report Workspace Id": object_workspace_id,
+            }
+            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+
+    return df