semantic-link-labs 0.10.1__py3-none-any.whl → 0.11.0__py3-none-any.whl

sempy_labs/_model_bpa_rules.py

@@ -2,8 +2,10 @@ import sempy
 import pandas as pd
 import re
 from typing import Optional
+from sempy._utils._log import log
 
 
+@log
 def model_bpa_rules(
     dependencies: Optional[pd.DataFrame] = None,
     **kwargs,

sempy_labs/_mounted_data_factories.py

@@ -1,20 +1,19 @@
 import pandas as pd
-import json
 from typing import Optional
 from sempy_labs._helper_functions import (
-    resolve_workspace_name_and_id,
+    resolve_workspace_id,
     _base_api,
     _create_dataframe,
     _update_dataframe_datatypes,
-    resolve_item_id,
-    _decode_b64,
-    delete_item,
     get_item_definition,
+    delete_item,
 )
 
 from uuid import UUID
+from sempy._utils._log import log
 
 
+@log
 def list_mounted_data_factories(
     workspace: Optional[str | UUID] = None,
 ) -> pd.DataFrame:
@@ -36,7 +35,7 @@ def list_mounted_data_factories(
         A pandas dataframe showing a list of mounted data factories from the specified workspace.
     """
 
-    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    workspace_id = resolve_workspace_id(workspace)
 
     columns = {
         "Mounted Data Factory Name": "str",
@@ -50,6 +49,7 @@ def list_mounted_data_factories(
         uses_pagination=True,
     )
 
+    dfs = []
     for r in responses:
         for v in r.get("value", []):
             new_data = {
@@ -58,13 +58,16 @@ def list_mounted_data_factories(
                 "Description": v.get("description"),
             }
 
-            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+            dfs.append(pd.DataFrame(new_data, index=[0]))
 
-    _update_dataframe_datatypes(dataframe=df, column_map=columns)
+    if dfs:
+        df = pd.concat(dfs, ignore_index=True)
+        _update_dataframe_datatypes(dataframe=df, column_map=columns)
 
     return df
 
 
+@log
 def get_mounted_data_factory_definition(
     mounted_data_factory: str | UUID, workspace: Optional[str | UUID] = None
 ) -> dict:
@@ -96,6 +99,7 @@ def get_mounted_data_factory_definition(
     )
 
 
+@log
 def delete_mounted_data_factory(
     mounted_data_factory: str | UUID, workspace: Optional[str | UUID]
 ):

sempy_labs/_notebooks.py

@@ -60,6 +60,7 @@ def _get_notebook_type(
     return file_extension[1:]
 
 
+@log
 def get_notebook_definition(
     notebook_name: str,
     workspace: Optional[str | UUID] = None,
@@ -172,6 +173,7 @@ def import_notebook_from_web(
         )
 
 
+@log
 def create_notebook(
     name: str,
     notebook_content: str,
@@ -227,6 +229,7 @@ def create_notebook(
     )
 
 
+@log
 def update_notebook_definition(
     name: str,
     notebook_content: str,

sempy_labs/_refresh_semantic_model.py

@@ -321,6 +321,7 @@ def cancel_dataset_refresh(
     )
 
 
+@log
 def get_semantic_model_refresh_history(
     dataset: str | UUID,
     request_id: Optional[str] = None,

sempy_labs/_semantic_models.py

@@ -13,8 +13,10 @@ from sempy_labs._helper_functions import (
 )
 import sempy_labs._icons as icons
 import re
+from sempy._utils._log import log
 
 
+@log
 def get_semantic_model_refresh_schedule(
     dataset: str | UUID, workspace: Optional[str | UUID] = None
 ) -> pd.DataFrame:
@@ -72,6 +74,7 @@ def get_semantic_model_refresh_schedule(
     return df
 
 
+@log
 def enable_semantic_model_scheduled_refresh(
     dataset: str | UUID,
     workspace: Optional[str | UUID] = None,
@@ -121,6 +124,7 @@ def enable_semantic_model_scheduled_refresh(
         )
 
 
+@log
 def delete_semantic_model(dataset: str | UUID, workspace: Optional[str | UUID] = None):
     """
     Deletes a semantic model.
@@ -140,6 +144,7 @@ def delete_semantic_model(dataset: str | UUID, workspace: Optional[str | UUID] =
     delete_item(item=dataset, type="SemanticModel", workspace=workspace)
 
 
+@log
 def update_semantic_model_refresh_schedule(
     dataset: str | UUID,
     days: Optional[str | List[str]] = None,
@@ -231,6 +236,7 @@ def update_semantic_model_refresh_schedule(
     )
 
 
+@log
 def list_semantic_model_datasources(
     dataset: str | UUID,
     workspace: Optional[str | UUID] = None,

sempy_labs/_spark.py

@@ -8,8 +8,10 @@ from sempy_labs._helper_functions import (
     _create_dataframe,
 )
 from uuid import UUID
+from sempy._utils._log import log
 
 
+@log
 def list_custom_pools(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     """
     Lists all `custom pools <https://learn.microsoft.com/fabric/data-engineering/create-custom-spark-pools>`_ within a workspace.
@@ -73,6 +75,7 @@ def list_custom_pools(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     return df
 
 
+@log
 def create_custom_pool(
     pool_name: str,
     node_size: str,
@@ -145,6 +148,7 @@ def create_custom_pool(
     )
 
 
+@log
 def update_custom_pool(
     pool_name: str,
     node_size: Optional[str] = None,
@@ -251,6 +255,7 @@ def update_custom_pool(
     )
 
 
+@log
 def delete_custom_pool(pool_name: str, workspace: Optional[str | UUID] = None):
     """
     Deletes a `custom pool <https://learn.microsoft.com/fabric/data-engineering/create-custom-spark-pools>`_ within a workspace.
@@ -286,6 +291,7 @@ def delete_custom_pool(pool_name: str, workspace: Optional[str | UUID] = None):
     )
 
 
+@log
 def get_spark_settings(
     workspace: Optional[str | UUID] = None, return_dataframe: bool = True
 ) -> pd.DataFrame | dict:
@@ -362,6 +368,7 @@ def get_spark_settings(
         return response.json()
 
 
+@log
 def update_spark_settings(
     automatic_log_enabled: Optional[bool] = None,
     high_concurrency_enabled: Optional[bool] = None,

sempy_labs/_sql_endpoints.py

@@ -7,10 +7,13 @@ from sempy_labs._helper_functions import (
     resolve_workspace_name_and_id,
     resolve_item_name_and_id,
     _update_dataframe_datatypes,
+    resolve_workspace_id,
 )
 import sempy_labs._icons as icons
+from sempy._utils._log import log
 
 
+@log
 def list_sql_endpoints(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     """
     Shows the SQL endpoints within a workspace.
@@ -35,12 +38,13 @@ def list_sql_endpoints(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     }
     df = _create_dataframe(columns=columns)
 
-    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    workspace_id = resolve_workspace_id(workspace)
 
     responses = _base_api(
         request=f"/v1/workspaces/{workspace_id}/sqlEndpoints", uses_pagination=True
     )
 
+    dfs = []
     for r in responses:
         for v in r.get("value", []):
 
@@ -49,11 +53,15 @@ def list_sql_endpoints(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
                 "SQL Endpoint Name": v.get("displayName"),
                 "Description": v.get("description"),
             }
-            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+            dfs.append(pd.DataFrame(new_data, index=[0]))
+
+    if dfs:
+        df = pd.concat(dfs, ignore_index=True)
 
     return df
 
 
+@log
 def refresh_sql_endpoint_metadata(
     item: str | UUID,
     type: Literal["Lakehouse", "MirroredDatabase"],
@@ -149,37 +157,52 @@ def refresh_sql_endpoint_metadata(
         "Error Message": "string",
     }
 
-    df = pd.json_normalize(result)
-
-    # Extract error code and message, set to None if no error
-    df['Error Code'] = df.get('error.errorCode', None)
-    df['Error Message'] = df.get('error.message', None)
-
-    # Friendly column renaming
-    df.rename(columns={
-        'tableName': 'Table Name',
-        'startDateTime': 'Start Time',
-        'endDateTime': 'End Time',
-        'status': 'Status',
-        'lastSuccessfulSyncDateTime': 'Last Successful Sync Time'
-    }, inplace=True)
-
-    # Drop the original 'error' column if present
-    df.drop(columns=[col for col in ['error'] if col in df.columns], inplace=True)
+    if result:
+        df = pd.json_normalize(result)
+
+        # Extract error code and message, set to None if no error
+        df["Error Code"] = df.get("error.errorCode", None)
+        df["Error Message"] = df.get("error.message", None)
+
+        # Friendly column renaming
+        df.rename(
+            columns={
+                "tableName": "Table Name",
+                "startDateTime": "Start Time",
+                "endDateTime": "End Time",
+                "status": "Status",
+                "lastSuccessfulSyncDateTime": "Last Successful Sync Time",
+            },
+            inplace=True,
+        )
 
-    # Optional: Reorder columns
-    column_order = [
-        'Table Name', 'Status', 'Start Time', 'End Time',
-        'Last Successful Sync Time', 'Error Code', 'Error Message'
-    ]
-    df = df[column_order]
+        # Drop the original 'error' column if present
+        df.drop(columns=[col for col in ["error"] if col in df.columns], inplace=True)
+
+        # Optional: Reorder columns
+        column_order = [
+            "Table Name",
+            "Status",
+            "Start Time",
+            "End Time",
+            "Last Successful Sync Time",
+            "Error Code",
+            "Error Message",
+        ]
+        df = df[column_order]
+
+        printout = f"{icons.green_dot} The metadata of the SQL endpoint for the '{item_name}' {type.lower()} within the '{workspace_name}' workspace has been refreshed"
+        if tables:
+            print(f"{printout} for the following tables: {tables}.")
+        else:
+            print(f"{printout} for all tables.")
+    else:
+        # If the target item has no tables to refresh the metadata for
+        df = pd.DataFrame(columns=columns.keys())
+        print(
+            f"{icons.yellow_dot} The SQL endpoint '{item_name}' {type.lower()} within the '{workspace_name}' workspace has no tables to refresh..."
+        )
 
     _update_dataframe_datatypes(df, columns)
 
-    printout = f"{icons.green_dot} The metadata of the SQL endpoint for the '{item_name}' {type.lower()} within the '{workspace_name}' workspace has been refreshed"
-    if tables:
-        print(f"{printout} for the following tables: {tables}.")
-    else:
-        print(f"{printout} for all tables.")
-
     return df

sempy_labs/_sqldatabase.py

@@ -1,5 +1,5 @@
 from sempy_labs._helper_functions import (
-    resolve_workspace_name_and_id,
+    resolve_workspace_id,
     _base_api,
     _create_dataframe,
     _update_dataframe_datatypes,
@@ -9,8 +9,10 @@ from sempy_labs._helper_functions import (
 import pandas as pd
 from typing import Optional
 from uuid import UUID
+from sempy._utils._log import log
 
 
+@log
 def create_sql_database(
     name: str, description: Optional[str] = None, workspace: Optional[str | UUID] = None
 ):
@@ -36,6 +38,7 @@ def create_sql_database(
     )
 
 
+@log
 def delete_sql_database(
     sql_database: str | UUID, workspace: Optional[str | UUID] = None
 ):
@@ -57,6 +60,7 @@ def delete_sql_database(
     delete_item(item=sql_database, type="SQLDatabase", workspace=workspace)
 
 
+@log
 def list_sql_databases(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     """
     Lists all SQL databases in the Fabric workspace.
@@ -78,7 +82,7 @@ def list_sql_databases(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
         A pandas dataframe showing a list of SQL databases in the Fabric workspace.
     """
 
-    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    workspace_id = resolve_workspace_id(workspace)
 
     columns = {
         "SQL Database Name": "string",
@@ -96,6 +100,7 @@ def list_sql_databases(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
         client="fabric_sp",
     )
 
+    dfs = []
     for r in responses:
         for v in r.get("value", []):
             prop = v.get("properties", {})
@@ -108,13 +113,16 @@ def list_sql_databases(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
                 "Server FQDN": prop.get("serverFqdn"),
             }
 
-            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+            dfs.append(pd.DataFrame(new_data, index=[0]))
 
-    _update_dataframe_datatypes(dataframe=df, column_map=columns)
+    if dfs:
+        df = pd.concat(dfs, ignore_index=True)
+        _update_dataframe_datatypes(dataframe=df, column_map=columns)
 
     return df
 
 
+@log
 def get_sql_database_tables(
     sql_database: str | UUID, workspace: Optional[str | UUID] = None
 ) -> pd.DataFrame:
@@ -150,6 +158,7 @@ def get_sql_database_tables(
     return df
 
 
+@log
 def get_sql_database_columns(
     sql_database: str | UUID, workspace: Optional[str | UUID] = None
 ) -> pd.DataFrame:

sempy_labs/_tags.py

@@ -10,8 +10,10 @@ import pandas as pd
 from typing import Optional, List
 from uuid import UUID
 import sempy_labs._icons as icons
+from sempy._utils._log import log
 
 
+@log
 def list_tags() -> pd.DataFrame:
     """
     Shows a list of all the tenant's tags.
@@ -39,7 +41,6 @@ def list_tags() -> pd.DataFrame:
     )
 
     dfs = []
-
     for r in responses:
         for v in r.get("value", []):
             new_data = {
@@ -55,6 +56,7 @@ def list_tags() -> pd.DataFrame:
     return df
 
 
+@log
 def resolve_tags(tags: str | List[str]) -> List[str]:
     """
     Resolves the tags to a list of strings.
@@ -92,6 +94,7 @@ def resolve_tags(tags: str | List[str]) -> List[str]:
     return tag_list
 
 
+@log
 def apply_tags(
     item: str | UUID,
     type: str,
@@ -143,6 +146,7 @@ def apply_tags(
     )
 
 
+@log
 def unapply_tags(
     item: str | UUID,
     type: str,

sempy_labs/_user_delegation_key.py

@@ -4,8 +4,10 @@ from sempy_labs._helper_functions import (
 )
 from datetime import datetime, timedelta, timezone
 import xml.etree.ElementTree as ET
+from sempy._utils._log import log
 
 
+@log
 def get_user_delegation_key():
     """
     Gets a key that can be used to sign a user delegation SAS (shared access signature). A user delegation SAS grants access to Azure Blob Storage resources by using Microsoft Entra credentials.

sempy_labs/_variable_libraries.py

@@ -1,5 +1,4 @@
 from sempy_labs._helper_functions import (
-    resolve_workspace_name_and_id,
     resolve_workspace_id,
     _base_api,
     _create_dataframe,
@@ -9,8 +8,10 @@ from sempy_labs._helper_functions import (
 import pandas as pd
 from typing import Optional
 from uuid import UUID
+from sempy._utils._log import log
 
 
+@log
 def list_variable_libraries(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     """
     Shows the variable libraries within a workspace.
@@ -68,6 +69,7 @@ def list_variable_libraries(workspace: Optional[str | UUID] = None) -> pd.DataFr
     return df
 
 
+@log
 def delete_variable_library(
     variable_library: str | UUID, workspace: Optional[str | UUID] = None
 ):

sempy_labs/_warehouses.py

@@ -4,13 +4,16 @@ from sempy_labs._helper_functions import (
     _create_dataframe,
     _update_dataframe_datatypes,
     delete_item,
+    resolve_workspace_id,
 )
 import pandas as pd
 from typing import Optional
 import sempy_labs._icons as icons
 from uuid import UUID
+from sempy._utils._log import log
 
 
+@log
 def create_warehouse(
     warehouse: str,
     description: Optional[str] = None,
@@ -68,6 +71,7 @@ def create_warehouse(
     return result.get("id")
 
 
+@log
 def list_warehouses(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     """
     Shows the warehouses within a workspace.
@@ -99,7 +103,7 @@ def list_warehouses(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     }
     df = _create_dataframe(columns=columns)
 
-    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    workspace_id = resolve_workspace_id(workspace)
 
     responses = _base_api(
         request=f"/v1/workspaces/{workspace_id}/warehouses",
@@ -107,6 +111,7 @@ def list_warehouses(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
         client="fabric_sp",
     )
 
+    dfs = []
     for r in responses:
         for v in r.get("value", []):
             prop = v.get("properties", {})
@@ -119,13 +124,16 @@ def list_warehouses(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
                 "Created Date": prop.get("createdDate"),
                 "Last Updated Time": prop.get("lastUpdatedTime"),
             }
-            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+            dfs.append(pd.DataFrame(new_data, index=[0]))
 
-    _update_dataframe_datatypes(dataframe=df, column_map=columns)
+    if dfs:
+        df = pd.concat(dfs, ignore_index=True)
+        _update_dataframe_datatypes(dataframe=df, column_map=columns)
 
     return df
 
 
+@log
 def delete_warehouse(name: str | UUID, workspace: Optional[str | UUID] = None):
     """
     Deletes a Fabric warehouse.
@@ -145,6 +153,7 @@ def delete_warehouse(name: str | UUID, workspace: Optional[str | UUID] = None):
     delete_item(item=name, type="Warehouse", workspace=workspace)
 
 
+@log
 def get_warehouse_tables(
     warehouse: str | UUID, workspace: Optional[str | UUID] = None
 ) -> pd.DataFrame:
@@ -180,6 +189,7 @@ def get_warehouse_tables(
     return df
 
 
+@log
 def get_warehouse_columns(
     warehouse: str | UUID, workspace: Optional[str | UUID] = None
 ) -> pd.DataFrame:

sempy_labs/_workloads.py

@@ -7,8 +7,10 @@ from sempy_labs._helper_functions import (
     _create_dataframe,
 )
 from uuid import UUID
+from sempy._utils._log import log
 
 
+@log
 def list_workloads(capacity: str | UUID, **kwargs) -> pd.DataFrame:
     """
     Returns the current state of the specified capacity workloads.
@@ -59,6 +61,7 @@ def list_workloads(capacity: str | UUID, **kwargs) -> pd.DataFrame:
     return df
 
 
+@log
 def patch_workload(
     capacity: str | UUID,
     workload_name: str,

sempy_labs/_workspace_identity.py

@@ -5,8 +5,10 @@ from sempy_labs._helper_functions import (
 from typing import Optional
 import sempy_labs._icons as icons
 from uuid import UUID
+from sempy._utils._log import log
 
 
+@log
 def provision_workspace_identity(workspace: Optional[str | UUID] = None):
     """
     Provisions a workspace identity for a workspace.
@@ -38,6 +40,7 @@ def provision_workspace_identity(workspace: Optional[str | UUID] = None):
     )
 
 
+@log
 def deprovision_workspace_identity(workspace: Optional[str | UUID] = None):
     """
     Deprovisions a workspace identity for a workspace.

sempy_labs/_workspaces.py

@@ -8,8 +8,10 @@ from sempy_labs._helper_functions import (
     _create_dataframe,
 )
 from uuid import UUID
+from sempy._utils._log import log
 
 
+@log
 def delete_user_from_workspace(
     email_address: str, workspace: Optional[str | UUID] = None
 ):
@@ -42,6 +44,7 @@ def delete_user_from_workspace(
     )
 
 
+@log
 def update_workspace_user(
     email_address: str,
     role_name: str,
@@ -102,6 +105,7 @@ def update_workspace_user(
     )
 
 
+@log
 def list_workspace_users(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     """
     A list of all the users of a workspace and their roles.
@@ -140,6 +144,7 @@ def list_workspace_users(workspace: Optional[str | UUID] = None) -> pd.DataFrame
         client="fabric_sp",
     )
 
+    dfs = []
     for r in responses:
         for v in r.get("value", []):
             p = v.get("principal", {})
@@ -150,11 +155,15 @@ def list_workspace_users(workspace: Optional[str | UUID] = None) -> pd.DataFrame
                 "Role": v.get("role"),
                 "Email Address": p.get("userDetails", {}).get("userPrincipalName"),
             }
-            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+            dfs.append(pd.DataFrame(new_data, index=[0]))
+
+    if dfs:
+        df = pd.concat(dfs, ignore_index=True)
 
     return df
 
 
+@log
 def add_user_to_workspace(
     email_address: str,
     role_name: str,
@@ -213,6 +222,7 @@ def add_user_to_workspace(
     )
 
 
+@log
 def assign_workspace_to_capacity(
     capacity: str | UUID,
     workspace: Optional[str | UUID] = None,
@@ -255,6 +265,7 @@ def assign_workspace_to_capacity(
     )
 
 
+@log
 def unassign_workspace_from_capacity(workspace: Optional[str | UUID] = None):
     """
     Unassigns a workspace from its assigned capacity.
@@ -284,6 +295,7 @@ def unassign_workspace_from_capacity(workspace: Optional[str | UUID] = None):
     )
 
 
+@log
 def list_workspace_role_assignments(
     workspace: Optional[str | UUID] = None,
 ) -> pd.DataFrame:
@@ -337,6 +349,7 @@ def list_workspace_role_assignments(
     return df
 
 
+@log
 def delete_workspace(workspace: Optional[str | UUID] = None):
     """
     Deletes a workspace.