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

sempy_labs/report/_report_rebind.py

@@ -1,14 +1,13 @@
 import sempy
 import sempy.fabric as fabric
 from sempy_labs._helper_functions import resolve_dataset_id, resolve_report_id
-from typing import List, Optional, Union
+from typing import Optional, List
 from sempy._utils._log import log
 import sempy_labs._icons as icons
 
-
 @log
 def report_rebind(
-    report: str,
+    report: str | List[str],
     dataset: str,
     report_workspace: Optional[str] = None,
     dataset_workspace: Optional[str] = None,
@@ -18,8 +17,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
@@ -36,36 +35,37 @@ def report_rebind(
 
     """
 
-    if report_workspace == None:
+    if report_workspace is None:
         report_workspace_id = fabric.get_workspace_id()
         report_workspace = fabric.resolve_workspace_name(report_workspace_id)
     else:
         report_workspace_id = fabric.resolve_workspace_id(report_workspace)
-    if dataset_workspace == None:
+    if dataset_workspace is None:
         dataset_workspace = report_workspace
 
     client = fabric.PowerBIRestClient()
 
-    reportId = resolve_report_id(report=report, workspace=report_workspace)
-    datasetId = resolve_dataset_id(dataset=dataset, workspace=dataset_workspace)
-
-    # Prepare API
-    request_body = {"datasetId": datasetId}
+    if isinstance(report, str):
+        report = [report]
+    
+    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."
-        )
-    else:
-        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."
+        response = client.post(
+            f"/v1.0/myorg/groups/{report_workspace_id}/reports/{reportId}/Rebind",
+            json=request_body,
         )
 
+        if response.status_code == 200:
+            print(
+                f"{icons.green_dot} The '{rpt}' report has been successfully rebinded to the '{dataset}' semantic model."
+            )
+        else:
+            raise ValueError(f"{icons.red_dot} The '{rpt}' report within the '{report_workspace}' workspace failed to rebind to the '{dataset}' semantic model within the '{dataset_workspace}' workspace.")
 
 @log
 def report_rebind_all(
@@ -73,7 +73,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.
@@ -94,8 +94,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.
 
@@ -104,28 +104,27 @@ def report_rebind_all(
 
     """
 
-    if dataset_workspace == 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 == None:
+    if new_dataset_workpace is None:
         new_dataset_workpace = dataset_workspace
 
-    if report_workspace == None:
+    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

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