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

sempy_labs/report/_report_functions.py

@@ -1,7 +1,10 @@
-import sempy
 import sempy.fabric as fabric
 import pandas as pd
-import json, os, time, base64, copy
+import json
+import os
+import time
+import base64
+import copy
 from anytree import Node, RenderTree
 from powerbiclient import Report
 from synapse.ml.services import Translate
@@ -20,6 +23,7 @@ from sempy_labs._helper_functions import (
 from typing import Any, List, Optional, Union
 from sempy._utils._log import log
 import sempy_labs._icons as icons
+from sempy.fabric.exceptions import FabricHTTPException
 
 
 def get_report_json(
@@ -55,10 +59,9 @@ def get_report_json(
     dfI_filt = dfI[(dfI["Display Name"] == report)]
 
     if len(dfI_filt) == 0:
-        print(
+        raise ValueError(
             f"{icons.red_dot} The '{report}' report does not exist in the '{workspace}' workspace."
         )
-        return
 
     itemId = dfI_filt["Id"].iloc[0]
     response = client.post(
@@ -74,10 +77,9 @@ def get_report_json(
     if save_to_file_name is not None:
         lakeAttach = lakehouse_attached()
         if lakeAttach is False:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} In order to save the report.json file, a lakehouse must be attached to the notebook. Please attach a lakehouse to this notebook."
             )
-            return
 
         lakehouse_id = fabric.get_lakehouse_id()
         lakehouse = resolve_lakehouse_name(lakehouse_id, workspace)
@@ -191,10 +193,9 @@ def export_report(
     lakeAttach = lakehouse_attached()
 
     if lakeAttach is False:
-        print(
+        raise ValueError(
             f"{icons.red_dot} In order to run the 'export_report' function, a lakehouse must be attached to the notebook. Please attach a lakehouse to this notebook."
         )
-        return
 
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
@@ -204,15 +205,14 @@ def export_report(
         visual_name = [visual_name]
 
     if bookmark_name is not None and (page_name is not None or visual_name is not None):
-        print(
+        raise ValueError(
             f"{icons.red_dot} If the 'bookmark_name' parameter is set, the 'page_name' and 'visual_name' parameters must not be set."
         )
-        return
+
     if visual_name is not None and page_name is None:
-        print(
+        raise ValueError(
             f"{icons.red_dot} If the 'visual_name' parameter is set, the 'page_name' parameter must be set."
         )
-        return
 
     validFormats = {
         "ACCESSIBLEPDF": ".pdf",
@@ -235,10 +235,9 @@ def export_report(
 
     fileExt = validFormats.get(export_format)
     if fileExt is None:
-        print(
+        raise ValueError(
             f"{icons.red_dot} The '{export_format}' format is not a valid format for exporting Power BI reports. Please enter a valid format. Options: {validFormats}"
         )
-        return
 
     if file_name is None:
         file_name = report + fileExt
@@ -255,10 +254,9 @@ def export_report(
     ]
 
     if len(dfI_filt) == 0:
-        print(
+        raise ValueError(
             f"{icons.red_dot} The '{report}' report does not exist in the '{workspace}' workspace."
         )
-        return
 
     reportType = dfI_filt["Type"].iloc[0]
 
@@ -279,23 +277,21 @@ def export_report(
     ]
 
     if reportType == "Report" and export_format in paginatedOnly:
-        print(
+        raise ValueError(
             f"{icons.red_dot} The '{export_format}' format is only supported for paginated reports."
         )
-        return
+
     if reportType == "PaginatedReport" and export_format in pbiOnly:
-        print(
+        raise ValueError(
             f"{icons.red_dot} The '{export_format}' format is only supported for Power BI reports."
         )
-        return
 
     if reportType == "PaginatedReport" and (
         bookmark_name is not None or page_name is not None or visual_name is not None
     ):
-        print(
+        raise ValueError(
             f"{icons.red_dot} Export for paginated reports does not support bookmarks/pages/visuals. Those parameters must not be set for paginated reports."
         )
-        return
 
     reportId = dfI_filt["Id"].iloc[0]
     client = fabric.PowerBIRestClient()
@@ -332,19 +328,19 @@ def export_report(
             for page in page_name:
                 dfPage_filt = dfPage[dfPage["Page ID"] == page]
                 if len(dfPage_filt) == 0:
-                    print(
+                    raise ValueError(
                         f"{icons.red_dot} The '{page}' page does not exist in the '{report}' report within the '{workspace}' workspace."
                     )
-                    return
+
                 page_dict = {"pageName": page}
                 request_body["powerBIReportConfiguration"]["pages"].append(page_dict)
 
     elif page_name is not None and visual_name is not None:
         if len(page_name) != len(visual_name):
-            print(
+            raise ValueError(
                 f"{icons.red_dot} Each 'visual_name' must map to a single 'page_name'."
             )
-            return
+
         if reportType == "Report":
             request_body = {"format": export_format, "powerBIReportConfiguration": {}}
 
@@ -356,10 +352,10 @@ def export_report(
                     (dfVisual["Page ID"] == page) & (dfVisual["Visual ID"] == visual)
                 ]
                 if len(dfVisual_filt) == 0:
-                    print(
+                    raise ValueError(
                         f"{icons.red_dot} The '{visual}' visual does not exist on the '{page}' in the '{report}' report within the '{workspace}' workspace."
                     )
-                    return
+
                 page_dict = {"pageName": page, "visualName": visual}
                 request_body["powerBIReportConfiguration"]["pages"].append(page_dict)
                 a += 1
@@ -393,7 +389,7 @@ def export_report(
             )
             response_body = json.loads(response.content)
         if response_body["status"] == "Failed":
-            print(
+            raise ValueError(
                 f"{icons.red_dot} The export for the '{report}' report within the '{workspace}' workspace in the '{export_format}' format has failed."
             )
         else:
@@ -447,10 +443,9 @@ def clone_report(
     dfI_filt = dfI[(dfI["Display Name"] == report)]
 
     if len(dfI_filt) == 0:
-        print(
+        raise ValueError(
             f"{icons.red_dot} The '{report}' report does not exist within the '{workspace}' workspace."
         )
-        return
 
     reportId = resolve_report_id(report, workspace)
 
@@ -462,8 +457,10 @@ def clone_report(
         dfW_filt = dfW[dfW["Name"] == target_workspace]
 
         if len(dfW_filt) == 0:
-            print(f"{icons.red_dot} The '{workspace}' is not a valid workspace.")
-            return
+            raise ValueError(
+                f"{icons.red_dot} The '{workspace}' is not a valid workspace."
+            )
+
         target_workspace_id = dfW_filt["Id"].iloc[0]
 
     if target_dataset is None:
@@ -478,10 +475,10 @@ def clone_report(
         dfD_filt = dfD[dfD["Dataset Name"] == target_dataset]
 
         if len(dfD_filt) == 0:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} The '{target_dataset}' target dataset does not exist in the '{target_workspace}' workspace."
             )
-            return
+
         target_dataset_id = dfD_filt["Dataset Id"].iloc[0]
 
     client = fabric.PowerBIRestClient()
@@ -503,14 +500,12 @@ def clone_report(
         f"/v1.0/myorg/groups/{workspace_id}/reports/{reportId}/Clone", json=request_body
     )
 
-    if response.status_code == 200:
-        print(
-            f"{icons.green_dot} The '{report}' report has been successfully cloned as the '{cloned_report}' report within the '{target_workspace}' workspace using the '{target_dataset}' semantic model."
-        )
-    else:
-        print(
-            f"{icons.red_dot} POST request failed with status code: {response.status_code}"
-        )
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+    print(
+        f"{icons.green_dot} The '{report}' report has been successfully cloned as the '{cloned_report}' report within the"
+        f" '{target_workspace}' workspace using the '{target_dataset}' semantic model."
+    )
 
 
 def launch_report(report: str, workspace: Optional[str] = None):
@@ -573,21 +568,21 @@ def list_report_pages(report: str, workspace: Optional[str] = None):
     reportJson = get_report_json(report=report, workspace=workspace)
 
     for section in reportJson["sections"]:
-        pageID = section["name"]
-        pageName = section["displayName"]
+        pageID = section.get("name")
+        pageName = section.get("displayName")
         # pageFilters = section['filters']
-        pageWidth = section["width"]
-        pageHeight = section["height"]
+        pageWidth = section.get("width")
+        pageHeight = section.get("height")
         visualCount = len(section["visualContainers"])
         pageHidden = False
-        pageConfig = section["config"]
+        pageConfig = section.get("config")
         pageConfigJson = json.loads(pageConfig)
 
         try:
             pageH = pageConfigJson["visibility"]
             if pageH == 1:
                 pageHidden = True
-        except:
+        except Exception:
             pass
 
         new_data = {
@@ -648,7 +643,7 @@ def list_report_visuals(report: str, workspace: Optional[str] = None):
                     "properties"
                 ]["text"]["expr"]["Literal"]["Value"]
                 title = title[1:-1]
-            except:
+            except Exception:
                 title = ""
 
             new_data = {
@@ -716,7 +711,7 @@ def list_report_bookmarks(report: str, workspace: Optional[str] = None):
                         ][vc]["singleVisual"]["display"]["mode"]
                         if hidden == "hidden":
                             vHidden = True
-                    except:
+                    except Exception:
                         pass
 
             new_data = {
@@ -744,7 +739,7 @@ def list_report_bookmarks(report: str, workspace: Optional[str] = None):
 
         return df
 
-    except:
+    except Exception:
         print(
             f"The '{report}' report within the '{workspace}' workspace has no bookmarks."
         )

sempy_labs/report/_report_rebind.py

@@ -1,13 +1,14 @@
-import sempy
 import sempy.fabric as fabric
 from sempy_labs._helper_functions import resolve_dataset_id, resolve_report_id
-from typing import Optional
+from typing import Optional, List
 from sempy._utils._log import log
 import sempy_labs._icons as icons
+from sempy.fabric.exceptions import FabricHTTPException
+
 
 @log
 def report_rebind(
-    report: str,
+    report: str | List[str],
     dataset: str,
     report_workspace: Optional[str] = None,
     dataset_workspace: Optional[str] = None,
@@ -17,8 +18,8 @@ def report_rebind(
 
     Parameters
     ----------
-    report : str
-        Name of the Power BI report.
+    report : str | List[str]
+        Name(s) of the Power BI report(s).
     dataset : str
         Name of the semantic model.
     report_workspace : str, default=None
@@ -45,24 +46,25 @@ def report_rebind(
 
     client = fabric.PowerBIRestClient()
 
-    reportId = resolve_report_id(report=report, workspace=report_workspace)
-    datasetId = resolve_dataset_id(dataset=dataset, workspace=dataset_workspace)
+    if isinstance(report, str):
+        report = [report]
 
-    # Prepare API
-    request_body = {"datasetId": datasetId}
+    for rpt in report:
+        reportId = resolve_report_id(report=rpt, workspace=report_workspace)
+        datasetId = resolve_dataset_id(dataset=dataset, workspace=dataset_workspace)
 
-    response = client.post(
-        f"/v1.0/myorg/groups/{report_workspace_id}/reports/{reportId}/Rebind",
-        json=request_body,
-    )
+        # Prepare API
+        request_body = {"datasetId": datasetId}
 
-    if response.status_code == 200:
-        print(
-            f"{icons.green_dot} The '{report}' report has been successfully rebinded to the '{dataset}' semantic model."
+        response = client.post(
+            f"/v1.0/myorg/groups/{report_workspace_id}/reports/{reportId}/Rebind",
+            json=request_body,
         )
-    else:
+
+        if response.status_code != 200:
+            raise FabricHTTPException(response)
         print(
-            f"{icons.red_dot} The '{report}' report within the '{report_workspace}' workspace failed to rebind to the '{dataset}' semantic model within the '{dataset_workspace}' workspace."
+            f"{icons.green_dot} The '{rpt}' report has been successfully rebinded to the '{dataset}' semantic model."
         )
 
 
@@ -72,7 +74,7 @@ def report_rebind_all(
     new_dataset: str,
     dataset_workspace: Optional[str] = None,
     new_dataset_workpace: Optional[str] = None,
-    report_workspace: Optional[str] = None,
+    report_workspace: Optional[str | List[str]] = None,
 ):
     """
     Rebinds all reports in a workspace which are bound to a specific semantic model to a new semantic model.
@@ -93,8 +95,8 @@ def report_rebind_all(
         The name of the Fabric workspace in which the new semantic model 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.
-    report_workspace : str, default=None
-        The name of the Fabric workspace in which the report resides.
+    report_workspace : str | List[str], default=None
+        The name(s) of the Fabric workspace(s) in which the report(s) reside(s).
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
 
@@ -103,11 +105,7 @@ def report_rebind_all(
 
     """
 
-    if dataset_workspace is None:
-        dataset_workspace_id = fabric.get_workspace_id()
-        dataset_workspace = fabric.resolve_workspace_name(dataset_workspace_id)
-    else:
-        dataset_workspace_id = fabric.resolve_workspace_id(dataset_workspace)
+    dataset_workspace = fabric.resolve_workspace_name()
 
     if new_dataset_workpace is None:
         new_dataset_workpace = dataset_workspace
@@ -115,16 +113,19 @@ def report_rebind_all(
     if report_workspace is None:
         report_workspace = dataset_workspace
 
-    datasetId = resolve_dataset_id(dataset, dataset_workspace)
+    if isinstance(report_workspace, str):
+        report_workspace = [report_workspace]
 
-    dfRep = fabric.list_reports(workspace=report_workspace)
-    dfRep_filt = dfRep[dfRep["Dataset Id"] == datasetId]
+    datasetId = resolve_dataset_id(dataset, dataset_workspace)
 
-    for i, r in dfRep_filt.iterrows():
-        rptName = r["Name"]
-        report_rebind(
-            report=rptName,
-            dataset=new_dataset,
-            report_workspace=report_workspace,
-            dataset_workspace=new_dataset_workpace,
-        )
+    for rw in report_workspace:
+        dfRep = fabric.list_reports(workspace=rw)
+        dfRep_filt = dfRep[dfRep["Dataset Id"] == datasetId]
+        for i, r in dfRep_filt.iterrows():
+            rptName = r["Name"]
+            report_rebind(
+                report=rptName,
+                dataset=new_dataset,
+                report_workspace=rw,
+                dataset_workspace=new_dataset_workpace,
+            )

sempy_labs/tom/__init__.py

@@ -1,6 +1,3 @@
 from sempy_labs.tom._model import TOMWrapper, connect_semantic_model
 
-__all__ = [
-    "TOMWrapper",
-    "connect_semantic_model"
-]
+__all__ = ["TOMWrapper", "connect_semantic_model"]