semantic-link-labs 0.8.3__py3-none-any.whl → 0.8.4__py3-none-any.whl

sempy_labs/admin/_domains.py

@@ -34,6 +34,8 @@ def list_domains(non_empty_only: bool = False) -> pd.DataFrame:
     """
     Shows a list of domains.
 
+    This is a wrapper function for the following API: `Domains - List Domains <https://learn.microsoft.com/rest/api/fabric/admin/domains/list-domains>`_.
+
     Parameters
     ----------
     non_empty_only : bool, default=False
@@ -46,8 +48,6 @@ def list_domains(non_empty_only: bool = False) -> pd.DataFrame:
         A pandas dataframe showing a list of the domains.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/domains/list-domains?tabs=HTTP
-
     df = pd.DataFrame(
         columns=[
             "Domain ID",
@@ -84,6 +84,8 @@ def list_domain_workspaces(domain_name: str) -> pd.DataFrame:
     """
     Shows a list of workspaces within the domain.
 
+    This is a wrapper function for the following API: `Domains - List Domain Workspaces <https://learn.microsoft.com/rest/api/fabric/admin/domains/list-domain-workspaces?tabs=HTTP>`_.
+
     Parameters
     ----------
     domain_name : str
@@ -95,8 +97,6 @@ def list_domain_workspaces(domain_name: str) -> pd.DataFrame:
         A pandas dataframe showing a list of workspaces within the domain.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/domains/list-domain-workspaces?tabs=HTTP
-
     domain_id = resolve_domain_id(domain_name)
 
     df = pd.DataFrame(columns=["Workspace ID", "Workspace Name"])
@@ -125,6 +125,8 @@ def create_domain(
     """
     Creates a new domain.
 
+    This is a wrapper function for the following API: `Domains - Create Domain <https://learn.microsoft.com/rest/api/fabric/admin/domains/create-domain>`_.
+
     Parameters
     ----------
     domain_name : str
@@ -135,8 +137,6 @@ def create_domain(
         The parent domain name.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/domains/create-domain?tabs=HTTP
-
     if parent_domain_name is not None:
         parent_domain_id = resolve_domain_id(parent_domain_name)
 
@@ -160,14 +160,14 @@ def delete_domain(domain_name: str):
     """
     Deletes a domain.
 
+    This is a wrapper function for the following API: `Domains - Delete Domain <https://learn.microsoft.com/rest/api/fabric/admin/domains/delete-domain>`_.
+
     Parameters
     ----------
     domain_name : str
         The domain name.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/domains/delete-domain?tabs=HTTP
-
     domain_id = resolve_domain_id(domain_name)
 
     client = fabric.FabricRestClient()
@@ -187,6 +187,8 @@ def update_domain(
     """
     Updates a domain's properties.
 
+    This is a wrapper function for the following API: `Domains - Update Domain <https://learn.microsoft.com/rest/api/fabric/admin/domains/update-domain>`_.
+
     Parameters
     ----------
     domain_name : str
@@ -197,8 +199,6 @@ def update_domain(
         The domain `contributor scope <https://learn.microsoft.com/rest/api/fabric/admin/domains/update-domain?tabs=HTTP#contributorsscopetype>`_.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/domains/update-domain?tabs=HTTP
-
     contributors_scopes = ["AdminsOnly", "AllTenant", "SpecificUsersAndGroups"]
 
     if contributors_scope not in contributors_scopes:
@@ -231,6 +231,8 @@ def assign_domain_workspaces_by_capacities(
     """
     Assigns all workspaces that reside on the specified capacities to the specified domain.
 
+    This is a wrapper function for the following API: `Domains - Assign Domain Workspaces By Capacities <https://learn.microsoft.com/rest/api/fabric/admin/domains/assign-domain-workspaces-by-capacities>`_.
+
     Parameters
     ----------
     domain_name : str
@@ -239,8 +241,6 @@ def assign_domain_workspaces_by_capacities(
         The capacity names.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/domains/assign-domain-workspaces-by-capacities?tabs=HTTP
-
     from sempy_labs.admin import list_capacities
 
     domain_id = resolve_domain_id(domain_name)
@@ -287,6 +287,8 @@ def assign_domain_workspaces(domain_name: str, workspace_names: str | List[str])
     """
     Assigns workspaces to the specified domain by workspace.
 
+    This is a wrapper function for the following API: `Domains - Assign Domain Workspaces By Ids <https://learn.microsoft.com/rest/api/fabric/admin/domains/assign-domain-workspaces-by-ids>`_.
+
     Parameters
     ----------
     domain_name : str
@@ -295,8 +297,6 @@ def assign_domain_workspaces(domain_name: str, workspace_names: str | List[str])
         The Fabric workspace(s).
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/domains/assign-domain-workspaces-by-ids?tabs=HTTP
-
     domain_id = resolve_domain_id(domain_name=domain_name)
 
     if isinstance(workspace_names, str):
@@ -340,14 +340,14 @@ def unassign_all_domain_workspaces(domain_name: str):
     """
     Unassigns all workspaces from the specified domain.
 
+    This is a wrapper function for the following API: `Domains - Unassign All Domain Workspaces <https://learn.microsoft.com/rest/api/fabric/admin/domains/unassign-all-domain-workspaces>`_.
+
     Parameters
     ----------
     domain_name : str
         The domain name.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/domains/unassign-all-domain-workspaces?tabs=HTTP
-
     domain_id = resolve_domain_id(domain_name=domain_name)
 
     client = fabric.FabricRestClient()
@@ -364,6 +364,8 @@ def unassign_domain_workspaces(domain_name: str, workspace_names: str | List[str
     """
     Unassigns workspaces from the specified domain by workspace.
 
+    This is a wrapper function for the following API: `Domains - Unassign Domain Workspaces By Ids <https://learn.microsoft.com/rest/api/fabric/admin/domains/unassign-domain-workspaces-by-ids>`_.
+
     Parameters
     ----------
     domain_name : str
@@ -372,8 +374,6 @@ def unassign_domain_workspaces(domain_name: str, workspace_names: str | List[str
         The Fabric workspace(s).
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/domains/unassign-domain-workspaces-by-ids?tabs=HTTP
-
     domain_id = resolve_domain_id(domain_name=domain_name)
 
     if isinstance(workspace_names, str):

sempy_labs/directlake/__init__.py

@@ -21,6 +21,7 @@ from sempy_labs.directlake._show_unsupported_directlake_objects import (
 )
 from sempy_labs.directlake._update_directlake_model_lakehouse_connection import (
     update_direct_lake_model_lakehouse_connection,
+    update_direct_lake_model_connection,
 )
 from sempy_labs.directlake._update_directlake_partition_entity import (
     update_direct_lake_partition_entity,
@@ -50,4 +51,5 @@ __all__ = [
     "add_table_to_direct_lake_semantic_model",
     "generate_direct_lake_semantic_model",
     "get_direct_lake_source",
+    "update_direct_lake_model_connection",
 ]

sempy_labs/directlake/_directlake_schema_sync.py

@@ -3,6 +3,7 @@ import sempy.fabric as fabric
 from sempy_labs.lakehouse import get_lakehouse_columns
 from sempy_labs.directlake._dl_helper import get_direct_lake_source
 from sempy_labs.tom import connect_semantic_model
+from sempy_labs._helper_functions import _convert_data_type
 from typing import Optional
 from sempy._utils._log import log
 import sempy_labs._icons as icons
@@ -88,7 +89,7 @@ def direct_lake_schema_sync(
                         f"{icons.yellow_dot} The '{lakeCName}' column exists in the '{lakeTName}' lakehouse table but not in the '{dataset}' semantic model within the '{workspace}' workspace."
                     )
                     if add_to_model:
-                        dt = icons.data_type_mapping.get(dType)
+                        dt = _convert_data_type(dType)
                         tom.add_data_column(
                             table_name=table_name,
                             column_name=lakeCName,

sempy_labs/directlake/_dl_helper.py

@@ -9,6 +9,7 @@ from sempy_labs._helper_functions import (
     retry,
     resolve_dataset_id,
     resolve_lakehouse_name,
+    _convert_data_type,
 )
 
 
@@ -130,6 +131,8 @@ def generate_direct_lake_semantic_model(
 
     dfLT = get_lakehouse_tables(lakehouse=lakehouse, workspace=lakehouse_workspace)
 
+    icons.sll_tags.append("GenerateDLModel")
+
     # Validate lakehouse tables
     for t in lakehouse_tables:
         if t not in dfLT["Table Name"].values:
@@ -178,7 +181,7 @@ def generate_direct_lake_semantic_model(
             for i, r in dfLC_filt.iterrows():
                 lakeCName = r["Column Name"]
                 dType = r["Data Type"]
-                dt = icons.data_type_mapping.get(dType)
+                dt = _convert_data_type(dType)
                 tom.add_data_column(
                     table_name=t,
                     column_name=lakeCName,

sempy_labs/directlake/_get_shared_expression.py

@@ -30,7 +30,7 @@ def get_shared_expression(
     workspace = fabric.resolve_workspace_name(workspace)
     if lakehouse is None:
         lakehouse_id = fabric.get_lakehouse_id()
-        lakehouse = resolve_lakehouse_name(lakehouse_id)
+        lakehouse = resolve_lakehouse_name(lakehouse_id, workspace)
 
     dfL = list_lakehouses(workspace=workspace)
     lakeDetail = dfL[dfL["Lakehouse Name"] == lakehouse]
@@ -39,6 +39,12 @@ def get_shared_expression(
     sqlepid = lakeDetail["SQL Endpoint ID"].iloc[0]
     provStatus = lakeDetail["SQL Endpoint Provisioning Status"].iloc[0]
 
+    parts = sqlEPCS.split(".", 1)
+    if parts:
+        parts[0] = parts[0].upper()
+
+    sqlEPCS = ".".join(parts)
+
     if provStatus == "InProgress":
         raise ValueError(
             f"{icons.red_dot} The SQL Endpoint for the '{lakehouse}' lakehouse within the '{workspace}' workspace has not yet been provisioned. Please wait until it has been provisioned."

sempy_labs/directlake/_guardrails.py

@@ -81,6 +81,7 @@ def get_directlake_guardrails_for_sku(sku_size: str) -> pd.DataFrame:
     """
 
     df = get_direct_lake_guardrails()
-    filtered_df = df[df["Fabric SKUs"] == sku_size]
+    col_name = df.columns[0]
+    filtered_df = df[df[col_name] == sku_size]
 
     return filtered_df

sempy_labs/directlake/_show_unsupported_directlake_objects.py

@@ -59,13 +59,7 @@ def show_unsupported_direct_lake_objects(
     dfR["From Column Data Type"] = merged_from["Data Type"]
     dfR["To Column Data Type"] = merged_to["Data Type"]
 
-    dfR_filt = dfR[
-        (
-            (dfR["From Column Data Type"] == "DateTime")
-            | (dfR["To Column Data Type"] == "DateTime")
-        )
-        | (dfR["From Column Data Type"] != dfR["To Column Data Type"])
-    ]
+    dfR_filt = dfR[(dfR["From Column Data Type"] != dfR["To Column Data Type"])]
     r = dfR_filt[
         [
             "From Table",

sempy_labs/directlake/_update_directlake_model_lakehouse_connection.py

@@ -1,5 +1,6 @@
 import sempy.fabric as fabric
 from sempy_labs.directlake._get_shared_expression import get_shared_expression
+from sempy_labs.directlake._generate_shared_expression import generate_shared_expression
 from sempy_labs._helper_functions import (
     resolve_lakehouse_name,
 )
@@ -53,6 +54,8 @@ def update_direct_lake_model_lakehouse_connection(
             f"Therefore it cannot be used to support the '{dataset}' semantic model within the '{workspace}' workspace."
         )
 
+    icons.sll_tags.append("UpdateDLConnection")
+
     shEx = get_shared_expression(lakehouse, lakehouse_workspace)
 
     with connect_semantic_model(
@@ -69,3 +72,78 @@ def update_direct_lake_model_lakehouse_connection(
     print(
         f"{icons.green_dot} The expression in the '{dataset}' semantic model has been updated to point to the '{lakehouse}' lakehouse in the '{lakehouse_workspace}' workspace."
     )
+
+
+def update_direct_lake_model_connection(
+    dataset: str,
+    workspace: Optional[str] = None,
+    source: Optional[str] = None,
+    source_type: Optional[str] = "Lakehouse",
+    source_workspace: Optional[str] = None,
+):
+    """
+    Remaps a Direct Lake semantic model's SQL Endpoint connection to a new lakehouse/warehouse.
+
+    Parameters
+    ----------
+    dataset : str
+        Name of the semantic model.
+    workspace : str, default=None
+        The Fabric workspace name in which the semantic model exists.
+        Defaults to None which resolves to the workspace of the attached lakehouse
+        or if no lakehouse attached, resolves to the workspace of the notebook.
+    source : str, default=None
+        The name of the Fabric lakehouse/warehouse used by the Direct Lake semantic model.
+        Defaults to None which resolves to the lakehouse attached to the notebook.
+    source_type : str, default="Lakehouse"
+        The type of source for the Direct Lake semantic model. Valid options: "Lakehouse", "Warehouse".
+    source_workspace : str, default=None
+        The Fabric workspace 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.
+    """
+
+    if workspace is None:
+        workspace = fabric.resolve_workspace_name(workspace)
+
+    source_type = source_type.capitalize()
+
+    if source_type not in ["Lakehouse", "Warehouse"]:
+        raise ValueError(
+            f"{icons.red_dot} The 'source_type' must be either 'Lakehouse' or 'Warehouse'."
+        )
+
+    if source_workspace is None:
+        source_workspace = workspace
+
+    if source is None:
+        source_id = fabric.get_lakehouse_id()
+        source = resolve_lakehouse_name(source_id, source_workspace)
+    else:
+        source_id = fabric.resolve_item_id(
+            item_name=source, type=source_type, workspace=source_workspace
+        )
+        source = fabric.resolve_item_name(
+            item_id=source_id, workspace=source_workspace, type=source_type
+        )
+
+    icons.sll_tags.append("UpdateDLConnection")
+
+    shEx = generate_shared_expression(
+        item_name=source, item_type=source_type, workspace=source_workspace
+    )
+
+    with connect_semantic_model(
+        dataset=dataset, readonly=False, workspace=workspace
+    ) as tom:
+
+        if not tom.is_direct_lake():
+            raise ValueError(
+                f"{icons.red_dot} The '{dataset}' 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}' semantic model has been updated to point to the '{source}' {source_type.lower()} in the '{source_workspace}' workspace."
+    )

sempy_labs/directlake/_update_directlake_partition_entity.py

@@ -3,6 +3,7 @@ import sempy.fabric as fabric
 from sempy_labs.tom import connect_semantic_model
 from sempy_labs._refresh_semantic_model import refresh_semantic_model
 from sempy_labs.directlake._dl_helper import get_direct_lake_source
+from sempy_labs._helper_functions import _convert_data_type
 from typing import List, Optional, Union
 import sempy_labs._icons as icons
 
@@ -12,7 +13,6 @@ def update_direct_lake_partition_entity(
     table_name: Union[str, List[str]],
     entity_name: Union[str, List[str]],
     workspace: Optional[str] = None,
-    **kwargs,
 ):
     """
     Remaps a table (or tables) in a Direct Lake semantic model to a table in a lakehouse.
@@ -31,28 +31,8 @@ def update_direct_lake_partition_entity(
         or if no lakehouse attached, resolves to the workspace of the notebook.
     """
 
-    if "lakehouse" in kwargs:
-        print(
-            "The 'lakehouse' parameter has been deprecated as it is no longer necessary. Please remove this parameter from the function going forward."
-        )
-        del kwargs["lakehouse"]
-    if "lakehouse_workspace" in kwargs:
-        print(
-            "The 'lakehouse_workspace' parameter has been deprecated as it is no longer necessary. Please remove this parameter from the function going forward."
-        )
-        del kwargs["lakehouse_workspace"]
-
-    workspace = fabric.resolve_workspace_name(workspace)
-
-    artifact_type, lakehouse_name, lakehouse_id, lakehouse_workspace_id = (
-        get_direct_lake_source(dataset=dataset, workspace=workspace)
-    )
-
-    if artifact_type == "Warehouse":
-        raise ValueError(
-            f"{icons.red_dot} This function is only valid for Direct Lake semantic models which source from lakehouses, not warehouses."
-        )
-    lakehouse_workspace = fabric.resolve_workspace_name(lakehouse_workspace_id)
+    if workspace is None:
+        workspace = fabric.resolve_workspace_name(workspace)
 
     # Support both str & list types
     if isinstance(table_name, str):
@@ -65,6 +45,8 @@ def update_direct_lake_partition_entity(
             f"{icons.red_dot} The 'table_name' and 'entity_name' arrays must be of equal length."
         )
 
+    icons.sll_tags.append("UpdateDLPartition")
+
     with connect_semantic_model(
         dataset=dataset, readonly=False, workspace=workspace
     ) as tom:
@@ -88,12 +70,11 @@ def update_direct_lake_partition_entity(
                 raise ValueError(
                     f"{icons.red_dot} The '{tName}' table in the '{dataset}' semantic model has not been updated."
                 )
-            else:
-                tom.model.Tables[tName].Partitions[part_name].EntityName = eName
-                print(
-                    f"{icons.green_dot} The '{tName}' table in the '{dataset}' semantic model has been updated to point to the '{eName}' table "
-                    f"in the '{lakehouse_name}' lakehouse within the '{lakehouse_workspace}' workspace."
-                )
+
+            tom.model.Tables[tName].Partitions[part_name].EntityName = eName
+            print(
+                f"{icons.green_dot} The '{tName}' table in the '{dataset}' semantic model has been updated to point to the '{eName}' table."
+            )
 
 
 def add_table_to_direct_lake_semantic_model(
@@ -149,9 +130,9 @@ def add_table_to_direct_lake_semantic_model(
 
         table_count = tom.model.Tables.Count
 
-        if tom.is_direct_lake() is False and table_count > 0:
+        if not tom.is_direct_lake() and table_count > 0:
             raise ValueError(
-                "This function is only valid for Direct Lake semantic models or semantic models with no tables."
+                f"{icons.red_dot} This function is only valid for Direct Lake semantic models or semantic models with no tables."
             )
 
         if any(
@@ -204,7 +185,7 @@ def add_table_to_direct_lake_semantic_model(
         for i, r in dfLC_filt.iterrows():
             lakeCName = r["Column Name"]
             dType = r["Data Type"]
-            dt = icons.data_type_mapping.get(dType)
+            dt = _convert_data_type(dType)
             tom.add_data_column(
                 table_name=table_name,
                 column_name=lakeCName,

sempy_labs/lakehouse/_get_lakehouse_tables.py

@@ -31,12 +31,14 @@ def get_lakehouse_tables(
     """
     Shows the tables of a lakehouse and their respective properties. Option to include additional properties relevant to Direct Lake guardrails.
 
+    This is a wrapper function for the following API: `Tables - List Tables <https://learn.microsoft.com/rest/api/fabric/lakehouse/tables/list-tables`_ plus extended capabilities.
+
     Parameters
     ----------
     lakehouse : str, default=None
         The Fabric lakehouse.
         Defaults to None which resolves to the lakehouse attached to the notebook.
-    lakehouse_workspace : str, default=None
+    workspace : str, default=None
         The Fabric workspace used by the lakehouse.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
@@ -155,7 +157,9 @@ def get_lakehouse_tables(
     if extended:
         intColumns = ["Files", "Row Groups", "Table Size"]
         df[intColumns] = df[intColumns].astype(int)
-        df["SKU"] = guardrail["Fabric SKUs"].iloc[0]
+
+        col_name = guardrail.columns[0]
+        df["SKU"] = guardrail[col_name].iloc[0]
         df["Parquet File Guardrail"] = guardrail["Parquet files per table"].iloc[0]
         df["Row Group Guardrail"] = guardrail["Row groups per table"].iloc[0]
         df["Row Count Guardrail"] = (

sempy_labs/lakehouse/_shortcuts.py

@@ -20,6 +20,8 @@ def create_shortcut_onelake(
     """
     Creates a `shortcut <https://learn.microsoft.com/fabric/onelake/onelake-shortcuts>`_ to a delta table in OneLake.
 
+    This is a wrapper function for the following API: `OneLake Shortcuts - Create Shortcut <https://learn.microsoft.com/rest/api/fabric/core/onelake-shortcuts/create-shortcut`_.
+
     Parameters
     ----------
     table_name : str
@@ -169,6 +171,8 @@ def delete_shortcut(
     """
     Deletes a shortcut.
 
+    This is a wrapper function for the following API: `OneLake Shortcuts - Delete Shortcut <https://learn.microsoft.com/rest/api/fabric/core/onelake-shortcuts/delete-shortcut`_.
+
     Parameters
     ----------
     shortcut_name : str

sempy_labs/migration/_migrate_calctables_to_lakehouse.py

@@ -282,6 +282,8 @@ def migrate_field_parameters(
     if new_dataset_workspace is None:
         new_dataset_workspace = workspace
 
+    icons.sll_tags.append("DirectLakeMigration")
+
     dfC = fabric.list_columns(dataset=dataset, workspace=workspace)
     dfC["Column Object"] = format_dax_object_name(dfC["Table Name"], dfC["Column Name"])
     dfP = fabric.list_partitions(dataset=dataset, workspace=workspace)
@@ -337,6 +339,7 @@ def migrate_field_parameters(
             try:
                 par = TOM.Partition()
                 par.Name = tName
+                par.Mode = TOM.ModeType.Import
 
                 parSource = TOM.CalculatedPartitionSource()
                 par.Source = parSource
@@ -345,7 +348,6 @@ def migrate_field_parameters(
                 tbl = TOM.Table()
                 tbl.Name = tName
                 tbl.LineageTag = generate_guid()
-                tbl.SourceLineageTag = generate_guid()
                 tbl.Partitions.Add(par)
 
                 columns = ["Value1", "Value2", "Value3"]
@@ -356,7 +358,6 @@ def migrate_field_parameters(
                     col.SourceColumn = "[" + colName + "]"
                     col.DataType = TOM.DataType.String
                     col.LineageTag = generate_guid()
-                    col.SourceLineageTag = generate_guid()
 
                     tbl.Columns.Add(col)
 

sempy_labs/migration/_migrate_calctables_to_semantic_model.py

@@ -64,6 +64,8 @@ def migrate_calc_tables_to_semantic_model(
         lakehouse_id = fabric.get_lakehouse_id()
         lakehouse = resolve_lakehouse_name(lakehouse_id, lakehouse_workspace)
 
+    icons.sll_tags.append("DirectLakeMigration")
+
     # Get calc tables but not field parameters
     dfP = fabric.list_partitions(dataset=dataset, workspace=workspace)
     dfP_filt = dfP[(dfP["Source Type"] == "Calculated")]

sempy_labs/migration/_migrate_model_objects_to_semantic_model.py

@@ -52,6 +52,8 @@ def migrate_model_objects_to_semantic_model(
     if new_dataset_workspace is None:
         new_dataset_workspace = workspace
 
+    icons.sll_tags.append("DirectLakeMigration")
+
     dfT = fabric.list_tables(dataset=dataset, workspace=workspace)
     dfC = fabric.list_columns(dataset=dataset, workspace=workspace)
     dfM = fabric.list_measures(dataset=dataset, workspace=workspace)
@@ -334,14 +336,6 @@ def migrate_model_objects_to_semantic_model(
                     print(
                         f"{icons.warning} The {relName} relationship was not created as Direct Lake does not support relationships based on columns with different data types."
                     )
-                # Direct Lake using DateTime columns
-                elif isDirectLake and (
-                    r.FromColumn.DataType == TOM.DataType.DateTime
-                    or r.ToColumn.DataType == TOM.DataType.DateTime
-                ):
-                    print(
-                        f"{icons.red_dot} The {relName} relationship was not created as Direct Lake does not support relationships based on columns of DateTime data type."
-                    )
                 # Columns do not exist in the new semantic model
                 elif not any(
                     c.Name == r.FromColumn.Name and c.Parent.Name == r.FromTable.Name

sempy_labs/migration/_migrate_tables_columns_to_semantic_model.py

@@ -61,6 +61,8 @@ def migrate_tables_columns_to_semantic_model(
         lakehouse_id = fabric.get_lakehouse_id()
         lakehouse = resolve_lakehouse_name(lakehouse_id, lakehouse_workspace)
 
+    icons.sll_tags.append("DirectLakeMigration")
+
     # Check that lakehouse is attached to the notebook
     if not lakehouse_attached() and (lakehouse is None and lakehouse_workspace is None):
         raise ValueError(
@@ -106,9 +108,24 @@ def migrate_tables_columns_to_semantic_model(
     with connect_semantic_model(
         dataset=new_dataset, readonly=False, workspace=new_dataset_workspace
     ) as tom:
+
+        # Additional updates
+        tom.set_annotation(
+            object=tom.model, name="__PBI_TimeIntelligenceEnabled", value="0"
+        )
+        tom.set_annotation(
+            object=tom.model, name="PBI_QueryOrder", value='["DatabaseQuery"]'
+        )
+
+        # Begin migration
         if not any(e.Name == "DatabaseQuery" for e in tom.model.Expressions):
             tom.add_expression("DatabaseQuery", expression=shEx)
             print(f"{icons.green_dot} The 'DatabaseQuery' expression has been added.")
+            tom.set_annotation(
+                object=tom.model.Expressions["DatabaseQuery"],
+                name="PBI_IncludeFutureArtifacts",
+                value="False",
+            )
 
         for i, r in dfT_filt.iterrows():
             tName = r["Name"]

sempy_labs/migration/_migration_validation.py

@@ -46,6 +46,8 @@ def migration_validation(
     if new_dataset_workspace is None:
         new_dataset_workspace = workspace
 
+    icons.sll_tags.append("DirectLakeMigration")
+
     dfA = list_semantic_model_objects(dataset=dataset, workspace=workspace)
     dfB = list_semantic_model_objects(
         dataset=new_dataset, workspace=new_dataset_workspace

sempy_labs/migration/_refresh_calc_tables.py

@@ -26,6 +26,7 @@ def refresh_calc_tables(dataset: str, workspace: Optional[str] = None):
 
     spark = SparkSession.builder.getOrCreate()
     workspace = fabric.resolve_workspace_name(workspace)
+    icons.sll_tags.append("DirectLakeMigration")
 
     @retry(
         sleep_time=1,

sempy_labs/report/__init__.py

@@ -1,7 +1,9 @@
 from sempy_labs.report._reportwrapper import (
     ReportWrapper,
 )
-
+from sempy_labs.report._paginated import (
+    get_report_datasources,
+)
 from sempy_labs.report._generate_report import (
     create_report_from_reportjson,
     get_report_definition,
@@ -39,4 +41,5 @@ __all__ = [
     "ReportWrapper",
     "report_bpa_rules",
     "run_report_bpa",
+    "get_report_datasources",
 ]

sempy_labs/report/_generate_report.py

@@ -23,6 +23,8 @@ def create_report_from_reportjson(
     """
     Creates a report based on a report.json file (and an optional themes.json file).
 
+    This is a wrapper function for the following API: `Items - Create Report <https://learn.microsoft.com/rest/api/fabric/report/items/create-report`_.
+
     Parameters
     ----------
     report : str
@@ -124,6 +126,8 @@ def update_report_from_reportjson(
     """
     Updates a report based on a report.json file.
 
+    This is a wrapper function for the following API: `Items - Update Report Definition <https://learn.microsoft.com/rest/api/fabric/report/items/update-report-definition`_.
+
     Parameters
     ----------
     report : str
@@ -179,6 +183,8 @@ def get_report_definition(report: str, workspace: Optional[str] = None) -> pd.Da
     """
     Gets the collection of definition files of a report.
 
+    This is a wrapper function for the following API: `Items - Get Report Definition <https://learn.microsoft.com/rest/api/fabric/report/items/get-report-definition`_.
+
     Parameters
     ----------
     report : str