semantic-link-labs 0.4.1__py3-none-any.whl → 0.5.0__py3-none-any.whl

sempy_labs/_dax.py

@@ -7,7 +7,7 @@ from sempy._utils._log import log
 
 
 @log
-def run_dax(
+def evaluate_dax_impersonation(
     dataset: str,
     dax_query: str,
     user_name: Optional[str] = None,
@@ -27,7 +27,6 @@ def run_dax(
         The DAX query.
     user_name : str
         The user name (i.e. hello@goodbye.com).
-        Defaults to None which resolves to no user impersonation.
     workspace : str, default=None
         The Fabric workspace name.
         Defaults to None which resolves to the workspace of the attached lakehouse
@@ -49,13 +48,10 @@ def run_dax(
 
     dataset_id = resolve_dataset_id(dataset=dataset, workspace=workspace)
 
-    if user_name is None:
-        request_body = {"queries": [{"query": dax_query}]}
-    else:
-        request_body = {
-            "queries": [{"query": dax_query}],
-            "impersonatedUserName": user_name,
-        }
+    request_body = {
+        "queries": [{"query": dax_query}],
+        "impersonatedUserName": user_name
+    }
 
     client = fabric.PowerBIRestClient()
     response = client.post(

sempy_labs/_generate_semantic_model.py

@@ -23,22 +23,21 @@ def create_blank_semantic_model(
     ----------
     dataset : str
         Name of the semantic model.
-    compatibility_level : int
+    compatibility_level : int, default=1605
         The compatibility level of the semantic model.
-        Defaults to 1605.
     workspace : str, default=None
         The Fabric workspace name.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
     """
 
-    if workspace == None:
-        workspace_id = fabric.get_workspace_id()
-        workspace = fabric.resolve_workspace_name(workspace_id)
+    if workspace is None:
+        workspace = fabric.resolve_workspace_name()
 
-    if compatibility_level < 1500:
-        print(f"{icons.red_dot} Compatiblity level must be at least 1500.")
-        return
+    min_compat = 1500
+
+    if compatibility_level < min_compat:
+        raise ValueError(f"{icons.red_dot} Compatiblity level must be at least {min_compat}.")
 
     tmsl = f"""
   {{
@@ -66,7 +65,7 @@ def create_blank_semantic_model(
 
 
 def create_semantic_model_from_bim(
-    dataset: str, bim_file: str, workspace: Optional[str] = None
+    dataset: str, bim_file: dict, workspace: Optional[str] = None
 ):
     """
     Creates a new semantic model based on a Model.bim file.
@@ -75,7 +74,7 @@ def create_semantic_model_from_bim(
     ----------
     dataset : str
         Name of the semantic model.
-    bim_file : str
+    bim_file : dict
         The model.bim file.
     workspace : str, default=None
         The Fabric workspace name.
@@ -91,10 +90,7 @@ def create_semantic_model_from_bim(
     dfI_filt = dfI[(dfI["Display Name"] == dataset)]
 
     if len(dfI_filt) > 0:
-        print(
-            f"WARNING: '{dataset}' already exists as a semantic model in the '{workspace}' workspace."
-        )
-        return
+        raise ValueError(f"{icons.red_dot} '{dataset}' already exists as a semantic model in the '{workspace}' workspace.")
 
     client = fabric.FabricRestClient()
     defPBIDataset = {"version": "1.0", "settings": {}}
@@ -132,7 +128,7 @@ def create_semantic_model_from_bim(
 
     if response.status_code == 201:
         print(
-            f"The '{dataset}' semantic model has been created within the '{workspace}' workspace."
+            f"{icons.green_dot} The '{dataset}' semantic model has been created within the '{workspace}' workspace."
         )
         print(response.json())
     elif response.status_code == 202:
@@ -145,7 +141,7 @@ def create_semantic_model_from_bim(
             response_body = json.loads(response.content)
         response = client.get(f"/v1/operations/{operationId}/result")
         print(
-            f"The '{dataset}' semantic model has been created within the '{workspace}' workspace."
+            f"{icons.green_dot} The '{dataset}' semantic model has been created within the '{workspace}' workspace."
         )
         print(response.json())
 
@@ -179,11 +175,9 @@ def deploy_semantic_model(
 
     """
 
-    if workspace == None:
-        workspace_id = fabric.get_workspace_id()
-        workspace = fabric.resolve_workspace_name(workspace_id)
+    workspace = fabric.resolve_workspace_name(workspace)
 
-    if new_dataset_workspace == None:
+    if new_dataset_workspace is None:
         new_dataset_workspace = workspace
 
     if new_dataset is None:
@@ -258,7 +252,7 @@ def get_semantic_model_bim(
 
     if save_to_file_name is not None:
         lakeAttach = lakehouse_attached()
-        if lakeAttach == False:
+        if lakeAttach is False:
             print(
                 f"In order to save the model.bim file, a lakehouse must be attached to the notebook. Please attach a lakehouse to this notebook."
             )

sempy_labs/_helper_functions.py

@@ -1,3 +1,4 @@
+import sempy
 import sempy.fabric as fabric
 import re
 import pandas as pd
@@ -100,7 +101,7 @@ def resolve_report_id(report: str, workspace: Optional[str] = None):
         The ID of the Power BI report.
     """
 
-    if workspace == None:
+    if workspace is None:
         workspace_id = fabric.get_workspace_id()
         workspace = fabric.resolve_workspace_name(workspace_id)
 
@@ -128,7 +129,7 @@ def resolve_report_name(report_id: UUID, workspace: Optional[str] = None):
         The name of the Power BI report.
     """
 
-    if workspace == None:
+    if workspace is None:
         workspace_id = fabric.get_workspace_id()
         workspace = fabric.resolve_workspace_name(workspace_id)
 
@@ -158,7 +159,7 @@ def resolve_dataset_id(dataset: str, workspace: Optional[str] = None):
         The ID of the semantic model.
     """
 
-    if workspace == None:
+    if workspace is None:
         workspace_id = fabric.get_workspace_id()
         workspace = fabric.resolve_workspace_name(workspace_id)
 
@@ -188,7 +189,7 @@ def resolve_dataset_name(dataset_id: UUID, workspace: Optional[str] = None):
         The name of the semantic model.
     """
 
-    if workspace == None:
+    if workspace is None:
         workspace_id = fabric.get_workspace_id()
         workspace = fabric.resolve_workspace_name(workspace_id)
 
@@ -199,14 +200,15 @@ def resolve_dataset_name(dataset_id: UUID, workspace: Optional[str] = None):
     return obj
 
 
-def resolve_lakehouse_name(lakehouse_id: UUID, workspace: Optional[str] = None):
+def resolve_lakehouse_name(lakehouse_id: Optional[UUID] = None, workspace: Optional[str] = None):
     """
     Obtains the name of the Fabric lakehouse.
 
     Parameters
     ----------
-    lakehouse_id : UUID
+    lakehouse_id : UUID, default=None
         The name of the Fabric lakehouse.
+        Defaults to None which resolves to the lakehouse attached to the notebook.
     workspace : str, default=None
         The Fabric workspace name.
         Defaults to None which resolves to the workspace of the attached lakehouse
@@ -218,9 +220,12 @@ def resolve_lakehouse_name(lakehouse_id: UUID, workspace: Optional[str] = None):
         The name of the Fabric lakehouse.
     """
 
-    if workspace == None:
+    if workspace is None:
         workspace_id = fabric.get_workspace_id()
         workspace = fabric.resolve_workspace_name(workspace_id)
+    
+    if lakehouse_id is None:
+        lakehouse_id = fabric.get_lakehouse_id()
 
     obj = fabric.resolve_item_name(
         item_id=lakehouse_id, type="Lakehouse", workspace=workspace
@@ -248,7 +253,7 @@ def resolve_lakehouse_id(lakehouse: str, workspace: Optional[str] = None):
         The ID of the Fabric lakehouse.
     """
 
-    if workspace == None:
+    if workspace is None:
         workspace_id = fabric.get_workspace_id()
         workspace = fabric.resolve_workspace_name(workspace_id)
 
@@ -278,7 +283,7 @@ def get_direct_lake_sql_endpoint(dataset: str, workspace: Optional[str] = None)
         The ID of SQL Endpoint.
     """
 
-    if workspace == None:
+    if workspace is None:
         workspace_id = fabric.get_workspace_id()
         workspace = fabric.resolve_workspace_name(workspace_id)
 
@@ -286,10 +291,9 @@ def get_direct_lake_sql_endpoint(dataset: str, workspace: Optional[str] = None)
     dfP_filt = dfP[dfP["Mode"] == "DirectLake"]
 
     if len(dfP_filt) == 0:
-        print(
+        raise ValueError(
             f"The '{dataset}' semantic model in the '{workspace}' workspace is not in Direct Lake mode."
         )
-        return
 
     dfE = fabric.list_expressions(dataset=dataset, workspace=workspace)
     dfE_filt = dfE[dfE["Name"] == "DatabaseQuery"]
@@ -416,16 +420,10 @@ def save_as_delta_table(
     write_mode = write_mode.lower()
 
     if write_mode not in writeModes:
-        print(
-            f"{icons.red_dot} Invalid 'write_type' parameter. Choose from one of the following values: {writeModes}."
-        )
-        return
+        raise ValueError(f"{icons.red_dot} Invalid 'write_type' parameter. Choose from one of the following values: {writeModes}.")
 
     if " " in delta_table_name:
-        print(
-            f"{icons.red_dot} Invalid 'delta_table_name'. Delta tables in the lakehouse cannot have spaces in their names."
-        )
-        return
+        raise ValueError(f"{icons.red_dot} Invalid 'delta_table_name'. Delta tables in the lakehouse cannot have spaces in their names.")
 
     dataframe.columns = dataframe.columns.str.replace(" ", "_")
 
@@ -472,10 +470,7 @@ def language_validate(language: str):
     elif len(df_filt2) == 1:
         lang = df_filt2["Language"].iloc[0]
     else:
-        print(
-            f"The '{language}' language is not a valid language code. Please refer to this link for a list of valid language codes: {url}."
-        )
-        return
+        raise ValueError(f"{icons.red_dot} The '{language}' language is not a valid language code. Please refer to this link for a list of valid language codes: {url}.")
 
     return lang
 
@@ -497,10 +492,10 @@ def resolve_workspace_name_and_id(workspace: Optional[str] = None) -> Tuple[str,
         The name and ID of the Fabric workspace.
     """
 
-    if workspace == None:
+    if workspace is None:
         workspace_id = fabric.get_workspace_id()
         workspace = fabric.resolve_workspace_name(workspace_id)
     else:
         workspace_id = fabric.resolve_workspace_id(workspace)
 
-    return workspace, workspace_id
+    return str(workspace), str(workspace_id)

sempy_labs/_icons.py

@@ -2,3 +2,9 @@ green_dot = "\U0001F7E2"
 yellow_dot = "\U0001F7E1"
 red_dot = "\U0001F534"
 in_progress = "⌛"
+checked = "\u2611"
+unchecked = "\u2610"
+start_bold = "\033[1m"
+end_bold = "\033[0m"
+bullet = "\u2022"
+warning = "⚠️"