semantic-link-labs 0.4.2__py3-none-any.whl → 0.6.0__py3-none-any.whl

{semantic_link_labs-0.4.2.dist-info → semantic_link_labs-0.6.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: semantic-link-labs
-Version: 0.4.2
+Version: 0.6.0
 Summary: Semantic Link Labs project
 Author: Microsoft Corporation
 License: MIT License
@@ -14,7 +14,7 @@ Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Framework :: Jupyter
 Requires-Python: <3.12,>=3.10
 License-File: LICENSE
-Requires-Dist: semantic-link-sempy >=0.7.5
+Requires-Dist: semantic-link-sempy >=0.7.6
 Requires-Dist: anytree
 Requires-Dist: powerbiclient
 Provides-Extra: test

semantic_link_labs-0.6.0.dist-info/RECORD

@@ -0,0 +1,54 @@
+sempy_labs/__init__.py,sha256=T_5thSoxVg83rUR_aac7P4kBKPrf5wylwRAxNMzYkjw,4976
+sempy_labs/_ai.py,sha256=1sWyWxM_fAqdT5Ih2F3QjqWepJ0R48RvffNHSZKfuXs,18079
+sempy_labs/_clear_cache.py,sha256=AfTCAwSK5EYhI2aBPsSun8FxJy-GHlByxH-rBxHfHzM,1362
+sempy_labs/_connections.py,sha256=w1dFC4WeTNFmLGD2EL_Syk0Wb1Eij18we2FVn_VaCD8,7641
+sempy_labs/_dax.py,sha256=dt1GgHceyM7f6phRBPxRKnmQy_KYKpcgFQHuOjGbpLo,2029
+sempy_labs/_generate_semantic_model.py,sha256=qdWL6GDuCstzGLzChNRZ_nr9CndIl-sKmYG1WrZHXcw,9689
+sempy_labs/_helper_functions.py,sha256=bcDDwy7LqOVV3x4scCSBxzMQuU2sLWBzgMSaluDrxEk,14306
+sempy_labs/_icons.py,sha256=RfKaIzFHvu9mtQo6N10w9BQUa9JPHQ6NtJUyASZBdTo,536
+sempy_labs/_list_functions.py,sha256=uNZsQD32jQqQYgOegLYO1m28pRqCpCqZMIm0fyA2QW0,82709
+sempy_labs/_model_auto_build.py,sha256=fX3bCLFCOMQHuheKIoB48fUABG7XAT7qqsMbUiWSrY0,5071
+sempy_labs/_model_bpa.py,sha256=PTC_vKKg-r66o0cXIGsx-0cW2ms-S1mWOLPZW8vGa2Q,14080
+sempy_labs/_model_bpa_rules.py,sha256=aDNudcvBLaKaDLy82MF2D8gbrAtVvrOt9u5AqDkltnk,47291
+sempy_labs/_model_dependencies.py,sha256=0uOhTMLpfhZ0mjIjhKacrvzOYs1uHoFfzc2hqTGAY50,12965
+sempy_labs/_one_lake_integration.py,sha256=r9gnLoFjzfJroAzlJ5aWpKs-rxemzJxnN7tADML0IQc,6246
+sempy_labs/_query_scale_out.py,sha256=-RMDO7T8QsIaDK3xXLyg1eOKXK5sKOBTrIISaKoSFRI,14901
+sempy_labs/_refresh_semantic_model.py,sha256=TGGdqMzpR0POXgaXkqxfD2ene-SwEvU3aujsVlkWX7E,6734
+sempy_labs/_translations.py,sha256=bWlLfz9ynFIQEBRHAWAyV1XZdYTUzLCTtipZKk6Fd_U,12920
+sempy_labs/_vertipaq.py,sha256=pIRfJKjsShxVDdMBoeO97vDvurZ98iblcSdtowSJgxQ,33303
+sempy_labs/directlake/__init__.py,sha256=yDoHDd4TyP7wTTjgMASTsjyXgRrTJwJjgml-IjVl460,1781
+sempy_labs/directlake/_directlake_schema_compare.py,sha256=xljJ60aymEJeQV9w-0NQTzFGA8AUiYvsByX7QqHxglg,4716
+sempy_labs/directlake/_directlake_schema_sync.py,sha256=VzTT9o6xceVfx_8wJkvXwy1jGwjH5HztIrQ_Vwh4B4Q,4940
+sempy_labs/directlake/_fallback.py,sha256=o6FswFiisMVbDqRTCJ2hHa1M65KafkGjaqQl5PQ6szQ,1992
+sempy_labs/directlake/_get_directlake_lakehouse.py,sha256=IkkpJe7iFbKT_9KCMdaLucuw_WCcdmEQd3bz6-Wmgx0,2346
+sempy_labs/directlake/_get_shared_expression.py,sha256=3KIrsaCX-t2r4_anC8Z4MMlbu3Iaa_gCHCC8Ok7NDxA,1936
+sempy_labs/directlake/_guardrails.py,sha256=pnLYM7ZO6_hq38UMNeJRx5Fp9XbcIpgPuAS3py24oA8,2259
+sempy_labs/directlake/_list_directlake_model_calc_tables.py,sha256=_rpnbgsFAz2W16PpgIOB0Rj_Fs1ZKrDbz3DUaaR_bfU,2143
+sempy_labs/directlake/_show_unsupported_directlake_objects.py,sha256=QNj2wHzFGtjnsAICmlc7BuhCYkw0An0XnditDTCG2JM,3358
+sempy_labs/directlake/_update_directlake_model_lakehouse_connection.py,sha256=b_Y5_GSfWC25wH6R7L37-AHO9fvKkmxRGaP6dVDC7-w,3233
+sempy_labs/directlake/_update_directlake_partition_entity.py,sha256=nMh2CqQrp64NipmryuLKewljmFt6aVKaecIwSedIFIw,7687
+sempy_labs/directlake/_warm_cache.py,sha256=RjpRjtzSaKw6gHKPw4O6phBwmnk7LHp4B1yrW8Z6VEY,8242
+sempy_labs/lakehouse/__init__.py,sha256=i6VRx4dR1SIN-1GxioiNwhC4FxbozRCIz5TfXjb9rKc,587
+sempy_labs/lakehouse/_get_lakehouse_columns.py,sha256=Bb_iCTlNwl0wdN4dW_E7tVnfbHhHwQT_l0SUqvcbYpo,2582
+sempy_labs/lakehouse/_get_lakehouse_tables.py,sha256=c2pqEyhiEJhnRaE3Zlz3EHRmR9bFI6J7CZxqLVa87iE,9091
+sempy_labs/lakehouse/_lakehouse.py,sha256=kUMiGi8SyY4oTWAB4OFzGQzlqUdAF2EolpZza15gFrE,2763
+sempy_labs/lakehouse/_shortcuts.py,sha256=oDYexjMxZX2JoaEqSyO1R_PQ2c5qTUNVbHGtVQJZorg,6977
+sempy_labs/migration/__init__.py,sha256=w4vvGk6wTWXVfofJDmio2yIFvSSJsxOpjv6mvNGmrOI,1043
+sempy_labs/migration/_create_pqt_file.py,sha256=XTG1BQJbfgKpwT8jKWrlCfDuMYdecM94qTLzrpQ7wck,9035
+sempy_labs/migration/_migrate_calctables_to_lakehouse.py,sha256=6Z1DIdZCi_LX0o-SfPO5zzqWa0hd_zEwcQzIsOer9SM,20551
+sempy_labs/migration/_migrate_calctables_to_semantic_model.py,sha256=Ne-2_PwW_LUR8vG2pNY3lxa9iib1dIpYkDHRaqyqyY8,6420
+sempy_labs/migration/_migrate_model_objects_to_semantic_model.py,sha256=T7Ijo9g9AmB8KMyqvYOTurSdabUKKaRcfdstSgL1QQ8,24399
+sempy_labs/migration/_migrate_tables_columns_to_semantic_model.py,sha256=1ZWQ9sawd2_ojTnWTsRaDEj7yMnUVOpCQZeI0YHhb-g,7066
+sempy_labs/migration/_migration_validation.py,sha256=phbUUIYA5z4dZvEKrVbByMTPPwMscY7Dy9HVxE8z_HM,2483
+sempy_labs/migration/_refresh_calc_tables.py,sha256=pImD8wMOMxs043e6s4G2TPrFzsGVHklK1aez_oVOgro,6086
+sempy_labs/report/__init__.py,sha256=78lLE4UlmwfSY-6FxMIgnshMQF90Qfq_qM7A3vrNtHw,848
+sempy_labs/report/_generate_report.py,sha256=fJUiUgsF2mS-dyaVjRvwAASvbSrSm8BUDB53Ru3rhQI,8545
+sempy_labs/report/_report_functions.py,sha256=6xpLyVrRkA2ZwwCj2By8_HX2k-EHoQUZtuUaTy0d7vM,30004
+sempy_labs/report/_report_rebind.py,sha256=PmZ3f3Rf3Yka8Eaea87JosSiFvEtzjfbUDQOzRx_vlQ,4614
+sempy_labs/tom/__init__.py,sha256=Qbs8leW0fjzvWwOjyWK3Hjeehu7IvpB1beASGsi28bk,121
+sempy_labs/tom/_model.py,sha256=kGRfpA6IHy8KX6QKVciekOa2Y97d6r0eRol5K8YMMIg,149938
+semantic_link_labs-0.6.0.dist-info/LICENSE,sha256=ws_MuBL-SCEBqPBFl9_FqZkaaydIJmxHrJG2parhU4M,1141
+semantic_link_labs-0.6.0.dist-info/METADATA,sha256=ZFdW982vsZ0btahdjKpAgiDMEXxnjn_nwiPY8iMMrdI,764
+semantic_link_labs-0.6.0.dist-info/WHEEL,sha256=Wyh-_nZ0DJYolHNn1_hMa4lM7uDedD_RGVwbmTjyItk,91
+semantic_link_labs-0.6.0.dist-info/top_level.txt,sha256=kiQX1y42Dbein1l3Q8jMUYyRulDjdlc2tMepvtrvixQ,11
+semantic_link_labs-0.6.0.dist-info/RECORD,,

{semantic_link_labs-0.4.2.dist-info → semantic_link_labs-0.6.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.43.0)
+Generator: setuptools (71.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

sempy_labs/__init__.py

@@ -9,13 +9,15 @@ from sempy_labs._dax import evaluate_dax_impersonation
 from sempy_labs._generate_semantic_model import (
     create_blank_semantic_model,
     create_semantic_model_from_bim,
-    # deploy_semantic_model,
+    deploy_semantic_model,
     get_semantic_model_bim,
 )
 from sempy_labs._list_functions import (
+    delete_custom_pool,
     list_semantic_model_objects,
     list_shortcuts,
     get_object_level_security,
+    list_capacities,
     # list_annotations,
     # list_columns,
     list_dashboards,
@@ -34,9 +36,21 @@ from sempy_labs._list_functions import (
     # list_sqlendpoints,
     # list_tables,
     list_warehouses,
-    # list_workspace_role_assignments,
+    list_workspace_role_assignments,
     create_warehouse,
     update_item,
+    list_custom_pools,
+    create_custom_pool,
+    update_custom_pool,
+    assign_workspace_to_capacity,
+    unassign_workspace_from_capacity,
+    get_spark_settings,
+    update_spark_settings,
+    add_user_to_workspace,
+    delete_user_from_workspace,
+    update_workspace_user,
+    list_workspace_users,
+    assign_workspace_to_dataflow_storage,
 )
 
 from sempy_labs._helper_functions import (
@@ -54,8 +68,10 @@ from sempy_labs._helper_functions import (
     resolve_report_name,
     #  language_validate
 )
+
 #  from sempy_labs._model_auto_build import model_auto_build
-from sempy_labs._model_bpa import model_bpa_rules, run_model_bpa
+from sempy_labs._model_bpa import run_model_bpa
+from sempy_labs._model_bpa_rules import model_bpa_rules
 from sempy_labs._model_dependencies import (
     measure_dependency_tree,
     get_measure_dependencies,
@@ -85,6 +101,7 @@ from sempy_labs._vertipaq import (
 )
 
 __all__ = [
+    "delete_custom_pool",
     "clear_cache",
     # create_connection_cloud,
     # create_connection_vnet,
@@ -92,7 +109,7 @@ __all__ = [
     "evaluate_dax_impersonation",
     "create_blank_semantic_model",
     "create_semantic_model_from_bim",
-    #'deploy_semantic_model',
+    "deploy_semantic_model",
     "get_semantic_model_bim",
     "get_object_level_security",
     #'list_annotations',
@@ -113,7 +130,7 @@ __all__ = [
     #'list_sqlendpoints',
     #'list_tables',
     "list_warehouses",
-    #'list_workspace_role_assignments',
+    "list_workspace_role_assignments",
     "create_warehouse",
     "update_item",
     "create_abfss_path",
@@ -129,20 +146,20 @@ __all__ = [
     "resolve_report_id",
     "resolve_report_name",
     #'language_validate',
-    #"model_auto_build",
+    # "model_auto_build",
     "model_bpa_rules",
     "run_model_bpa",
     "measure_dependency_tree",
     "get_measure_dependencies",
     "get_model_calc_dependencies",
     "export_model_to_onelake",
-    'qso_sync',
-    'qso_sync_status',
-    'set_qso',
-    'list_qso_settings',
-    'disable_qso',
-    'set_semantic_model_storage_format',
-    'set_workspace_default_storage_format',
+    "qso_sync",
+    "qso_sync_status",
+    "set_qso",
+    "list_qso_settings",
+    "disable_qso",
+    "set_semantic_model_storage_format",
+    "set_workspace_default_storage_format",
     "refresh_semantic_model",
     "cancel_dataset_refresh",
     "translate_semantic_model",
@@ -150,5 +167,18 @@ __all__ = [
     #'visualize_vertipaq',
     "import_vertipaq_analyzer",
     "list_semantic_model_objects",
-    "list_shortcuts"
+    "list_shortcuts",
+    "list_custom_pools",
+    "create_custom_pool",
+    "update_custom_pool",
+    "assign_workspace_to_capacity",
+    "unassign_workspace_from_capacity",
+    "get_spark_settings",
+    "update_spark_settings",
+    "add_user_to_workspace",
+    "delete_user_from_workspace",
+    "update_workspace_user",
+    "list_workspace_users",
+    "assign_workspace_to_dataflow_storage",
+    "list_capacities",
 ]

sempy_labs/_ai.py

@@ -40,7 +40,8 @@ def optimize_semantic_model(dataset: str, workspace: Optional[str] = None):
 
         if len(fallback_filt) > 0:
             print(
-                f"{icons.yellow_dot} The '{dataset}' semantic model is a Direct Lake semantic model which contains views. Since views always fall back to DirectQuery, it is recommended to only use lakehouse tables and not views."
+                f"{icons.yellow_dot} The '{dataset}' semantic model is a Direct Lake semantic model which contains views. "
+                "Since views always fall back to DirectQuery, it is recommended to only use lakehouse tables and not views."
             )
 
     # Potential model reduction estimate
@@ -78,10 +79,9 @@ def generate_measure_descriptions(
 
     validModels = ["gpt-35-turbo", "gpt-35-turbo-16k", "gpt-4"]
     if gpt_model not in validModels:
-        print(
+        raise ValueError(
             f"{icons.red_dot} The '{gpt_model}' model is not a valid model. Enter a gpt_model from this list: {validModels}."
         )
-        return
 
     dfM = fabric.list_measures(dataset=dataset, workspace=workspace)
 
@@ -116,8 +116,7 @@ def generate_measure_descriptions(
     )
 
     # Update the model to use the new descriptions
-    tom_server = fabric.create_tom_server(readonly=False, workspace=workspace)
-    m = tom_server.Databases.GetByName(dataset).Model
+    # with connect_semantic_model(dataset=dataset, workspace=workspace, readonly=False) as tom:
 
     # for t in m.Tables:
     # tName = t.Name
@@ -148,10 +147,10 @@ def generate_aggs(
     import System
 
     # columns = {
-    #'SalesAmount': 'Sum',
-    #'ProductKey': 'GroupBy',
-    #'OrderDateKey': 'GroupBy'
-    # }
+    # 'SalesAmount': 'Sum',
+    # 'ProductKey': 'GroupBy',
+    # 'OrderDateKey': 'GroupBy'
+    #  }
 
     if workspace is None:
         workspace_id = fabric.get_workspace_id()
@@ -173,48 +172,44 @@ def generate_aggs(
     numericTypes = ["Int64", "Double", "Decimal"]
 
     if any(value not in aggTypes for value in columns.values()):
-        print(
+        raise ValueError(
             f"{icons.red_dot} Invalid aggregation type(s) have been specified in the 'columns' parameter. Valid aggregation types: {aggTypes}."
         )
-        return
 
     dfC = fabric.list_columns(dataset=dataset, workspace=workspace)
     dfP = fabric.list_partitions(dataset=dataset, workspace=workspace)
     dfM = fabric.list_measures(dataset=dataset, workspace=workspace)
     dfR = fabric.list_relationships(dataset=dataset, workspace=workspace)
     if not any(r["Mode"] == "DirectLake" for i, r in dfP.iterrows()):
-        print(
+        raise ValueError(
             f"{icons.red_dot} The '{dataset}' semantic model within the '{workspace}' workspace is not in Direct Lake mode. This function is only relevant for Direct Lake semantic models."
         )
-        return
 
     dfC_filtT = dfC[dfC["Table Name"] == table_name]
 
     if len(dfC_filtT) == 0:
-        print(
+        raise ValueError(
             f"{icons.red_dot} The '{table_name}' table does not exist in the '{dataset}' semantic model within the '{workspace}' workspace."
         )
-        return
 
     dfC_filt = dfC[
         (dfC["Table Name"] == table_name) & (dfC["Column Name"].isin(columnValues))
     ]
 
     if len(columns) != len(dfC_filt):
-        print(
+        raise ValueError(
             f"{icons.red_dot} Columns listed in '{columnValues}' do not exist in the '{table_name}' table in the '{dataset}' semantic model within the '{workspace}' workspace."
         )
-        return
 
     # Check if doing sum/count/min/max etc. on a non-number column
-    for col, agg in columns.items():
-        dfC_col = dfC_filt[dfC_filt["Column Name"] == col]
+    for cm, agg in columns.items():
+        dfC_col = dfC_filt[dfC_filt["Column Name"] == cm]
         dataType = dfC_col["Data Type"].iloc[0]
         if agg in aggTypesAggregate and dataType not in numericTypes:
-            print(
-                f"{icons.red_dot} The '{col}' column in the '{table_name}' table is of '{dataType}' data type. Only columns of '{numericTypes}' data types can be aggregated as '{aggTypesAggregate}' aggregation types."
+            raise ValueError(
+                f"{icons.red_dot} The '{cm}' column in the '{table_name}' table is of '{dataType}' data type. Only columns of '{numericTypes}' data types"
+                f" can be aggregated as '{aggTypesAggregate}' aggregation types."
             )
-            return
 
     # Create/update lakehouse delta agg table
     aggSuffix = "_agg"
@@ -230,10 +225,10 @@ def generate_aggs(
     dfI_filt = dfI[(dfI["Id"] == sqlEndpointId)]
 
     if len(dfI_filt) == 0:
-        print(
-            f"{icons.red_dot} The lakehouse (SQL Endpoint) used by the '{dataset}' semantic model does not reside in the '{lakehouse_workspace}' workspace. Please update the lakehouse_workspace parameter."
+        raise ValueError(
+            f"{icons.red_dot} The lakehouse (SQL Endpoint) used by the '{dataset}' semantic model does not reside in"
+            f" the '{lakehouse_workspace}' workspace. Please update the lakehouse_workspace parameter."
         )
-        return
 
     lakehouseName = dfI_filt["Display Name"].iloc[0]
     lakehouse_id = resolve_lakehouse_id(
@@ -243,8 +238,8 @@ def generate_aggs(
     # Generate SQL query
     query = "SELECT"
     groupBy = "\nGROUP BY"
-    for col, agg in columns.items():
-        colFilt = dfC_filt[dfC_filt["Column Name"] == col]
+    for cm, agg in columns.items():
+        colFilt = dfC_filt[dfC_filt["Column Name"] == cm]
         sourceCol = colFilt["Source"].iloc[0]
 
         if agg == "GroupBy":
@@ -284,7 +279,7 @@ def generate_aggs(
     # Create/update semantic model agg table
     tom_server = fabric.create_tom_server(readonly=False, workspace=workspace)
     m = tom_server.Databases.GetByName(dataset).Model
-    f"\n{icons.in_progress} Updating the '{dataset}' semantic model..."
+    print(f"\n{icons.in_progress} Updating the '{dataset}' semantic model...")
     dfC_agg = dfC[dfC["Table Name"] == aggTableName]
 
     if len(dfC_agg) == 0:
@@ -348,7 +343,9 @@ def generate_aggs(
             col.DataType = System.Enum.Parse(TOM.DataType, dType)
 
             m.Tables[aggTableName].Columns.Add(col)
-            print(f"{icons.green_dot} The '{aggTableName}'[{cName}] column has been added.")
+            print(
+                f"{icons.green_dot} The '{aggTableName}'[{cName}] column has been added."
+            )
 
     # Create relationships
     relMap = {"m": "Many", "1": "One", "0": "None"}
@@ -387,10 +384,11 @@ def generate_aggs(
                 print(
                     f"{icons.green_dot} '{aggTableName}'[{fromColumn}] -> '{toTable}'[{toColumn}] relationship has been added."
                 )
-            except:
+            except Exception as e:
                 print(
                     f"{icons.red_dot} '{aggTableName}'[{fromColumn}] -> '{toTable}'[{toColumn}] relationship has not been created."
                 )
+                print(f"Exception occured: {e}")
         elif toTable == table_name:
             try:
                 rel.ToColumn = m.Tables[aggTableName].Columns[toColumn]
@@ -398,11 +396,12 @@ def generate_aggs(
                 print(
                     f"{icons.green_dot} '{fromTable}'[{fromColumn}] -> '{aggTableName}'[{toColumn}] relationship has been added."
                 )
-            except:
+            except Exception as e:
                 print(
                     f"{icons.red_dot} '{fromTable}'[{fromColumn}] -> '{aggTableName}'[{toColumn}] relationship has not been created."
                 )
-    f"Relationship creation is complete."
+                print(f"Exception occured: {e}")
+    "Relationship creation is complete."
 
     # Create IF measure
     f"\n{icons.in_progress} Creating measure to check if the agg table can be used..."

sempy_labs/_clear_cache.py

@@ -1,7 +1,6 @@
-import sempy
 import sempy.fabric as fabric
 from ._helper_functions import resolve_dataset_id
-from typing import List, Optional, Union
+from typing import Optional
 import sempy_labs._icons as icons
 
 
@@ -20,17 +19,15 @@ def clear_cache(dataset: str, workspace: Optional[str] = None):
         or if no lakehouse attached, resolves to the workspace of the notebook.
     """
 
-    if workspace is None:
-        workspace_id = fabric.get_workspace_id()
-        workspace = fabric.resolve_workspace_name(workspace_id)
+    workspace = fabric.resolve_workspace_name(workspace)
 
     datasetID = resolve_dataset_id(dataset=dataset, workspace=workspace)
 
     xmla = f"""
-            <ClearCache xmlns="http://schemas.microsoft.com/analysisservices/2003/engine">  
+            <ClearCache xmlns="http://schemas.microsoft.com/analysisservices/2003/engine">
                 <Object>  
-                    <DatabaseID>{datasetID}</DatabaseID>  
-                </Object>  
+                    <DatabaseID>{datasetID}</DatabaseID>
+                </Object>
             </ClearCache>
             """
     fabric.execute_xmla(dataset=dataset, xmla_command=xmla, workspace=workspace)

sempy_labs/_connections.py

@@ -1,7 +1,6 @@
-import sempy
 import sempy.fabric as fabric
 import pandas as pd
-from typing import List, Optional, Union
+from sempy.fabric.exceptions import FabricHTTPException
 
 
 def create_connection_cloud(
@@ -11,7 +10,7 @@ def create_connection_cloud(
     user_name: str,
     password: str,
     privacy_level: str,
-):
+) -> pd.DataFrame:
 
     # https://review.learn.microsoft.com/en-us/rest/api/fabric/core/connections/create-connection?branch=features%2Fdmts&tabs=HTTP
 
@@ -55,29 +54,32 @@ def create_connection_cloud(
         },
     }
 
-    response = client.post(f"/v1/connections", json=request_body)
-
-    if response.status_code == 200:
-        o = response.json()
-        new_data = {
-            "Connection Id": o["id"],
-            "Connection Name": o["name"],
-            "Connectivity Type": o["connectivityType"],
-            "Connection Type": o["connectionDetails"]["type"],
-            "Connection Path": o["connectionDetails"]["path"],
-            "Privacy Level": o["privacyLevel"],
-            "Credential Type": o["credentialDetails"]["credentialType"],
-            "Single Sign On Type": o["credentialDetails"]["singleSignOnType"],
-            "Connection Encryption": o["credentialDetails"]["connectionEncryption"],
-            "Skip Test Connection": o["credentialDetails"]["skipTestConnection"],
-        }
-        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+    response = client.post("/v1/connections", json=request_body)
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+    o = response.json()
+    new_data = {
+        "Connection Id": o.get("id"),
+        "Connection Name": o.get("name"),
+        "Connectivity Type": o.get("connectivityType"),
+        "Connection Type": o.get("connectionDetails", {}).get("type"),
+        "Connection Path": o.get("connectionDetails", {}).get("path"),
+        "Privacy Level": o.get("privacyLevel"),
+        "Credential Type": o.get("credentialDetails", {}).get("credentialType"),
+        "Single Sign On Type": o.get("credentialDetails", {}).get("singleSignOnType"),
+        "Connection Encryption": o.get("credentialDetails", {}).get(
+            "connectionEncryption"
+        ),
+        "Skip Test Connection": o.get("credentialDetails", {}).get(
+            "skipTestConnection"
+        ),
+    }
+    df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
-        df["Skip Test Connection"] = df["Skip Test Connection"].astype(bool)
+    df["Skip Test Connection"] = df["Skip Test Connection"].astype(bool)
 
-        return df
-    else:
-        print(response.status_code)
+    return df
 
 
 def create_connection_on_prem(
@@ -87,7 +89,7 @@ def create_connection_on_prem(
     database_name: str,
     credentials: str,
     privacy_level: str,
-):
+) -> pd.DataFrame:
 
     df = pd.DataFrame(
         columns=[
@@ -130,30 +132,33 @@ def create_connection_on_prem(
         },
     }
 
-    response = client.post(f"/v1/connections", json=request_body)
-
-    if response.status_code == 200:
-        o = response.json()
-        new_data = {
-            "Connection Id": o["id"],
-            "Connection Name": o["name"],
-            "Gateway ID": o["gatewayId"],
-            "Connectivity Type": o["connectivityType"],
-            "Connection Type": o["connectionDetails"]["type"],
-            "Connection Path": o["connectionDetails"]["path"],
-            "Privacy Level": o["privacyLevel"],
-            "Credential Type": o["credentialDetails"]["credentialType"],
-            "Single Sign On Type": o["credentialDetails"]["singleSignOnType"],
-            "Connection Encryption": o["credentialDetails"]["connectionEncryption"],
-            "Skip Test Connection": o["credentialDetails"]["skipTestConnection"],
-        }
-        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+    response = client.post("/v1/connections", json=request_body)
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+    o = response.json()
+    new_data = {
+        "Connection Id": o.get("id"),
+        "Connection Name": o.get("name"),
+        "Gateway ID": o.get("gatewayId"),
+        "Connectivity Type": o.get("connectivityType"),
+        "Connection Type": o.get("connectionDetails", {}).get("type"),
+        "Connection Path": o.get("connectionDetails", {}).get("path"),
+        "Privacy Level": o.get("privacyLevel"),
+        "Credential Type": o.get("credentialDetails", {}).get("credentialType"),
+        "Single Sign On Type": o.get("credentialDetails", {}).get("singleSignOnType"),
+        "Connection Encryption": o.get("credentialDetails", {}).get(
+            "connectionEncryption"
+        ),
+        "Skip Test Connection": o.get("credentialDetails", {}).get(
+            "skipTestConnection"
+        ),
+    }
+    df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
-        df["Skip Test Connection"] = df["Skip Test Connection"].astype(bool)
+    df["Skip Test Connection"] = df["Skip Test Connection"].astype(bool)
 
-        return df
-    else:
-        print(response.status_code)
+    return df
 
 
 def create_connection_vnet(
@@ -164,7 +169,7 @@ def create_connection_vnet(
     user_name: str,
     password: str,
     privacy_level: str,
-):
+) -> pd.DataFrame:
 
     df = pd.DataFrame(
         columns=[
@@ -208,27 +213,30 @@ def create_connection_vnet(
         },
     }
 
-    response = client.post(f"/v1/connections", json=request_body)
-
-    if response.status_code == 200:
-        o = response.json()
-        new_data = {
-            "Connection Id": o["id"],
-            "Connection Name": o["name"],
-            "Gateway ID": o["gatewayId"],
-            "Connectivity Type": o["connectivityType"],
-            "Connection Type": o["connectionDetails"]["type"],
-            "Connection Path": o["connectionDetails"]["path"],
-            "Privacy Level": o["privacyLevel"],
-            "Credential Type": o["credentialDetails"]["credentialType"],
-            "Single Sign On Type": o["credentialDetails"]["singleSignOnType"],
-            "Connection Encryption": o["credentialDetails"]["connectionEncryption"],
-            "Skip Test Connection": o["credentialDetails"]["skipTestConnection"],
-        }
-        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
-
-        df["Skip Test Connection"] = df["Skip Test Connection"].astype(bool)
-
-        return df
-    else:
-        print(response.status_code)
+    response = client.post("/v1/connections", json=request_body)
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+    o = response.json()
+    new_data = {
+        "Connection Id": o.get("id"),
+        "Connection Name": o.get("name"),
+        "Gateway ID": o.get("gatewayId"),
+        "Connectivity Type": o.get("connectivityType"),
+        "Connection Type": o.get("connectionDetails", {}).get("type"),
+        "Connection Path": o.get("connectionDetails", {}).get("path"),
+        "Privacy Level": o.get("privacyLevel"),
+        "Credential Type": o.get("credentialDetails", {}).get("credentialType"),
+        "Single Sign On Type": o.get("credentialDetails", {}).get("singleSignOnType"),
+        "Connection Encryption": o.get("credentialDetails", {}).get(
+            "connectionEncryption"
+        ),
+        "Skip Test Connection": o.get("credentialDetails", {}).get(
+            "skipTestConnection"
+        ),
+    }
+    df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+
+    df["Skip Test Connection"] = df["Skip Test Connection"].astype(bool)
+
+    return df

sempy_labs/_dax.py

@@ -1,8 +1,10 @@
-import sempy
 import sempy.fabric as fabric
 import pandas as pd
-from sempy_labs._helper_functions import resolve_dataset_id
-from typing import List, Optional, Union
+from sempy_labs._helper_functions import (
+    resolve_dataset_id,
+    resolve_workspace_name_and_id,
+)
+from typing import Optional
 from sempy._utils._log import log
 
 
@@ -40,17 +42,13 @@ def evaluate_dax_impersonation(
 
     # https://learn.microsoft.com/en-us/rest/api/power-bi/datasets/execute-queries-in-group
 
-    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)
+    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     dataset_id = resolve_dataset_id(dataset=dataset, workspace=workspace)
 
     request_body = {
         "queries": [{"query": dax_query}],
-        "impersonatedUserName": user_name
+        "impersonatedUserName": user_name,
     }
 
     client = fabric.PowerBIRestClient()