semantic-link-labs 0.9.3__py3-none-any.whl → 0.9.5__py3-none-any.whl

sempy_labs/_capacity_migration.py

@@ -2,10 +2,11 @@ import sempy.fabric as fabric
 from typing import Optional, List
 from sempy._utils._log import log
 import sempy_labs._icons as icons
-from sempy.fabric.exceptions import FabricHTTPException
 from sempy_labs._workspaces import assign_workspace_to_capacity
-from sempy_labs.admin._basic_functions import (
+from sempy_labs.admin import (
     assign_workspaces_to_capacity,
+)
+from sempy_labs.admin._capacities import (
     _list_capacities_meta,
     list_capacities,
 )
@@ -15,6 +16,7 @@ from sempy_labs._helper_functions import (
     _base_api,
 )
 from sempy_labs._capacities import create_fabric_capacity
+from uuid import UUID
 
 
 def _migrate_settings(source_capacity: str, target_capacity: str):
@@ -530,7 +532,7 @@ def _migrate_delegated_tenant_settings(source_capacity: str, target_capacity: st
 
 
 @log
-def _migrate_spark_settings(source_capacity: str, target_capacity: str):
+def _migrate_spark_settings(source_capacity: str | UUID, target_capacity: str | UUID):
     """
     This function migrates a capacity's spark settings to another capacity.
 
@@ -538,14 +540,14 @@ def _migrate_spark_settings(source_capacity: str, target_capacity: str):
 
     Parameters
     ----------
-    source_capacity : str
-        Name of the source capacity.
-    target_capacity : str
-        Name of the target capacity.
+    source_capacity : str | uuid.UUID
+        Name or ID of the source capacity.
+    target_capacity : str | uuid.UUID
+        Name or ID of the target capacity.
     """
 
-    source_capacity_id = resolve_capacity_id(capacity_name=source_capacity)
-    target_capacity_id = resolve_capacity_id(capacity_name=target_capacity)
+    source_capacity_id = resolve_capacity_id(capacity=source_capacity)
+    target_capacity_id = resolve_capacity_id(capacity=target_capacity)
 
     # Get source capacity server dns
     response = _base_api(request=f"metadata/capacityInformation/{source_capacity_id}")

sempy_labs/_dashboards.py

@@ -0,0 +1,60 @@
+from typing import Optional
+from uuid import UUID
+import pandas as pd
+from sempy_labs._helper_functions import (
+    _create_dataframe,
+    _base_api,
+    resolve_workspace_name_and_id,
+    _update_dataframe_datatypes,
+)
+
+
+def list_dashboards(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
+    """
+    Shows a list of the dashboards within a workspace.
+
+    Parameters
+    ----------
+    workspace : str | uuid.UUID, default=None
+        The Fabric workspace name or ID.
+        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 dashboards within a workspace.
+    """
+
+    columns = {
+        "Dashboard ID": "string",
+        "Dashboard Name": "string",
+        "Read Only": "bool",
+        "Web URL": "string",
+        "Embed URL": "string",
+        "Data Classification": "string",
+        "Users": "string",
+        "Subscriptions": "string",
+    }
+    df = _create_dataframe(columns=columns)
+
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+
+    response = _base_api(request=f"/v1.0/myorg/groups/{workspace_id}/dashboards")
+
+    for v in response.json().get("value", []):
+        new_data = {
+            "Dashboard ID": v.get("id"),
+            "Dashboard Name": v.get("displayName"),
+            "Read Only": v.get("isReadOnly"),
+            "Web URL": v.get("webUrl"),
+            "Embed URL": v.get("embedUrl"),
+            "Data Classification": v.get("dataClassification"),
+            "Users": v.get("users"),
+            "Subscriptions": v.get("subscriptions"),
+        }
+        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+
+    _update_dataframe_datatypes(dataframe=df, column_map=columns)
+
+    return df

sempy_labs/_data_pipelines.py

@@ -1,13 +1,13 @@
-import sempy.fabric as fabric
 import pandas as pd
 from typing import Optional
 from sempy_labs._helper_functions import (
     resolve_workspace_name_and_id,
     _decode_b64,
     _base_api,
-    _print_success,
     resolve_item_id,
     _create_dataframe,
+    delete_item,
+    create_item,
 )
 from uuid import UUID
 
@@ -76,25 +76,8 @@ def create_data_pipeline(
         or if no lakehouse attached, resolves to the workspace of the notebook.
     """
 
-    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
-
-    payload = {"displayName": name}
-
-    if description:
-        payload["description"] = description
-
-    _base_api(
-        request=f"/v1/workspaces/{workspace_id}/dataPipelines",
-        method="post",
-        payload=payload,
-        status_codes=[201, 202],
-        lro_return_status_code=True,
-    )
-    _print_success(
-        item_name=name,
-        item_type="data pipeline",
-        workspace_name=workspace_name,
-        action="created",
+    create_item(
+        name=name, description=description, type="DataPipeline", workspace=workspace
     )
 
 
@@ -114,16 +97,7 @@ def delete_data_pipeline(name: str | UUID, workspace: Optional[str | UUID] = Non
         or if no lakehouse attached, resolves to the workspace of the notebook.
     """
 
-    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
-    item_id = resolve_item_id(item=name, type="DataPipeline", workspace=workspace)
-
-    fabric.delete_item(item_id=item_id, workspace=workspace)
-    _print_success(
-        item_name=name,
-        item_type="data pipeline",
-        workspace_name=workspace_name,
-        action="deleted",
-    )
+    delete_item(item=name, type="DataPipeline", workspace=workspace)
 
 
 def get_data_pipeline_definition(

sempy_labs/_dax.py

@@ -62,9 +62,23 @@ def evaluate_dax_impersonation(
         payload=payload,
     )
     data = response.json()["results"][0]["tables"]
-    column_names = data[0]["rows"][0].keys()
-    data_rows = [row.values() for item in data for row in item["rows"]]
-    df = pd.DataFrame(data_rows, columns=column_names)
+
+    # Get all possible column names from all rows because null columns aren't returned
+    all_columns = set()
+    for item in data:
+        for row in item["rows"]:
+            all_columns.update(row.keys())
+
+    # Create rows with all columns, filling missing values with None
+    rows = []
+    for item in data:
+        for row in item["rows"]:
+            # Create a new row with all columns, defaulting to None
+            new_row = {col: row.get(col) for col in all_columns}
+            rows.append(new_row)
+
+    # Create DataFrame from the processed rows
+    df = pd.DataFrame(rows)
 
     return df