semantic-link-labs 0.9.6__py3-none-any.whl → 0.9.8__py3-none-any.whl

sempy_labs/_helper_functions.py

@@ -183,7 +183,7 @@ def resolve_report_name(report_id: UUID, workspace: Optional[str | UUID] = None)
         The name of the Power BI report.
     """
 
-    return resolve_item_name(item_id=report_id, type="Report", workspace=workspace)
+    return resolve_item_name(item_id=report_id, workspace=workspace)
 
 
 def delete_item(
@@ -392,7 +392,7 @@ def resolve_lakehouse_name_and_id(
     lakehouse: Optional[str | UUID] = None, workspace: Optional[str | UUID] = None
 ) -> Tuple[str, UUID]:
 
-    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    workspace_id = resolve_workspace_id(workspace)
     type = "Lakehouse"
 
     if lakehouse is None:
@@ -469,9 +469,7 @@ def resolve_dataset_name(
         The name of the semantic model.
     """
 
-    return resolve_item_name(
-        item_id=dataset_id, type="SemanticModel", workspace=workspace
-    )
+    return resolve_item_name(item_id=dataset_id, workspace=workspace)
 
 
 def resolve_lakehouse_name(
@@ -503,9 +501,7 @@ def resolve_lakehouse_name(
                 f"{icons.red_dot} Cannot resolve a lakehouse. Please enter a valid lakehouse or make sure a lakehouse is attached to the notebook."
             )
 
-    return resolve_item_name(
-        item_id=lakehouse_id, type="Lakehouse", workspace=workspace
-    )
+    return resolve_item_name(item_id=lakehouse_id, workspace=workspace)
 
 
 def resolve_lakehouse_id(
@@ -1308,10 +1304,8 @@ class FabricTokenCredential(TokenCredential):
 
         import notebookutils
 
-        token = notebookutils.credentials.getToken(scopes)
-        access_token = AccessToken(token, 0)
-
-        return access_token
+        token = notebookutils.credentials.getToken("storage")
+        return AccessToken(token, 0)
 
 
 def _get_adls_client(account_name):
@@ -1320,11 +1314,21 @@ def _get_adls_client(account_name):
 
     account_url = f"https://{account_name}.dfs.core.windows.net"
 
-    service_client = DataLakeServiceClient(
-        account_url, credential=FabricTokenCredential()
+    return DataLakeServiceClient(account_url, credential=FabricTokenCredential())
+
+
+def _get_blob_client(workspace_id: UUID, item_id: UUID):
+
+    from azure.storage.blob import BlobServiceClient
+
+    endpoint = _get_fabric_context_setting(name="trident.onelake.endpoint").replace(
+        ".dfs.", ".blob."
     )
+    url = f"https://{endpoint}/{workspace_id}/{item_id}"
+
+    # account_url = f"https://{account_name}.blob.core.windows.net"
 
-    return service_client
+    return BlobServiceClient(url, credential=FabricTokenCredential())
 
 
 def resolve_warehouse_id(
@@ -1889,7 +1893,9 @@ def _run_spark_sql_query(query):
     return spark.sql(query)
 
 
-def _mount(lakehouse, workspace) -> str:
+def _mount(
+    lakehouse: Optional[str | UUID] = None, workspace: Optional[str | UUID] = None
+) -> str:
     """
     Mounts a lakehouse to a notebook if it is not already mounted. Returns the local path to the lakehouse.
     """
@@ -1901,6 +1907,16 @@ def _mount(lakehouse, workspace) -> str:
         lakehouse=lakehouse, workspace=workspace
     )
 
+    # Hide display mounts
+    current_setting = ""
+    try:
+        current_setting = notebookutils.conf.get(
+            "spark.notebookutils.displaymountpoint.enabled"
+        )
+        notebookutils.conf.set("spark.notebookutils.displaymountpoint.enabled", "false")
+    except Exception:
+        pass
+
     lake_path = create_abfss_path(lakehouse_id, workspace_id)
     mounts = notebookutils.fs.mounts()
     mount_point = f"/{workspace_name.replace(' ', '')}{lakehouse_name.replace(' ', '')}"
@@ -1912,6 +1928,16 @@ def _mount(lakehouse, workspace) -> str:
         )
 
     mounts = notebookutils.fs.mounts()
+
+    # Set display mounts to original setting
+    try:
+        if current_setting != "false":
+            notebookutils.conf.set(
+                "spark.notebookutils.displaymountpoint.enabled", "true"
+            )
+    except Exception:
+        pass
+
     local_path = next(
         i.get("localPath") for i in mounts if i.get("source") == lake_path
     )
@@ -2015,3 +2041,26 @@ def _get_or_create_warehouse(
     )
 
     return (warehouse, warehouse_id)
+
+
+def _xml_to_dict(element):
+    data = {element.tag: {} if element.attrib else None}
+    children = list(element)
+    if children:
+        temp_dict = {}
+        for child in children:
+            child_dict = _xml_to_dict(child)
+            for key, value in child_dict.items():
+                if key in temp_dict:
+                    if isinstance(temp_dict[key], list):
+                        temp_dict[key].append(value)
+                    else:
+                        temp_dict[key] = [temp_dict[key], value]
+                else:
+                    temp_dict[key] = value
+        data[element.tag] = temp_dict
+    else:
+        data[element.tag] = (
+            element.text.strip() if element.text and element.text.strip() else None
+        )
+    return data

sempy_labs/_icons.py

@@ -1,6 +1,6 @@
-green_dot = "\U0001F7E2"
-yellow_dot = "\U0001F7E1"
-red_dot = "\U0001F534"
+green_dot = "\U0001f7e2"
+yellow_dot = "\U0001f7e1"
+red_dot = "\U0001f534"
 in_progress = "⌛"
 checked = "\u2611"
 unchecked = "\u2610"
@@ -8,11 +8,11 @@ start_bold = "\033[1m"
 end_bold = "\033[0m"
 bullet = "\u2022"
 warning = "⚠️"
-error = "\u274C"
+error = "\u274c"
 info = "ℹ️"
 measure_icon = "\u2211"
-table_icon = "\u229E"
-column_icon = "\u229F"
+table_icon = "\u229e"
+column_icon = "\u229f"
 model_bpa_name = "ModelBPA"
 report_bpa_name = "ReportBPA"
 severity_mapping = {warning: "Warning", error: "Error", info: "Info"}

sempy_labs/_list_functions.py

@@ -1476,7 +1476,9 @@ def list_server_properties(workspace: Optional[str | UUID] = None) -> pd.DataFra
         A pandas dataframe showing a list of the server properties.
     """
 
-    tom_server = fabric.create_tom_server(readonly=True, workspace=workspace)
+    tom_server = fabric.create_tom_server(
+        dataset=None, readonly=True, workspace=workspace
+    )
 
     rows = [
         {

sempy_labs/_model_bpa_bulk.py

@@ -2,7 +2,6 @@ import sempy.fabric as fabric
 import pandas as pd
 import datetime
 from sempy_labs._helper_functions import (
-    resolve_lakehouse_name,
     save_as_delta_table,
     resolve_workspace_capacity,
     retry,
@@ -26,7 +25,7 @@ def run_model_bpa_bulk(
     rules: Optional[pd.DataFrame] = None,
     extended: bool = False,
     language: Optional[str] = None,
-    workspace: Optional[str | List[str]] = None,
+    workspace: Optional[str | UUID | List[str | UUID]] = None,
     skip_models: Optional[str | List[str]] = ["ModelBPA", "Fabric Capacity Metrics"],
     skip_models_in_workspace: Optional[dict] = None,
 ):
@@ -44,8 +43,8 @@ def run_model_bpa_bulk(
     language : str, default=None
         The language (code) in which the rules will appear. For example, specifying 'it-IT' will show the Rule Name, Category and Description in Italian.
         Defaults to None which resolves to English.
-    workspace : str | List[str], default=None
-        The workspace or list of workspaces to scan.
+    workspace : str | uuid.UUID | List[str | uuid.UUID], default=None
+        The workspace or list of workspaces to scan. Supports both the workspace name and the workspace id.
         Defaults to None which scans all accessible workspaces.
     skip_models : str | List[str], default=['ModelBPA', 'Fabric Capacity Metrics']
         The semantic models to always skip when running this analysis.
@@ -71,7 +70,7 @@ def run_model_bpa_bulk(
     output_table = "modelbparesults"
     lakeT = get_lakehouse_tables()
     lakeT_filt = lakeT[lakeT["Table Name"] == output_table]
-    if len(lakeT_filt) == 0:
+    if lakeT_filt.empty:
         runId = 1
     else:
         max_run_id = _get_column_aggregate(table_name=output_table)
@@ -84,14 +83,14 @@ def run_model_bpa_bulk(
     if workspace is None:
         dfW_filt = dfW.copy()
     else:
-        dfW_filt = dfW[dfW["Name"].isin(workspace)]
+        dfW_filt = dfW[(dfW["Name"].isin(workspace)) | (dfW["Id"].isin(workspace))]
 
-    if len(dfW_filt) == 0:
+    if dfW_filt.empty:
         raise ValueError(
             f"{icons.red_dot} There are no valid workspaces to assess. This is likely due to not having proper permissions to the workspace(s) entered in the 'workspace' parameter."
         )
 
-    for i, r in dfW_filt.iterrows():
+    for _, r in dfW_filt.iterrows():
         wksp = r["Name"]
         wksp_id = r["Id"]
         capacity_id, capacity_name = resolve_workspace_capacity(workspace=wksp)
@@ -106,7 +105,7 @@ def run_model_bpa_bulk(
             dfD = dfD[~dfD["Dataset Name"].isin(skip_models_wkspc)]
 
         # Exclude default semantic models
-        if len(dfD) > 0:
+        if not dfD.empty:
             dfI = fabric.list_items(workspace=wksp)
             filtered_df = dfI.groupby("Display Name").filter(
                 lambda x: set(["Warehouse", "SemanticModel"]).issubset(set(x["Type"]))
@@ -116,7 +115,7 @@ def run_model_bpa_bulk(
             skip_models.extend(default_semantic_models)
             dfD_filt = dfD[~dfD["Dataset Name"].isin(skip_models)]
 
-            if len(dfD_filt) > 0:
+            if not dfD_filt.empty:
                 for _, r2 in dfD_filt.iterrows():
                     dataset_id = r2["Dataset Id"]
                     dataset_name = r2["Dataset Name"]
@@ -159,7 +158,7 @@ def run_model_bpa_bulk(
                         )
                         print(e)
 
-                if len(df) == 0:
+                if df.empty:
                     print(
                         f"{icons.yellow_dot} No BPA results to save for the '{wksp}' workspace."
                     )

sempy_labs/_model_bpa_rules.py

@@ -556,7 +556,7 @@ def model_bpa_rules(
                 "Warning",
                 "Use the DIVIDE function for division",
                 lambda obj, tom: re.search(
-                    r"\]\s*\/(?!\/)(?!\*)\" or \"\)\s*\/(?!\/)(?!\*)",
+                    r"\]\s*\/(?!\/)(?!\*)|\)\s*\/(?!\/)(?!\*)",
                     obj.Expression,
                     flags=re.IGNORECASE,
                 ),

sempy_labs/admin/_basic_functions.py

@@ -350,14 +350,40 @@ def list_workspace_access_details(
     return df
 
 
+def _resolve_workspace_name(workspace_id: Optional[UUID] = None) -> str:
+    from sempy_labs._helper_functions import _get_fabric_context_setting
+    from sempy.fabric.exceptions import FabricHTTPException
+
+    if workspace_id is None:
+        workspace_id = _get_fabric_context_setting(name="trident.workspace.id")
+
+    try:
+        workspace_name = (
+            _base_api(
+                request=f"/v1/admin/workspaces/{workspace_id}", client="fabric_sp"
+            )
+            .json()
+            .get("name")
+        )
+    except FabricHTTPException:
+        raise ValueError(
+            f"{icons.red_dot} The '{workspace_id}' workspace was not found."
+        )
+    return workspace_name
+
+
 def _resolve_workspace_name_and_id(
     workspace: str | UUID,
 ) -> Tuple[str, UUID]:
 
-    from sempy_labs._helper_functions import resolve_workspace_name_and_id
+    from sempy_labs._helper_functions import _get_fabric_context_setting
 
     if workspace is None:
-        (workspace_name, workspace_id) = resolve_workspace_name_and_id()
+        workspace_id = _get_fabric_context_setting(name="trident.workspace.id")
+        workspace_name = _resolve_workspace_name(workspace_id)
+    elif _is_valid_uuid(workspace):
+        workspace_id = workspace
+        workspace_name = _resolve_workspace_name(workspace_id)
     else:
         dfW = list_workspaces(workspace=workspace)
         if not dfW.empty:

sempy_labs/admin/_reports.py

@@ -113,7 +113,7 @@ def list_report_users(report: str | UUID) -> pd.DataFrame:
     """
     Shows a list of users that have access to the specified report.
 
-    This is a wrapper function for the following API: `Admin - Reports GetDatasetUsersAsAdmin <https://learn.microsoft.com/rest/api/power-bi/admin/datasets-get-report-users-as-admin>`_.
+    This is a wrapper function for the following API: `Admin - Reports GetDatasetUsersAsAdmin <https://learn.microsoft.com/rest/api/power-bi/admin/reports-get-report-users-as-admin>`_.
 
     Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
 

sempy_labs/admin/_scanner.py

@@ -115,6 +115,4 @@ def scan_workspaces(
         client="fabric_sp",
     )
 
-    print(f"{icons.green_dot} Status: {scan_status}")
-
     return response.json()

sempy_labs/admin/_tenant.py

@@ -126,11 +126,10 @@ def list_capacity_tenant_settings_overrides(
             if capacity_id is None:
                 # If capacity_id is None, we access 'Overrides' -> 'tenantSettings'
                 for override in r.get("overrides", []):
+                    capacity_id = override.get("id")
                     tenant_settings = override.get("tenantSettings", [])
                     for setting in tenant_settings:
-                        data.append(
-                            create_new_data(setting)
-                        )  # No capacity_id needed here
+                        data.append(create_new_data(setting, capacity_id))
             else:
                 # If capacity_id is provided, we access 'value' directly for tenantSettings
                 for setting in r.get("value", []):
@@ -391,6 +390,7 @@ def list_workspaces_tenant_settings_overrides() -> pd.DataFrame:
     """
 
     columns = {
+        "Workspace Id": "string",
         "Setting Name": "string",
         "Title": "string",
         "Enabled": "bool",
@@ -409,8 +409,10 @@ def list_workspaces_tenant_settings_overrides() -> pd.DataFrame:
 
     for r in responses:
         for v in r.get("value", []):
+            workspace_id = v.get("id")
             for setting in v.get("tenantSettings", []):
                 new_data = {
+                    "Workspace Id": workspace_id,
                     "Setting Name": setting.get("settingName"),
                     "Title": setting.get("title"),
                     "Enabled": setting.get("enabled"),
@@ -447,6 +449,7 @@ def list_domain_tenant_settings_overrides() -> pd.DataFrame:
     """
 
     columns = {
+        "Domain Id": "string",
         "Setting Name": "string",
         "Title": "string",
         "Enabled": "bool",
@@ -466,8 +469,10 @@ def list_domain_tenant_settings_overrides() -> pd.DataFrame:
 
     for r in responses:
         for v in r.get("value", []):
+            domain_id = v.get("id")
             for setting in v.get("tenantSettings", []):
                 new_data = {
+                    "Domain Id": domain_id,
                     "Setting Name": setting.get("settingName"),
                     "Title": setting.get("title"),
                     "Enabled": setting.get("enabled"),

sempy_labs/directlake/_generate_shared_expression.py

@@ -13,6 +13,7 @@ def generate_shared_expression(
     item_name: Optional[str] = None,
     item_type: str = "Lakehouse",
     workspace: Optional[str | UUID] = None,
+    use_sql_endpoint: bool = True,
 ) -> str:
     """
     Dynamically generates the M expression used by a Direct Lake model for a given lakehouse/warehouse.
@@ -28,6 +29,9 @@ def generate_shared_expression(
         The Fabric workspace name or ID used by the item.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
+    use_sql_endpoint : bool, default=True
+        Whether to use the SQL Endpoint for the lakehouse/warehouse.
+        If False, the expression will be generated without using the SQL Endpoint.
 
     Returns
     -------
@@ -78,4 +82,8 @@ def generate_shared_expression(
     end_expr = "\nin\n\tdatabase"
     mid_expr = f'Sql.Database("{sqlEPCS}", "{sqlepid}")'
 
-    return f"{start_expr}{mid_expr}{end_expr}"
+    # Build DL/OL expression
+    if not use_sql_endpoint and item_type == "Lakehouse":
+        return f'AzureDataLakeStorage{{"server":"onelake.dfs.fabric.microsoft.com","path":"/{workspace_id}/{item_id}/"}}'
+    else:
+        return f"{start_expr}{mid_expr}{end_expr}"

sempy_labs/directlake/_update_directlake_model_lakehouse_connection.py

@@ -1,18 +1,64 @@
-import sempy.fabric as fabric
 from sempy_labs.directlake._generate_shared_expression import generate_shared_expression
 from sempy_labs._helper_functions import (
-    resolve_lakehouse_name,
     resolve_dataset_name_and_id,
     resolve_workspace_name_and_id,
     resolve_item_name_and_id,
     resolve_lakehouse_name_and_id,
 )
+from sempy._utils._log import log
 from sempy_labs.tom import connect_semantic_model
 from typing import Optional
 import sempy_labs._icons as icons
 from uuid import UUID
+import re
 
 
+def _extract_expression_list(expression):
+    """
+    Finds the pattern for DL/SQL & DL/OL expressions in the semantic model.
+    """
+
+    pattern_sql = r'Sql\.Database\s*\(\s*"([^"]+)"\s*,\s*"([^"]+)"\s*\)'
+    pattern_no_sql = r'AzureDataLakeStorage\s*\{\s*"server".*?:\s*onelake\.dfs\.fabric\.microsoft\.com"\s*,\s*"path"\s*:\s*"/([\da-fA-F-]+)\s*/\s*([\da-fA-F-]+)\s*/"\s*\}'
+
+    match_sql = re.search(pattern_sql, expression)
+    match_no_sql = re.search(pattern_no_sql, expression)
+
+    result = []
+    if match_sql:
+        value_1, value_2 = match_sql.groups()
+        result = [value_1, value_2, True]
+    elif match_no_sql:
+        value_1, value_2 = match_no_sql.groups()
+        result = [value_1, value_2, False]
+
+    return result
+
+
+def _get_direct_lake_expressions(
+    dataset: str | UUID, workspace: Optional[str | UUID] = None
+) -> dict:
+    """
+    Extracts a dictionary of all Direct Lake expressions from a semantic model.
+    """
+
+    from sempy_labs.tom import connect_semantic_model
+
+    result = {}
+
+    with connect_semantic_model(dataset=dataset, workspace=workspace) as tom:
+        for e in tom.model.Expressions:
+            expr_name = e.Name
+            expr = e.Expression
+
+            list_values = _extract_expression_list(expr)
+            if list_values:
+                result[expr_name] = list_values
+
+    return result
+
+
+@log
 def update_direct_lake_model_lakehouse_connection(
     dataset: str | UUID,
     workspace: Optional[str | UUID] = None,
@@ -39,41 +85,23 @@ def update_direct_lake_model_lakehouse_connection(
         or if no lakehouse attached, resolves to the workspace of the notebook.
     """
 
-    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
-    (dataset_name, dataset_id) = resolve_dataset_name_and_id(dataset, workspace_id)
-
-    (lakehouse_name, lakehouse_id) = resolve_lakehouse_name_and_id(
-        lakehouse=lakehouse, workspace=lakehouse_workspace
-    )
-
-    icons.sll_tags.append("UpdateDLConnection")
-
-    shEx = generate_shared_expression(
-        item_name=lakehouse, item_type="Lakehouse", workspace=lakehouse_workspace
-    )
-
-    with connect_semantic_model(
-        dataset=dataset_id, readonly=False, workspace=workspace_id
-    ) as tom:
-
-        if not tom.is_direct_lake():
-            raise ValueError(
-                f"{icons.red_dot} The '{dataset_name}' semantic model is not in Direct Lake. This function is only applicable to Direct Lake semantic models."
-            )
-
-        tom.model.Expressions["DatabaseQuery"].Expression = shEx
-
-    print(
-        f"{icons.green_dot} The expression in the '{dataset_name}' semantic model has been updated to point to the '{lakehouse}' lakehouse in the '{lakehouse_workspace}' workspace."
+    update_direct_lake_model_connection(
+        dataset=dataset,
+        workspace=workspace,
+        source=lakehouse,
+        source_type="Lakehouse",
+        source_workspace=lakehouse_workspace,
     )
 
 
+@log
 def update_direct_lake_model_connection(
     dataset: str | UUID,
     workspace: Optional[str | UUID] = None,
     source: Optional[str] = None,
     source_type: str = "Lakehouse",
     source_workspace: Optional[str | UUID] = None,
+    use_sql_endpoint: bool = True,
 ):
     """
     Remaps a Direct Lake semantic model's SQL Endpoint connection to a new lakehouse/warehouse.
@@ -95,7 +123,14 @@ def update_direct_lake_model_connection(
         The Fabric workspace name or ID used by the lakehouse/warehouse.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
+    use_sql_endpoint : bool, default=True
+        If True, the SQL Endpoint will be used for the connection.
+        If False, Direct Lake over OneLake will be used.
     """
+    if use_sql_endpoint:
+        icons.sll_tags.append("UpdateDLConnection_SQL")
+    else:
+        icons.sll_tags.append("UpdateDLConnection_DLOL")
 
     (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
     (dataset_name, dataset_id) = resolve_dataset_name_and_id(dataset, workspace_id)
@@ -119,12 +154,16 @@ def update_direct_lake_model_connection(
             item=source, type=source_type, workspace=source_workspace
         )
 
-    icons.sll_tags.append("UpdateDLConnection")
-
-    shEx = generate_shared_expression(
-        item_name=source_name, item_type=source_type, workspace=source_workspace
+    shared_expression = generate_shared_expression(
+        item_name=source_name,
+        item_type=source_type,
+        workspace=source_workspace,
+        use_sql_endpoint=use_sql_endpoint,
     )
 
+    expression_dict = _get_direct_lake_expressions(dataset=dataset, workspace=workspace)
+    expressions = list(expression_dict.keys())
+
     with connect_semantic_model(
         dataset=dataset_id, readonly=False, workspace=workspace_id
     ) as tom:
@@ -134,8 +173,15 @@ def update_direct_lake_model_connection(
                 f"{icons.red_dot} The '{dataset_name}' semantic model within the '{workspace_name}' workspace is not in Direct Lake. This function is only applicable to Direct Lake semantic models."
             )
 
-        tom.model.Expressions["DatabaseQuery"].Expression = shEx
+        # Update the single connection expression
+        if len(expressions) == 1:
+            expr = expressions[0]
+            tom.model.Expressions[expr].Expression = shared_expression
 
-    print(
-        f"{icons.green_dot} The expression in the '{dataset_name}' semantic model within the '{workspace_name}' workspace has been updated to point to the '{source}' {source_type.lower()} in the '{source_workspace}' workspace."
-    )
+            print(
+                f"{icons.green_dot} The expression in the '{dataset_name}' semantic model within the '{workspace_name}' workspace has been updated to point to the '{source}' {source_type.lower()} in the '{source_workspace}' workspace."
+            )
+        else:
+            print(
+                f"{icons.info} Multiple expressions found in the model. Please use the update_direct_lake_partition_entity function to update specific tables."
+            )

sempy_labs/directlake/_update_directlake_partition_entity.py

@@ -8,11 +8,13 @@ from sempy_labs._helper_functions import (
     resolve_workspace_name_and_id,
     resolve_workspace_name,
 )
+from sempy._utils._log import log
 from typing import List, Optional, Union
 import sempy_labs._icons as icons
 from uuid import UUID
 
 
+@log
 def update_direct_lake_partition_entity(
     dataset: str | UUID,
     table_name: Union[str, List[str]],
@@ -96,6 +98,7 @@ def update_direct_lake_partition_entity(
             )
 
 
+@log
 def add_table_to_direct_lake_semantic_model(
     dataset: str | UUID,
     table_name: str,

sempy_labs/graph/_groups.py

@@ -6,6 +6,7 @@ from sempy_labs._helper_functions import (
     _create_dataframe,
     _update_dataframe_datatypes,
 )
+from sempy._utils._log import log
 import sempy_labs._icons as icons
 from typing import List, Literal
 
@@ -38,6 +39,7 @@ def resolve_group_id(group: str | UUID) -> UUID:
     return group_id
 
 
+@log
 def list_groups() -> pd.DataFrame:
     """
     Shows a list of groups and their properties.
@@ -158,6 +160,7 @@ def _get_group(group_id: UUID) -> pd.DataFrame:
     return df
 
 
+@log
 def list_group_members(group: str | UUID) -> pd.DataFrame:
     """
     Shows a list of the members of a group.
@@ -217,6 +220,7 @@ def list_group_members(group: str | UUID) -> pd.DataFrame:
     return df
 
 
+@log
 def list_group_owners(group: str | UUID) -> pd.DataFrame:
     """
     Shows a list of the owners of a group.
@@ -332,6 +336,7 @@ def _base_add_to_group(
     )
 
 
+@log
 def add_group_members(
     group: str | UUID,
     user: str | UUID | List[str | UUID],
@@ -376,6 +381,7 @@ def add_group_owners(
     _base_add_to_group(group=group, object=user, object_type="owners")
 
 
+@log
 def renew_group(group: str | UUID):
     """
     Renews the group.

sempy_labs/graph/_teams.py

@@ -1,5 +1,6 @@
 import pandas as pd
 from uuid import UUID
+from sempy._utils._log import log
 from sempy_labs._helper_functions import (
     _base_api,
     _create_dataframe,
@@ -7,6 +8,7 @@ from sempy_labs._helper_functions import (
 )
 
 
+@log
 def list_teams() -> pd.DataFrame:
     """
     Shows a list of teams and their properties.