semantic-link-labs 0.11.3__py3-none-any.whl → 0.12.0__py3-none-any.whl

sempy_labs/_model_bpa.py

@@ -3,8 +3,8 @@ import pandas as pd
 import warnings
 import datetime
 from IPython.display import display, HTML
-from ._model_dependencies import get_model_calc_dependencies
-from ._helper_functions import (
+from sempy_labs._model_dependencies import get_model_calc_dependencies
+from sempy_labs._helper_functions import (
     format_dax_object_name,
     create_relationship_name,
     save_as_delta_table,
@@ -15,9 +15,9 @@ from ._helper_functions import (
     resolve_workspace_name_and_id,
     _create_spark_session,
 )
-from .lakehouse import get_lakehouse_tables, lakehouse_attached
-from .tom import connect_semantic_model
-from ._model_bpa_rules import model_bpa_rules
+from sempy_labs.lakehouse import get_lakehouse_tables, lakehouse_attached
+from sempy_labs.tom import connect_semantic_model
+from sempy_labs._model_bpa_rules import model_bpa_rules
 from typing import Optional
 from sempy._utils._log import log
 import sempy_labs._icons as icons
@@ -391,7 +391,7 @@ def run_model_bpa(
         lakeT = get_lakehouse_tables()
         lakeT_filt = lakeT[lakeT["Table Name"] == delta_table_name]
 
-        dfExport["Severity"].replace(icons.severity_mapping, inplace=True)
+        dfExport["Severity"] = dfExport["Severity"].replace(icons.severity_mapping)
 
         if len(lakeT_filt) == 0:
             runId = 1

sempy_labs/_model_bpa_bulk.py

@@ -1,7 +1,7 @@
 import sempy.fabric as fabric
 import pandas as pd
 import datetime
-from ._helper_functions import (
+from sempy_labs._helper_functions import (
     save_as_delta_table,
     resolve_workspace_capacity,
     retry,
@@ -9,11 +9,11 @@ from ._helper_functions import (
     resolve_workspace_id,
     resolve_lakehouse_name_and_id,
 )
-from .lakehouse import (
+from sempy_labs.lakehouse import (
     get_lakehouse_tables,
     lakehouse_attached,
 )
-from ._model_bpa import run_model_bpa
+from sempy_labs._model_bpa import run_model_bpa
 from typing import Optional, List
 from sempy._utils._log import log
 import sempy_labs._icons as icons

sempy_labs/_model_dependencies.py

@@ -1,6 +1,6 @@
 import sempy.fabric as fabric
 import pandas as pd
-from ._helper_functions import (
+from sempy_labs._helper_functions import (
     format_dax_object_name,
     resolve_dataset_name_and_id,
     resolve_workspace_name_and_id,

sempy_labs/_mounted_data_factories.py

@@ -1,11 +1,11 @@
 import pandas as pd
 from typing import Optional
-from ._helper_functions import (
+from sempy_labs._helper_functions import (
     resolve_workspace_id,
     _base_api,
     _create_dataframe,
     _update_dataframe_datatypes,
-    get_item_definition,
+    _get_item_definition,
     delete_item,
 )
 
@@ -91,7 +91,7 @@ def get_mounted_data_factory_definition(
         The 'mountedDataFactory-content.json' file from the mounted data factory definition.
     """
 
-    return get_item_definition(
+    return _get_item_definition(
         item=mounted_data_factory,
         type="MountedDataFactory",
         workspace=workspace,

sempy_labs/_notebooks.py

@@ -17,6 +17,7 @@ from sempy_labs._helper_functions import (
 from sempy.fabric.exceptions import FabricHTTPException
 from os import PathLike
 from uuid import UUID
+import os
 
 _notebook_prefix = "notebook-content."
 
@@ -315,7 +316,7 @@ def list_notebooks(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     Returns
     -------
     pandas.DataFrame
-        A pandas dataframe showing the SQL endpoints within a workspace.
+        A pandas dataframe showing the notebooks within a workspace.
     """
 
     columns = {

sempy_labs/_query_scale_out.py

@@ -1,6 +1,6 @@
 import sempy.fabric as fabric
 import pandas as pd
-from ._helper_functions import (
+from sempy_labs._helper_functions import (
     resolve_workspace_name_and_id,
     resolve_dataset_name_and_id,
     _update_dataframe_datatypes,
@@ -406,7 +406,7 @@ def set_workspace_default_storage_format(
     (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     # Check current storage format
-    dfW = fabric.list_workspaces(filter=f"name eq '{workspace_name}'")
+    dfW = fabric.list_workspaces(filter=f"id eq '{workspace_id}'")
     if len(dfW) == 0:
         raise ValueError()
     current_storage_format = dfW["Default Dataset Storage Format"].iloc[0]

sempy_labs/_refresh_semantic_model.py

@@ -1,6 +1,6 @@
 import sempy.fabric as fabric
 import time
-from ._helper_functions import (
+from sempy_labs._helper_functions import (
     resolve_workspace_name_and_id,
     _get_partition_map,
     _process_and_display_chart,

sempy_labs/_semantic_models.py

@@ -1,7 +1,7 @@
 from uuid import UUID
 from typing import Optional, List
 import pandas as pd
-from ._helper_functions import (
+from sempy_labs._helper_functions import (
     _create_dataframe,
     _base_api,
     _update_dataframe_datatypes,
@@ -23,6 +23,8 @@ def get_semantic_model_refresh_schedule(
     """
     Gets the refresh schedule for the specified dataset from the specified workspace.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     dataset : str | uuid.UUID
@@ -38,7 +40,7 @@ def get_semantic_model_refresh_schedule(
         Shows the refresh schedule for the specified dataset from the specified workspace.
     """
 
-    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    workspace_id = resolve_workspace_id(workspace)
     (dataset_name, dataset_id) = resolve_dataset_name_and_id(dataset, workspace)
 
     columns = {
@@ -60,7 +62,8 @@ def get_semantic_model_refresh_schedule(
     df = _create_dataframe(columns)
 
     result = _base_api(
-        request=f"/v1.0/myorg/groups/{workspace_id}/datasets/{dataset_id}/refreshSchedule"
+        request=f"/v1.0/myorg/groups/{workspace_id}/datasets/{dataset_id}/refreshSchedule",
+        client="fabric_sp",
     ).json()
 
     df = (
@@ -83,6 +86,8 @@ def enable_semantic_model_scheduled_refresh(
     """
     Enables the scheduled refresh for the specified dataset from the specified workspace.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     dataset : str | uuid.UUID
@@ -117,6 +122,7 @@ def enable_semantic_model_scheduled_refresh(
             request=f"/v1.0/myorg/groups/{workspace_id}/datasets/{dataset_id}/refreshSchedule",
             method="patch",
             payload=payload,
+            client="fabric_sp",
         )
 
         print(
@@ -131,6 +137,8 @@ def delete_semantic_model(dataset: str | UUID, workspace: Optional[str | UUID] =
 
     This is a wrapper function for the following API: `Items - Delete Semantic Model <https://learn.microsoft.com/rest/api/fabric/semanticmodel/items/delete-semantic-model>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     dataset: str | uuid.UUID
@@ -157,6 +165,8 @@ def update_semantic_model_refresh_schedule(
 
     This is a wrapper function for the following API: `Datasets - Update Refresh Schedule In Group <https://learn.microsoft.com/rest/api/power-bi/datasets/update-refresh-schedule-in-group>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     dataset : str | uuid.UUID
@@ -247,6 +257,8 @@ def list_semantic_model_datasources(
 
     This is a wrapper function for the following API: `Datasets - Get Datasources In Group <https://learn.microsoft.com/rest/api/power-bi/datasets/get-datasources-in-group>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     dataset : str | uuid.UUID

sempy_labs/_spark.py

@@ -1,7 +1,7 @@
 import pandas as pd
 import sempy_labs._icons as icons
 from typing import Optional
-from ._helper_functions import (
+from sempy_labs._helper_functions import (
     resolve_workspace_name_and_id,
     _update_dataframe_datatypes,
     _base_api,

sempy_labs/_sql.py

@@ -3,7 +3,7 @@ from typing import Optional, Union, List
 from sempy._utils._log import log
 import struct
 from itertools import chain, repeat
-from ._helper_functions import (
+from sempy_labs._helper_functions import (
     resolve_lakehouse_name_and_id,
     resolve_item_name_and_id,
     resolve_workspace_name_and_id,
@@ -39,7 +39,7 @@ class ConnectBase:
         timeout: Optional[int] = None,
         endpoint_type: str = "warehouse",
     ):
-        from sempy.fabric._token_provider import SynapseTokenProvider
+        from sempy.fabric._credentials import get_access_token
         import pyodbc
 
         (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
@@ -82,7 +82,7 @@ class ConnectBase:
             )
 
         # Set up the connection string
-        access_token = SynapseTokenProvider()("sql")
+        access_token = get_access_token("sql").token
         tokenstruct = _bytes2mswin_bstr(access_token.encode())
         if endpoint_type == "sqldatabase":
             conn_str = f"DRIVER={{ODBC Driver 18 for SQL Server}};SERVER={tds_endpoint};DATABASE={resource_name}-{resource_id};Encrypt=Yes;"

sempy_labs/_sql_endpoints.py

@@ -1,7 +1,7 @@
 from typing import Optional, Literal
 from uuid import UUID
 import pandas as pd
-from ._helper_functions import (
+from sempy_labs._helper_functions import (
     _base_api,
     _create_dataframe,
     resolve_workspace_name_and_id,
@@ -73,6 +73,8 @@ def refresh_sql_endpoint_metadata(
 
     This is a wrapper function for the following API: `Items - Refresh Sql Endpoint Metadata <https://learn.microsoft.com/rest/api/fabric/sqlendpoint/items/refresh-sql-endpoint-metadata>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     item : str | uuid.UUID
@@ -140,7 +142,7 @@ def refresh_sql_endpoint_metadata(
         }
 
     result = _base_api(
-        request=f"v1/workspaces/{workspace_id}/sqlEndpoints/{sql_endpoint_id}/refreshMetadata?preview=true",
+        request=f"v1/workspaces/{workspace_id}/sqlEndpoints/{sql_endpoint_id}/refreshMetadata",
         method="post",
         client="fabric_sp",
         status_codes=[200, 202],
@@ -159,7 +161,7 @@ def refresh_sql_endpoint_metadata(
     }
 
     if result:
-        df = pd.json_normalize(result)
+        df = pd.json_normalize(result.get("value"))
 
         # Extract error code and message, set to None if no error
         df["Error Code"] = df.get("error.errorCode", None)

sempy_labs/_sqldatabase.py

@@ -1,4 +1,4 @@
-from ._helper_functions import (
+from sempy_labs._helper_functions import (
     resolve_workspace_id,
     _base_api,
     _create_dataframe,
@@ -21,6 +21,8 @@ def create_sql_database(
 
     This is a wrapper function for the following API: `Items - Create SQL Database <https://learn.microsoft.com/rest/api/fabric/sqldatabase/items/create-sql-database>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     name: str
@@ -47,6 +49,8 @@ def delete_sql_database(
 
     This is a wrapper function for the following API: `Items - Delete SQL Database <https://learn.microsoft.com/rest/api/fabric/sqldatabase/items/delete-sql-database>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     sql_database: str | uuid.UUID

sempy_labs/_tags.py

@@ -1,4 +1,4 @@
-from ._helper_functions import (
+from sempy_labs._helper_functions import (
     _base_api,
     _create_dataframe,
     _update_dataframe_datatypes,
@@ -62,6 +62,8 @@ def resolve_tags(tags: str | List[str]) -> List[str]:
     """
     Resolves the tags to a list of strings.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     tags : str | List[str]