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

sempy_labs/report/_paginated.py

@@ -0,0 +1,74 @@
+import sempy.fabric as fabric
+from typing import Optional
+from sempy.fabric.exceptions import FabricHTTPException
+import pandas as pd
+
+
+def get_report_datasources(
+    report: str,
+    workspace: Optional[str] = None,
+) -> pd.DataFrame:
+    """
+    Returns a list of data sources for the specified paginated report (RDL) from the specified workspace.
+
+    Parameters
+    ----------
+    report : str | List[str]
+        Name(s) of the Power BI report(s).
+    workspace : str, default=None
+        The name of the Fabric workspace in which the report resides.
+        Defaults to None which resolves to the workspace of the attached lakehouse
+        or if no lakehouse attached, resolves to the workspace of the notebook.
+
+    Returns
+    -------
+    pandas.DataFrame
+        A pandas dataframe showing a list of data sources for the specified paginated report (RDL) from the specified workspace.
+    """
+
+    df = pd.DataFrame(
+        columns=[
+            "Report Name",
+            "Report Id",
+            "Datasource Id",
+            "Datasource Type",
+            "Gateway Id",
+            "Server",
+            "Database",
+        ]
+    )
+
+    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)
+
+    report_id = fabric.resolve_item_id(
+        item_name=report, type="PaginatedReport", workspace=workspace
+    )
+
+    client = fabric.PowerBIRestClient()
+
+    response = client.get(
+        f"/v1.0/myorg/groups/{workspace_id}/reports/{report_id}/datasources"
+    )
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    for i in response.json().get("value", []):
+        conn = i.get("connectionDetails", {})
+        new_data = {
+            "Report Name": report,
+            "Report Id": report_id,
+            "Datasource Id": i.get("datasourceId"),
+            "Datasource Type": i.get("datasourceType"),
+            "Gateway Id": i.get("gatewayId"),
+            "Server": conn.get("server") if conn else None,
+            "Database": conn.get("database") if conn else None,
+        }
+
+        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+
+    return df

sempy_labs/report/_report_functions.py

@@ -33,6 +33,8 @@ def get_report_json(
     """
     Gets the report.json file content of a Power BI 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
@@ -157,6 +159,8 @@ def export_report(
     """
     Exports a Power BI report to a file in your lakehouse.
 
+    This is a wrapper function for the following APIs: `Reports - Export To File In Group <https://learn.microsoft.com/rest/api/power-bi/reports/export-to-file-in-group`_, `Reports - Get Export To File Status In Group <https://learn.microsoft.com/rest/api/power-bi/reports/get-export-to-file-status-in-group>`_, `Reports - Get File Of Export To File In Group <https://learn.microsoft.com/rest/api/power-bi/reports/get-file-of-export-to-file-in-group>`_.
+
     Parameters
     ----------
     report : str
@@ -401,6 +405,8 @@ def clone_report(
     """
     Clones a Power BI report.
 
+    This is a wrapper function for the following API: `Reports - Clone Report In Group <https://learn.microsoft.com/rest/api/power-bi/reports/clone-report-in-group`_.
+
     Parameters
     ----------
     report : str

sempy_labs/report/_report_rebind.py

@@ -19,6 +19,8 @@ def report_rebind(
     """
     Rebinds a report to a semantic model.
 
+    This is a wrapper function for the following API: `Reports - Rebind Report In Group <https://learn.microsoft.com/rest/api/power-bi/reports/rebind-report-in-group`_.
+
     Parameters
     ----------
     report : str | List[str]

sempy_labs/tom/_model.py

@@ -6,6 +6,7 @@ from datetime import datetime
 from sempy_labs._helper_functions import (
     format_dax_object_name,
     generate_guid,
+    _make_list_unique,
 )
 from sempy_labs._list_functions import list_relationships
 from sempy_labs._refresh_semantic_model import refresh_semantic_model
@@ -15,6 +16,7 @@ from typing import List, Iterator, Optional, Union, TYPE_CHECKING
 from sempy._utils._log import log
 import sempy_labs._icons as icons
 from sempy.fabric.exceptions import FabricHTTPException
+import ast
 
 if TYPE_CHECKING:
     import Microsoft.AnalysisServices.Tabular
@@ -914,6 +916,7 @@ class TOMWrapper:
         part = TOM.Partition()
         part.Name = name
         part.Source = TOM.CalculationGroupSource()
+        part.Mode = TOM.ModeType.Import
         tbl.Partitions.Add(part)
 
         sortCol = "Ordinal"
@@ -988,10 +991,13 @@ class TOMWrapper:
         """
         import Microsoft.AnalysisServices.Tabular as TOM
 
-        cul = TOM.Culture()
-        cul.Name = language
-
         if not any(c.Name == language for c in self.model.Cultures):
+            cul = TOM.Culture()
+            cul.Name = language
+            lm = TOM.LinguisticMetadata()
+            lm.ContentType = TOM.ContentType.Json
+            lm.Content = f'{{"Version": "1.0.0", "Language": "{language}"}}'
+            cul.LinguisticMetadata = lm
             self.model.Cultures.Add(cul)
 
     def add_perspective(self, perspective_name: str):
@@ -2109,7 +2115,9 @@ class TOMWrapper:
             if c.Parent.Name == table_name and c.Parent.DataCategory == "Time"
         )
 
-    def mark_as_date_table(self, table_name: str, column_name: str):
+    def mark_as_date_table(
+        self, table_name: str, column_name: str, validate: bool = False
+    ):
         """
         Marks a table as a `date table <https://learn.microsoft.com/power-bi/transform-model/desktop-date-tables>`_.
 
@@ -2119,6 +2127,8 @@ class TOMWrapper:
             Name of the table.
         column_name : str
             Name of the date column in the table.
+        validate : bool, default=False
+            If True, performs a validation on if the the date table is viable.
         """
         import Microsoft.AnalysisServices.Tabular as TOM
 
@@ -2129,31 +2139,32 @@ class TOMWrapper:
                 f"{icons.red_dot} The column specified in the 'column_name' parameter in this function must be of DateTime data type."
             )
 
-        daxQuery = f"""
-        define measure '{table_name}'[test] = 
-        var mn = MIN('{table_name}'[{column_name}])
-        var ma = MAX('{table_name}'[{column_name}])
-        var x = COUNTROWS(DISTINCT('{table_name}'[{column_name}]))
-        var y = DATEDIFF(mn, ma, DAY) + 1
-        return if(y = x, 1,0)
-
-        EVALUATE
-        SUMMARIZECOLUMNS(
-        "1",[test]
-        )
-        """
-        df = fabric.evaluate_dax(
-            dataset=self._dataset, workspace=self._workspace, dax_string=daxQuery
-        )
-        value = df["1"].iloc[0]
-        if value != "1":
-            raise ValueError(
-                f"{icons.red_dot} The '{column_name}' within the '{table_name}' table does not contain contiguous date values."
+        if validate:
+            dax_query = f"""
+            define measure '{table_name}'[test] = 
+            var mn = MIN('{table_name}'[{column_name}])
+            var ma = MAX('{table_name}'[{column_name}])
+            var x = COUNTROWS(DISTINCT('{table_name}'[{column_name}]))
+            var y = DATEDIFF(mn, ma, DAY) + 1
+            return if(y = x, 1,0)
+
+            EVALUATE
+            SUMMARIZECOLUMNS(
+            "1",[test]
+            )
+            """
+            df = fabric.evaluate_dax(
+                dataset=self._dataset, workspace=self._workspace, dax_string=dax_query
             )
+            value = df["[1]"].iloc[0]
+            if value != "1":
+                raise ValueError(
+                    f"{icons.red_dot} The '{column_name}' within the '{table_name}' table does not contain contiguous date values."
+                )
 
         # Mark as a date table
         t.DataCategory = "Time"
-        c.Columns[column_name].IsKey = True
+        c.IsKey = True
         print(
             f"{icons.green_dot} The '{table_name}' table has been marked as a date table using the '{column_name}' column as its primary date key."
         )
@@ -2692,6 +2703,7 @@ class TOMWrapper:
 
         par = TOM.Partition()
         par.Name = name
+        par.Mode = TOM.ModeType.Import
 
         parSource = TOM.CalculatedPartitionSource()
         parSource.Expression = expression
@@ -2949,6 +2961,8 @@ class TOMWrapper:
             runId = "1"
         self.set_annotation(object=self.model, name="Vertipaq_Run", value=runId)
 
+        icons.sll_tags.append("VertipaqAnnotations")
+
     def row_count(self, object: Union["TOM.Partition", "TOM.Table"]):
         """
         Obtains the row count of a table or partition within a semantic model.
@@ -4182,15 +4196,7 @@ class TOMWrapper:
             Name of the column to use for sorting. Must be of integer (Int64) data type.
         """
 
-        import Microsoft.AnalysisServices.Tabular as TOM
-
         sbc = self.model.Tables[table_name].Columns[sort_by_column]
-
-        if sbc.DataType != TOM.DataType.Int64:
-            raise ValueError(
-                f"{icons.red_dot} Invalid sort by column data type. The sort by column must be of 'Int64' data type."
-            )
-
         self.model.Tables[table_name].Columns[column_name].SortByColumn = sbc
 
     def remove_sort_by_column(self, table_name: str, column_name: str):
@@ -4476,6 +4482,7 @@ class TOMWrapper:
         #     executor.map(process_measure, self.all_measures())
 
     def close(self):
+
         if not self._readonly and self.model is not None:
 
             import Microsoft.AnalysisServices.Tabular as TOM
@@ -4500,6 +4507,30 @@ class TOMWrapper:
                     if self._column_map.get(c.LineageTag)[1] != c.DataType:
                         self.add_changed_property(object=c, property="DataType")
 
+            tags = [f"{icons.sll_prefix}{a}" for a in icons.sll_tags]
+            tags.append("SLL")
+
+            if not any(a.Name == icons.sll_ann_name for a in self.model.Annotations):
+                ann_list = _make_list_unique(tags)
+                new_ann_value = str(ann_list).replace("'", '"')
+                self.set_annotation(
+                    object=self.model, name=icons.sll_ann_name, value=new_ann_value
+                )
+            else:
+                try:
+                    ann_value = self.get_annotation_value(
+                        object=self.model, name=icons.sll_ann_name
+                    )
+                    ann_list = ast.literal_eval(ann_value)
+                    ann_list += tags
+                    ann_list = _make_list_unique(ann_list)
+                    new_ann_value = str(ann_list).replace("'", '"')
+                    self.set_annotation(
+                        object=self.model, name=icons.sll_ann_name, value=new_ann_value
+                    )
+                except Exception:
+                    pass
+
             self.model.SaveChanges()
 
             if len(self._tables_added) > 0: