semantic-link-labs 0.8.10__py3-none-any.whl → 0.9.0__py3-none-any.whl

sempy_labs/_eventstreams.py

@@ -8,9 +8,10 @@ from sempy_labs._helper_functions import (
     pagination,
 )
 from sempy.fabric.exceptions import FabricHTTPException
+from uuid import UUID
 
 
-def list_eventstreams(workspace: Optional[str] = None) -> pd.DataFrame:
+def list_eventstreams(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     """
     Shows the eventstreams within a workspace.
 
@@ -18,8 +19,8 @@ def list_eventstreams(workspace: Optional[str] = None) -> pd.DataFrame:
 
     Parameters
     ----------
-    workspace : str, default=None
-        The Fabric workspace name.
+    workspace : str | uuid.UUID, default=None
+        The Fabric workspace name or ID.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
 
@@ -31,7 +32,7 @@ def list_eventstreams(workspace: Optional[str] = None) -> pd.DataFrame:
 
     df = pd.DataFrame(columns=["Eventstream Name", "Eventstream Id", "Description"])
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     client = fabric.FabricRestClient()
     response = client.get(f"/v1/workspaces/{workspace_id}/eventstreams")
@@ -53,7 +54,7 @@ def list_eventstreams(workspace: Optional[str] = None) -> pd.DataFrame:
 
 
 def create_eventstream(
-    name: str, description: Optional[str] = None, workspace: Optional[str] = None
+    name: str, description: Optional[str] = None, workspace: Optional[str | UUID] = None
 ):
     """
     Creates a Fabric eventstream.
@@ -66,13 +67,13 @@ def create_eventstream(
         Name of the eventstream.
     description : str, default=None
         A description of the environment.
-    workspace : str, default=None
-        The Fabric workspace name.
+    workspace : str | uuid.UUID, default=None
+        The Fabric workspace name or ID.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
     """
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     request_body = {"displayName": name}
 
@@ -87,11 +88,11 @@ def create_eventstream(
     lro(client, response, status_codes=[201, 202])
 
     print(
-        f"{icons.green_dot} The '{name}' eventstream has been created within the '{workspace}' workspace."
+        f"{icons.green_dot} The '{name}' eventstream has been created within the '{workspace_name}' workspace."
     )
 
 
-def delete_eventstream(name: str, workspace: Optional[str] = None):
+def delete_eventstream(name: str, workspace: Optional[str | UUID] = None):
     """
     Deletes a Fabric eventstream.
 
@@ -101,16 +102,16 @@ def delete_eventstream(name: str, workspace: Optional[str] = None):
     ----------
     name: str
         Name of the eventstream.
-    workspace : str, default=None
-        The Fabric workspace name.
+    workspace : str | uuid.UUID, default=None
+        The Fabric workspace name or ID.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
     """
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     item_id = fabric.resolve_item_id(
-        item_name=name, type="Eventstream", workspace=workspace
+        item_name=name, type="Eventstream", workspace=workspace_id
     )
 
     client = fabric.FabricRestClient()
@@ -120,5 +121,5 @@ def delete_eventstream(name: str, workspace: Optional[str] = None):
         raise FabricHTTPException(response)
 
     print(
-        f"{icons.green_dot} The '{name}' eventstream within the '{workspace}' workspace has been deleted."
+        f"{icons.green_dot} The '{name}' eventstream within the '{workspace_name}' workspace has been deleted."
     )

sempy_labs/_external_data_shares.py

@@ -15,7 +15,7 @@ def create_external_data_share(
     item_type: str,
     paths: str | List[str],
     recipient: str,
-    workspace: Optional[str] = None,
+    workspace: Optional[str | UUID] = None,
 ):
     """
     Creates an external data share for a given path or list of paths in the specified item.
@@ -32,17 +32,15 @@ def create_external_data_share(
         The path or list of paths that are to be externally shared. Currently, only a single path is supported.
     recipient : str
         The email address of the recipient.
-    workspace : str, default=None
-        The Fabric workspace name.
+    workspace : str | uuid.UUID, default=None
+        The Fabric workspace name or ID.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/core/external-data-shares/create-external-data-share?tabs=HTTP
-
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
     item_id = fabric.resolve_item_id(
-        item_name=item_name, type=item_type, workspace=workspace
+        item_name=item_name, type=item_type, workspace=workspace_id
     )
 
     if isinstance(paths, str):
@@ -60,7 +58,7 @@ def create_external_data_share(
         raise FabricHTTPException(response)
 
     print(
-        f"{icons.green_dot} An external data share was created for the '{item_name}' {item_type} within the '{workspace}' workspace for the {paths} paths."
+        f"{icons.green_dot} An external data share was created for the '{item_name}' {item_type} within the '{workspace_name}' workspace for the {paths} paths."
     )
 
 
@@ -68,7 +66,7 @@ def revoke_external_data_share(
     external_data_share_id: UUID,
     item_name: str,
     item_type: str,
-    workspace: Optional[str] = None,
+    workspace: Optional[str | UUID] = None,
 ):
     """
     Revokes the specified external data share. Note: This action cannot be undone.
@@ -77,21 +75,21 @@ def revoke_external_data_share(
 
     Parameters
     ----------
-    external_data_share_id : UUID
+    external_data_share_id : uuid.UUID
         The external data share ID.
     item_name : str
         The item name.
     item_type : str
         The `item type <https://learn.microsoft.com/rest/api/fabric/core/items/list-items?tabs=HTTP#itemtype>`_.
-    workspace : str, default=None
-        The Fabric workspace name.
+    workspace : str | uuid.UUID, default=None
+        The Fabric workspace name or ID.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
     """
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
     item_id = fabric.resolve_item_id(
-        item_name=item_name, type=item_type, workspace=workspace
+        item_name=item_name, type=item_type, workspace=workspace_id
     )
 
     client = fabric.FabricRestClient()
@@ -103,12 +101,12 @@ def revoke_external_data_share(
         raise FabricHTTPException(response)
 
     print(
-        f"{icons.green_dot} The '{external_data_share_id}' external data share for the '{item_name}' {item_type} within the '{workspace}' workspace has been revoked."
+        f"{icons.green_dot} The '{external_data_share_id}' external data share for the '{item_name}' {item_type} within the '{workspace_name}' workspace has been revoked."
     )
 
 
 def list_external_data_shares_in_item(
-    item_name: str, item_type: str, workspace: Optional[str] = None
+    item_name: str, item_type: str, workspace: Optional[str | UUID] = None
 ) -> pd.DataFrame:
     """
     Returns a list of the external data shares that exist for the specified item.
@@ -121,8 +119,8 @@ def list_external_data_shares_in_item(
         The item name.
     item_type : str
         The `item type <https://learn.microsoft.com/rest/api/fabric/core/items/list-items?tabs=HTTP#itemtype>`_.
-    workspace : str, default=None
-        The Fabric workspace name.
+    workspace : str | uuid.UUID, default=None
+        The Fabric workspace name or ID.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
 
@@ -132,9 +130,9 @@ def list_external_data_shares_in_item(
         A pandas dataframe showing a list of the external data shares that exist for the specified item.
     """
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
     item_id = fabric.resolve_item_id(
-        item_name=item_name, type=item_type, workspace=workspace
+        item_name=item_name, type=item_type, workspace=workspace_id
     )
 
     client = fabric.FabricRestClient()

sempy_labs/_gateways.py

@@ -1,4 +1,5 @@
 import sempy.fabric as fabric
+from sempy._utils._log import log
 import pandas as pd
 from typing import Optional
 from sempy.fabric.exceptions import FabricHTTPException
@@ -13,6 +14,7 @@ from uuid import UUID
 import sempy_labs._icons as icons
 
 
+@log
 def list_gateways() -> pd.DataFrame:
     """
     Returns a list of all gateways the user has permission for, including on-premises, on-premises (personal mode), and virtual network gateways.
@@ -95,7 +97,7 @@ def delete_gateway(gateway: str | UUID):
 
     Parameters
     ----------
-    gateway : str | UUID
+    gateway : str | uuid.UUID
         The gateway name or ID.
     """
 
@@ -117,7 +119,7 @@ def list_gateway_role_assigments(gateway: str | UUID) -> pd.DataFrame:
 
     Parameters
     ----------
-    gateway : str | UUID
+    gateway : str | uuid.UUID
         The gateway name or ID.
 
     Returns
@@ -159,9 +161,9 @@ def delete_gateway_role_assignment(gateway: str | UUID, role_assignement_id: UUI
 
     Parameters
     ----------
-    gateway : str | UUID
+    gateway : str | uuid.UUID
         The gateway name or ID.
-    role_assignement_id : UUID
+    role_assignement_id : uuid.UUID
         The role assignment ID.
     """
 
@@ -204,9 +206,9 @@ def delete_gateway_member(gateway: str | UUID, gateway_member: str | UUID):
 
     Parameters
     ----------
-    gateway : str | UUID
+    gateway : str | uuid.UUID
         The gateway name or ID.
-    gateway_member : str | UUID
+    gateway_member : str | uuid.UUID
         The gateway member name or ID.
     """
 
@@ -234,7 +236,7 @@ def list_gateway_members(gateway: str | UUID) -> pd.DataFrame:
 
     Parameters
     ----------
-    gateway : str | UUID
+    gateway : str | uuid.UUID
         The gateway name or ID.
 
     Returns
@@ -298,7 +300,7 @@ def create_vnet_gateway(
     ----------
     name : str
         The gateway name.
-    capacity : str | UUID
+    capacity : str | uuid.UUID
         The capacity name or Id.
     inactivity_minutes_before_sleep : int
         The minutes of inactivity before the virtual network gateway goes into auto-sleep. Must be one of the following values: 30, 60, 90, 120, 150, 240, 360, 480, 720, 1440.
@@ -353,7 +355,7 @@ def update_on_premises_gateway(
 
     Parameters
     ----------
-    gateway : str | UUID
+    gateway : str | uuid.UUID
         The gateway name or ID.
     allow_cloud_connection_refresh : bool, default=None
         Whether to allow cloud connections to refresh through this on-premises gateway. True - Allow, False - Do not allow.
@@ -403,9 +405,9 @@ def update_vnet_gateway(
 
     Parameters
     ----------
-    gateway : str | UUID
+    gateway : str | uuid.UUID
         The gateway name or ID.
-    capacity: str | UUID
+    capacity: str | uuid.UUID
         The capacity name or ID.
     inactivity_minutes_before_sleep : int, default=None
         The minutes of inactivity before the virtual network gateway goes into auto-sleep. Must be one of the following values: 30, 60, 90, 120, 150, 240, 360, 480, 720, 1440.
@@ -451,11 +453,11 @@ def bind_semantic_model_to_gateway(
 
     Parameters
     ----------
-    dataset : str | UUID
+    dataset : str | uuid.UUID
         The name or ID of the semantic model.
-    gateway : str | UUID
+    gateway : str | uuid.UUID
         The name or ID of the gateway.
-    workspace : str | UUID, default=None
+    workspace : str | uuid.UUID, default=None
         The Fabric workspace name.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.