semantic-link-labs 0.8.8__py3-none-any.whl → 0.8.10__py3-none-any.whl

sempy_labs/admin/__init__.py

@@ -1,3 +1,6 @@
+from sempy_labs.admin._scanner import (
+    scan_workspaces,
+)
 from sempy_labs.admin._basic_functions import (
     assign_workspaces_to_capacity,
     unassign_workspaces_from_capacity,
@@ -11,6 +14,7 @@ from sempy_labs.admin._basic_functions import (
     list_capacities_delegated_tenant_settings,
     list_access_entities,
     list_activity_events,
+    get_capacity_assignment_status,
 )
 from sempy_labs.admin._domains import (
     list_domains,
@@ -64,4 +68,6 @@ __all__ = [
     "list_modified_workspaces",
     "list_git_connections",
     "list_reports",
+    "get_capacity_assignment_status",
+    "scan_workspaces",
 ]

sempy_labs/admin/_basic_functions.py

@@ -11,7 +11,6 @@ from sempy_labs._helper_functions import (
 import numpy as np
 import pandas as pd
 from dateutil.parser import parse as dtparser
-import urllib.parse
 
 
 def list_workspaces(
@@ -126,7 +125,7 @@ def list_capacities(
     capacity: Optional[str | UUID] = None,
 ) -> pd.DataFrame:
     """
-    Shows the a list of capacities and their properties. This function is the admin version.
+    Shows the a list of capacities and their properties.
 
     This is a wrapper function for the following API: `Admin - Get Capacities As Admin <https://learn.microsoft.com/rest/api/power-bi/admin/get-capacities-as-admin>`_.
 
@@ -138,7 +137,7 @@ def list_capacities(
     Returns
     -------
     pandas.DataFrame
-        A pandas dataframe showing the capacities and their properties
+        A pandas dataframe showing the capacities and their properties.
     """
     client = fabric.FabricRestClient()
 
@@ -214,6 +213,7 @@ def assign_workspaces_to_capacity(
         if source_capacity is None:
             dfW = list_workspaces()
         else:
+            source_capacity_id = _resolve_capacity_name_and_id(source_capacity)[1]
             dfW = list_workspaces(capacity=source_capacity_id)
 
         # Extract names and IDs that are mapped in dfW
@@ -230,10 +230,10 @@ def assign_workspaces_to_capacity(
             if item not in workspaces_names and item not in workspaces_ids
         ]
 
-    if len(workspace) != len(workspaces):
-        raise ValueError(
-            f"{icons.red_dot} The following workspaces are invalid: {unmapped_workspaces}."
-        )
+        if len(workspace) != len(workspaces):
+            raise ValueError(
+                f"{icons.red_dot} The following workspaces are invalid or not found in source capacity: {unmapped_workspaces}."
+            )
 
     target_capacity_id = _resolve_capacity_name_and_id(target_capacity)[1]
 
@@ -777,38 +777,38 @@ def list_activity_events(
             "Object Display Name",
             "Experience",
             "Refresh Enforcement Policy",
+            "Is Success",
+            "Activity Id",
+            "Item Name",
+            "Dataset Name",
+            "Report Name",
+            "Capacity Id",
+            "Capacity Name",
+            "App Name",
+            "Dataset Id",
+            "Report Id",
+            "Artifact Id",
+            "Artifact Name",
+            "Report Type",
+            "App Report Id",
+            "Distribution Method",
+            "Consumption Method",
+            "Artifact Kind",
         ]
     )
 
-    resposeJson = {"activityEventEntities": []}
-
-    tic = "%27"
-    space = "%20"
+    response_json = {"activityEventEntities": []}
     client = fabric.PowerBIRestClient()
-
-    params = {}
-    url = "/v1.0/myorg/admin/activityevents"
-
-    if start_dt is not None:
-        params["startDateTime"] = f"'{start_dt.isoformat(timespec='milliseconds')}'"
-
-    if end_dt is not None:
-        params["endDateTime"] = f"'{end_dt.isoformat(timespec='milliseconds')}'"
+    url = f"/v1.0/myorg/admin/activityevents?startDateTime='{start_time}'&endDateTime='{end_time}'"
 
     conditions = []
-
     if activity_filter is not None:
-        conditions.append(f"Activity{space}eq{space}{tic}{activity_filter}{tic}")
-
+        conditions.append(f"Activity eq '{activity_filter}'")
     if user_id_filter is not None:
-        conditions.append(f"UserId{space}eq{space}{tic}{user_id_filter}{tic}")
+        conditions.append(f"UserId eq '{user_id_filter}'")
 
     if conditions:
-        params["filder"] = f"{f'{space}and{space}'.join(conditions)}"
-
-    url_parts = list(urllib.parse.urlparse(url))
-    url_parts[4] = urllib.parse.urlencode(params)
-    url = urllib.parse.urlunparse(url_parts)
+        url += f"&$filter={f' and '.join(conditions)}"
 
     response = client.get(url)
 
@@ -842,21 +842,38 @@ def list_activity_events(
                     "Object Display Name": i.get("ObjectDisplayName"),
                     "Experience": i.get("Experience"),
                     "Refresh Enforcement Policy": i.get("RefreshEnforcementPolicy"),
+                    "Is Success": i.get("IsSuccess"),
+                    "Activity Id": i.get("ActivityId"),
+                    "Item Name": i.get("ItemName"),
+                    "Dataset Name": i.get("DatasetName"),
+                    "Report Name": i.get("ReportName"),
+                    "Capacity Id": i.get("CapacityId"),
+                    "Capacity Name": i.get("CapacityName"),
+                    "App Name": i.get("AppName"),
+                    "Dataset Id": i.get("DatasetId"),
+                    "Report Id": i.get("ReportId"),
+                    "Artifact Id": i.get("ArtifactId"),
+                    "Artifact Name": i.get("ArtifactName"),
+                    "Report Type": i.get("ReportType"),
+                    "App Report Id": i.get("AppReportId"),
+                    "Distribution Method": i.get("DistributionMethod"),
+                    "Consumption Method": i.get("ConsumptionMethod"),
+                    "Artifact Kind": i.get("ArtifactKind"),
                 }
                 df = pd.concat(
                     [df, pd.DataFrame(new_data, index=[0])],
                     ignore_index=True,
                 )
         else:
-            resposeJson["activityEventEntities"].extend(r.get("activityEventEntities"))
+            response_json["activityEventEntities"].extend(
+                r.get("activityEventEntities")
+            )
 
     if return_dataframe:
         df["Creation Time"] = pd.to_datetime(df["Creation Time"])
-        activity_events = df
+        return df
     else:
-        activity_events = resposeJson
-
-    return activity_events
+        return response_json
 
 
 def _resolve_capacity_name_and_id(
@@ -917,12 +934,18 @@ def _resolve_workspace_name_and_id(
     workspace: str | UUID,
 ) -> Tuple[str, UUID]:
 
-    dfW = list_workspaces(workspace=workspace)
-    try:
-        workspace_name = dfW["Name"].iloc[0]
-        workspace_id = dfW["Id"].iloc[0]
-    except Exception:
-        raise ValueError(f"{icons.red_dot} The '{workspace}' workspace was not found.")
+    if workspace is None:
+        workspace_id = fabric.get_workspace_id()
+        workspace_name = fabric.resolve_workspace_name(workspace_id)
+    else:
+        dfW = list_workspaces(workspace=workspace)
+        if not dfW.empty:
+            workspace_name = dfW["Name"].iloc[0]
+            workspace_id = dfW["Id"].iloc[0]
+        else:
+            raise ValueError(
+                f"{icons.red_dot} The '{workspace}' workspace was not found."
+            )
 
     return workspace_name, workspace_id
 
@@ -1004,3 +1027,60 @@ def list_reports(
     df["Modified Date"] = pd.to_datetime(df["Modified Date"], errors="coerce")
 
     return df
+
+
+def get_capacity_assignment_status(workspace: Optional[str | UUID] = None):
+    """
+    Gets the status of the assignment-to-capacity operation for the specified workspace.
+
+    This is a wrapper function for the following API: `Capacities - Groups CapacityAssignmentStatus <https://learn.microsoft.com/rest/api/power-bi/capacities/groups-capacity-assignment-status>`_.
+
+    Parameters
+    ----------
+    workspace : str | 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 status of the assignment-to-capacity operation for the specified workspace.
+    """
+
+    (workspace_name, workspace_id) = _resolve_workspace_name_and_id(workspace)
+
+    df = pd.DataFrame(
+        columns=[
+            "Status",
+            "Activity Id",
+            "Start Time",
+            "End Time",
+            "Capacity Id",
+            "Capacity Name",
+        ]
+    )
+
+    client = fabric.FabricRestClient()
+    response = client.get(f"/v1.0/myorg/groups/{workspace_id}/CapacityAssignmentStatus")
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    v = response.json()
+    capacity_id = v.get("capacityId")
+
+    (capacity_name, capacity_id) = _resolve_capacity_name_and_id(capacity=capacity_id)
+
+    new_data = {
+        "Status": v.get("status"),
+        "Activity Id": v.get("activityId"),
+        "Start Time": v.get("startTime"),
+        "End Time": v.get("endTime"),
+        "Capacity Id": capacity_id,
+        "Capacity Name": capacity_name,
+    }
+
+    df = pd.concat([df, pd.DataFrame([new_data])], ignore_index=True)
+
+    return df

sempy_labs/admin/_domains.py

@@ -5,6 +5,7 @@ from sempy_labs._helper_functions import lro
 from sempy.fabric.exceptions import FabricHTTPException
 import pandas as pd
 from uuid import UUID
+from sempy_labs.admin._basic_functions import list_workspaces
 
 
 def resolve_domain_id(domain_name: str) -> UUID:
@@ -302,7 +303,7 @@ def assign_domain_workspaces(domain_name: str, workspace_names: str | List[str])
     if isinstance(workspace_names, str):
         workspace_names = [workspace_names]
 
-    dfW = fabric.list_workspaces()
+    dfW = list_workspaces()
 
     # Check for invalid capacities
     invalid_workspaces = [
@@ -379,7 +380,7 @@ def unassign_domain_workspaces(domain_name: str, workspace_names: str | List[str
     if isinstance(workspace_names, str):
         workspace_names = [workspace_names]
 
-    dfW = fabric.list_workspaces()
+    dfW = list_workspaces()
 
     # Check for invalid capacities
     invalid_workspaces = [

sempy_labs/admin/_scanner.py

@@ -16,12 +16,12 @@ def scan_workspaces(
     workspace: Optional[str | List[str] | UUID | List[UUID]] = None,
 ) -> dict:
     """
-    Get the inventory and details of the tenant.
+    Gets the scan result for the specified scan.
 
     This is a wrapper function for the following APIs:
-        `Admin - WorkspaceInfo PostWorkspaceInfo <https://learn.microsoft.com/en-gb/rest/api/power-bi/admin/workspace-info-post-workspace-info>`_.
-        `Admin - WorkspaceInfo GetScanStatus <https://learn.microsoft.com/en-gb/rest/api/power-bi/admin/workspace-info-get-scan-status>`_.
-        `Admin - WorkspaceInfo GetScanResult <https://learn.microsoft.com/en-gb/rest/api/power-bi/admin/workspace-info-get-scan-result>`_.
+        `Admin - WorkspaceInfo PostWorkspaceInfo <https://learn.microsoft.com/rest/api/power-bi/admin/workspace-info-post-workspace-info>`_.
+        `Admin - WorkspaceInfo GetScanStatus <https://learn.microsoft.com/rest/api/power-bi/admin/workspace-info-get-scan-status>`_.
+        `Admin - WorkspaceInfo GetScanResult <https://learn.microsoft.com/rest/api/power-bi/admin/workspace-info-get-scan-result>`_.
 
     Parameters
     ----------
@@ -40,7 +40,7 @@ def scan_workspaces(
 
     Returns
     -------
-    dictionary
+    dict
         A json object with the scan result.
     """
     scan_result = {

sempy_labs/directlake/_dl_helper.py

@@ -10,19 +10,21 @@ from sempy_labs._helper_functions import (
     resolve_dataset_id,
     resolve_lakehouse_name,
     _convert_data_type,
+    resolve_dataset_name_and_id,
+    resolve_workspace_name_and_id,
 )
 
 
 def check_fallback_reason(
-    dataset: str, workspace: Optional[str] = None
+    dataset: str | UUID, workspace: Optional[str] = None
 ) -> pd.DataFrame:
     """
     Shows the reason a table in a Direct Lake semantic model would fallback to DirectQuery.
 
     Parameters
     ----------
-    dataset : str
-        Name of the semantic model.
+    dataset : str | UUID
+        Name or ID of the semantic model.
     workspace : str, default=None
         The Fabric workspace name.
         Defaults to None which resolves to the workspace of the attached lakehouse
@@ -35,19 +37,22 @@ def check_fallback_reason(
     """
     from sempy_labs.tom import connect_semantic_model
 
-    workspace = fabric.resolve_workspace_name(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (dataset_name, dataset_id) = resolve_dataset_name_and_id(
+        dataset, workspace=workspace_id
+    )
 
     with connect_semantic_model(
-        dataset=dataset, workspace=workspace, readonly=True
+        dataset=dataset_id, workspace=workspace_id, readonly=True
     ) as tom:
         if not tom.is_direct_lake():
             raise ValueError(
-                f"{icons.red_dot} The '{dataset}' semantic model is not in Direct Lake. This function is only applicable to Direct Lake semantic models."
+                f"{icons.red_dot} The '{dataset_name}' semantic model is not in Direct Lake. This function is only applicable to Direct Lake semantic models."
             )
 
     df = fabric.evaluate_dax(
-        dataset=dataset,
-        workspace=workspace,
+        dataset=dataset_id,
+        workspace=workspace_id,
         dax_string="""
             SELECT [TableName] AS [Table Name],[FallbackReason] AS [FallbackReasonID]
             FROM $SYSTEM.TMSCHEMA_DELTA_TABLE_METADATA_STORAGES

sempy_labs/report/_reportwrapper.py

@@ -970,6 +970,7 @@ class ReportWrapper:
                 "Sparkline",
                 "Visual Calc",
                 "Format",
+                "Object Display Name",
             ]
         )
 
@@ -1038,23 +1039,26 @@ class ReportWrapper:
 
                 entity_property_pairs = find_entity_property_pairs(visual_json)
                 query_state = (
-                    visual_json.get("visual", {})
-                    .get("query", {})
-                    .get("queryState", {})
-                    .get("Values", {})
+                    visual_json.get("visual", {}).get("query", {}).get("queryState", {})
                 )
+
                 format_mapping = {}
-                for p in query_state.get("projections", []):
-                    query_ref = p.get("queryRef")
-                    fmt = p.get("format")
-                    if fmt is not None:
-                        format_mapping[query_ref] = fmt
+                obj_display_mapping = {}
+                for a, p in query_state.items():
+                    for proj in p.get("projections", []):
+                        query_ref = proj.get("queryRef")
+                        fmt = proj.get("format")
+                        obj_display_name = proj.get("displayName")
+                        if fmt is not None:
+                            format_mapping[query_ref] = fmt
+                        obj_display_mapping[query_ref] = obj_display_name
 
                 for object_name, properties in entity_property_pairs.items():
                     table_name = properties[0]
                     obj_full = f"{table_name}.{object_name}"
                     is_agg = properties[2]
                     format_value = format_mapping.get(obj_full)
+                    obj_display = obj_display_mapping.get(obj_full)
 
                     if is_agg:
                         for k, v in format_mapping.items():
@@ -1071,6 +1075,7 @@ class ReportWrapper:
                         "Sparkline": properties[4],
                         "Visual Calc": properties[3],
                         "Format": format_value,
+                        "Object Display Name": obj_display,
                     }
 
                     df = pd.concat(