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

sempy_labs/admin/_users.py

@@ -0,0 +1,133 @@
+from sempy_labs._helper_functions import (
+    _base_api,
+    _create_dataframe,
+    _update_dataframe_datatypes,
+)
+from uuid import UUID
+import pandas as pd
+
+
+def list_access_entities(
+    user_email_address: str,
+) -> pd.DataFrame:
+    """
+    Shows a list of permission details for Fabric and Power BI items the specified user can access.
+
+    This is a wrapper function for the following API: `Users - List Access Entities <https://learn.microsoft.com/rest/api/fabric/admin/users/list-access-entities>`_.
+
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
+    Parameters
+    ----------
+    user_email_address : str
+        The user's email address.
+
+    Returns
+    -------
+    pandas.DataFrame
+        A pandas dataframe showing a list of permission details for Fabric and Power BI items the specified user can access.
+    """
+
+    columns = {
+        "Item Id": "string",
+        "Item Name": "string",
+        "Item Type": "string",
+        "Permissions": "string",
+        "Additional Permissions": "string",
+    }
+    df = _create_dataframe(columns=columns)
+
+    responses = _base_api(
+        request=f"/v1/admin/users/{user_email_address}/access",
+        client="fabric_sp",
+        uses_pagination=True,
+    )
+
+    for r in responses:
+        for v in r.get("accessEntities", []):
+            new_data = {
+                "Item Id": v.get("id"),
+                "Item Name": v.get("displayName"),
+                "Item Type": v.get("itemAccessDetails", {}).get("type"),
+                "Permissions": v.get("itemAccessDetails", {}).get("permissions"),
+                "Additional Permissions": v.get("itemAccessDetails", {}).get(
+                    "additionalPermissions"
+                ),
+            }
+            df = pd.concat([df, pd.DataFrame([new_data])], ignore_index=True)
+
+    return df
+
+
+def list_user_subscriptions(user: str | UUID) -> pd.DataFrame:
+    """
+    Shows a list of subscriptions for the specified user. This is a preview API call.
+
+    This is a wrapper function for the following API: `Admin - Users GetUserSubscriptionsAsAdmin <https://learn.microsoft.com/rest/api/power-bi/admin/users-get-user-subscriptions-as-admin>`_.
+
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
+    Parameters
+    ----------
+    user : str | uuid.UUID
+        The graph ID or user principal name (UPN) of the user.
+
+    Returns
+    -------
+    pandas.DataFrame
+        A pandas dataframe showing a list of subscriptions for the specified user. This is a preview API call.
+    """
+
+    columns = {
+        "Subscription Id": "string",
+        "Title": "string",
+        "Artifact Id": "string",
+        "Artifact Name": "string",
+        "Sub Artifact Name": "string",
+        "Artifact Type": "string",
+        "Is Enabled": "bool",
+        "Frequency": "string",
+        "Start Date": "datetime",
+        "End Date": "string",
+        "Link To Content": "bool",
+        "Preview Image": "bool",
+        "Attachment Format": "string",
+        "Users": "string",
+    }
+
+    df = _create_dataframe(columns=columns)
+
+    responses = _base_api(
+        request=f"/v1.0/myorg/admin/users/{user}/subscriptions",
+        client="fabric_sp",
+        uses_pagination=True,
+    )
+
+    rows = []
+    for r in responses:
+        for v in r.get("subscriptionEntities", []):
+            rows.append(
+                {
+                    "Subscription Id": v.get("id"),
+                    "Title": v.get("title"),
+                    "Artifact Id": v.get("artifactId"),
+                    "Artifact Name": v.get("artifactDisplayName"),
+                    "Sub Artifact Name": v.get("subArtifactDisplayName"),
+                    "Artifact Type": v.get("artifactType"),
+                    "Is Enabled": v.get("isEnabled"),
+                    "Frequency": v.get("frequency"),
+                    "Start Date": v.get("startDate"),
+                    "End Date": v.get("endDate"),
+                    "Link To Content": v.get("linkToContent"),
+                    "Preview Image": v.get("previewImage"),
+                    "Attachment Format": v.get("attachmentFormat"),
+                    "Users": str(v.get("users")),
+                }
+            )
+
+    if rows:
+        df = pd.DataFrame(rows, columns=list(columns.keys()))
+
+    _update_dataframe_datatypes(dataframe=df, column_map=columns)
+
+    return df

sempy_labs/admin/_workspaces.py

@@ -0,0 +1,148 @@
+from sempy_labs._helper_functions import (
+    _base_api,
+    _build_url,
+    _encode_user,
+)
+from uuid import UUID
+from typing import Optional
+from sempy_labs.admin._basic_functions import (
+    _resolve_workspace_name_and_id,
+)
+import sempy_labs._icons as icons
+
+
+def add_user_to_workspace(
+    user: str | UUID,
+    role: str = "Member",
+    principal_type: str = "User",
+    workspace: Optional[str | UUID] = None,
+):
+    """
+    Grants user permissions to the specified workspace.
+
+    This is a wrapper function for the following API: `Admin - Groups AddUserAsAdmin <https://learn.microsoft.com/rest/api/power-bi/admin/groups-add-user-as-admin>`_.
+
+    Parameters
+    ----------
+    user : str | uuid.UUID
+        The user identifier or email address. For service principals and groups you must use the user identifier.
+    role : str, default="Member"
+        The role of the user in the workspace. Options are: 'Admin', 'Contributor', 'Member', 'None', 'Viewer'.
+    principal_type : str, default="User"
+        The principal type of the user. Options are: 'App', 'Group', 'None', 'User'.
+    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.
+    """
+
+    (workspace_name, workspace_id) = _resolve_workspace_name_and_id(workspace)
+
+    # Validation
+    role = role.capitalize()
+    roles = ["Admin", "Contributor", "Member", "None", "Viewer"]
+    if role not in roles:
+        raise ValueError(f"{icons.red_dot} Invalid role. Please choose from {roles}")
+    principal_types = ["App", "Group", "None", "User"]
+    if principal_type not in principal_types:
+        raise ValueError(
+            f"{icons.red_dot} Invalid principal type. Please choose from {principal_types}"
+        )
+
+    user = _encode_user(user)
+
+    payload = {
+        "identifier": user,  # identifier or emailAddress?
+        "principalType": principal_type,
+        "groupUserAccessRight": role,
+    }
+
+    _base_api(
+        request=f"/v1.0/myorg/admin/groups/{workspace_id}/users",
+        method="post",
+        payload=payload,
+    )
+
+    print(
+        f"{icons.green_dot} The '{user}' user has been added with '{role.lower()}' permissions to the '{workspace_name}' workspace."
+    )
+
+
+def delete_user_from_workspace(
+    user: str | UUID,
+    workspace: Optional[str | UUID] = None,
+    is_group: Optional[bool] = None,
+    profile_id: Optional[str] = None,
+):
+    """
+    Removes user permissions from the specified workspace.
+
+    This is a wrapper function for the following API: `Admin - Groups DeleteUserAsAdmin <https://learn.microsoft.com/rest/api/power-bi/admin/groups-delete-user-as-admin>`_.
+
+    Parameters
+    ----------
+    user : str | uuid.UUID
+        The user identifier or email address. For service principals and groups you must use the user identifier.
+    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.
+    is_group : bool, default=None
+        Whether a given user is a group or not. This parameter is required when user to delete is group.
+    profile_id : str, default=None
+        The service principal profile ID to delete.
+    """
+
+    (workspace_name, workspace_id) = _resolve_workspace_name_and_id(workspace)
+
+    user = _encode_user(user)
+    url = f"/v1.0/myorg/admin/groups/{workspace_id}/users/{user}"
+
+    params = {}
+    if profile_id is not None:
+        params["profileId"] = profile_id
+    if is_group is not None:
+        params["isGroup"] = is_group
+
+    url = _build_url(url, params)
+
+    _base_api(
+        request=url,
+        method="delete",
+    )
+
+    print(
+        f"{icons.green_dot} The '{user}' user has been removed from the '{workspace_name}' workspace."
+    )
+
+
+def restore_deleted_workspace(workspace_id: UUID, name: str, email_address: str):
+    """
+    Restores a deleted workspace.
+
+    This is a wrapper function for the following API: `Admin - Groups RestoreDeletedGroupAsAdmin <https://learn.microsoft.com/rest/api/power-bi/admin/groups-restore-deleted-group-as-admin>`_.
+
+    Parameters
+    ----------
+    workspace_id : uuid.UUID
+        The ID of the workspace to restore.
+    name : str
+        The name of the group to be restored
+    email_address : str
+        The email address of the owner of the group to be restored
+    """
+
+    payload = {
+        "name": name,
+        "emailAddress": email_address,
+    }
+
+    _base_api(
+        request=f"/v1.0/myorg/admin/groups/{workspace_id}/restore",
+        method="post",
+        payload=payload,
+    )
+
+    print(
+        f"{icons.green_dot} The '{workspace_id}' workspace has been restored as '{name}'."
+    )

sempy_labs/directlake/_update_directlake_partition_entity.py

@@ -17,6 +17,7 @@ def update_direct_lake_partition_entity(
     dataset: str | UUID,
     table_name: Union[str, List[str]],
     entity_name: Union[str, List[str]],
+    schema: Optional[str] = None,
     workspace: Optional[str | UUID] = None,
 ):
     """
@@ -30,6 +31,9 @@ def update_direct_lake_partition_entity(
         Name of the table(s) in the semantic model.
     entity_name : str, List[str]
         Name of the lakehouse table to be mapped to the semantic model table.
+    schema : str, default=None
+        The schema of the lakehouse table to be mapped to the semantic model table.
+        Defaults to None which resolves to the existing schema of the lakehouse table.
     workspace : str | uuid.UUID, default=None
         The Fabric workspace name or ID in which the semantic model exists.
         Defaults to None which resolves to the workspace of the attached lakehouse
@@ -79,9 +83,13 @@ def update_direct_lake_partition_entity(
             tom.model.Tables[tName].Partitions[part_name].Source.EntityName = eName
 
             # Update source lineage tag
-            schema = (
+            existing_schema = (
                 tom.model.Tables[tName].Partitions[part_name].Source.SchemaName or "dbo"
             )
+            if schema is None:
+                schema = existing_schema
+
+            tom.model.Tables[tName].Partitions[part_name].Source.SchemaName = schema
             tom.model.Tables[tName].SourceLineageTag = f"[{schema}].[{eName}]"
             print(
                 f"{icons.green_dot} The '{tName}' table in the '{dataset_name}' semantic model within the '{workspace_name}' workspace has been updated to point to the '{eName}' table."

sempy_labs/lakehouse/__init__.py

@@ -12,6 +12,7 @@ from sempy_labs.lakehouse._shortcuts import (
     create_shortcut_onelake,
     delete_shortcut,
     reset_shortcut_cache,
+    list_shortcuts,
 )
 
 __all__ = [
@@ -25,4 +26,5 @@ __all__ = [
     "vacuum_lakehouse_tables",
     "reset_shortcut_cache",
     "run_table_maintenance",
+    "list_shortcuts",
 ]

sempy_labs/lakehouse/_lakehouse.py

@@ -35,7 +35,7 @@ def lakehouse_attached() -> bool:
 @log
 def optimize_lakehouse_tables(
     tables: Optional[Union[str, List[str]]] = None,
-    lakehouse: Optional[str] = None,
+    lakehouse: Optional[str | UUID] = None,
     workspace: Optional[str | UUID] = None,
 ):
     """
@@ -46,8 +46,8 @@ def optimize_lakehouse_tables(
     tables : str | List[str], default=None
         The table(s) to optimize.
         Defaults to None which resovles to optimizing all tables within the lakehouse.
-    lakehouse : str, default=None
-        The Fabric lakehouse.
+    lakehouse : str | uuid.UUID, default=None
+        The Fabric lakehouse name or ID.
         Defaults to None which resolves to the lakehouse attached to the notebook.
     workspace : str | uuid.UUID, default=None
         The Fabric workspace name or ID used by the lakehouse.
@@ -82,7 +82,7 @@ def optimize_lakehouse_tables(
 @log
 def vacuum_lakehouse_tables(
     tables: Optional[Union[str, List[str]]] = None,
-    lakehouse: Optional[str] = None,
+    lakehouse: Optional[str | UUID] = None,
     workspace: Optional[str | UUID] = None,
     retain_n_hours: Optional[int] = None,
 ):
@@ -93,8 +93,8 @@ def vacuum_lakehouse_tables(
     ----------
     tables : str | List[str] | None
         The table(s) to vacuum. If no tables are specified, all tables in the lakehouse will be optimized.
-    lakehouse : str, default=None
-        The Fabric lakehouse.
+    lakehouse : str | uuid.UUID, default=None
+        The Fabric lakehouse name or ID.
         Defaults to None which resolves to the lakehouse attached to the notebook.
     workspace : str | uuid.UUID, default=None
         The Fabric workspace name or ID used by the lakehouse.
@@ -107,7 +107,6 @@ def vacuum_lakehouse_tables(
         The default retention period is 168 hours (7 days) unless manually configured via table properties.
     """
 
-    from pyspark.sql import SparkSession
     from sempy_labs.lakehouse._get_lakehouse_tables import get_lakehouse_tables
     from delta import DeltaTable