semantic-link-labs 0.8.6__py3-none-any.whl → 0.8.7__py3-none-any.whl

sempy_labs/_gateways.py

@@ -0,0 +1,439 @@
+import sempy.fabric as fabric
+import pandas as pd
+from typing import Optional
+from sempy.fabric.exceptions import FabricHTTPException
+from sempy_labs._helper_functions import (
+    pagination,
+    _is_valid_uuid,
+    resolve_capacity_id,
+)
+from uuid import UUID
+import sempy_labs._icons as icons
+
+
+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.
+
+    This is a wrapper function for the following API: `Gateways - List Gateways <https://learn.microsoft.com/rest/api/fabric/core/gateways/list-gateways>`_.
+
+    Returns
+    -------
+    pandas.DataFrame
+        A pandas dataframe showing a list of all gateways the user has permission for, including on-premises, on-premises (personal mode), and virtual network gateways.
+    """
+
+    client = fabric.FabricRestClient()
+    response = client.get("/v1/gateways")
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    responses = pagination(client, response)
+
+    df = pd.DataFrame(
+        columns=[
+            "Gateway Name",
+            "Gateway Id",
+            "Type",
+            "Public Key Exponent",
+            "Public Key Modulus",
+            "Version",
+            "Number Of Member Gateways",
+            "Load Balancing Setting",
+            "Allow Cloud Connection Refresh",
+            "Allow Custom Connectors",
+        ]
+    )
+
+    for r in responses:
+        for v in r.get("value", []):
+            new_data = {
+                "Gateway Name": v.get("displayName"),
+                "Gateway Id": v.get("id"),
+                "Type": v.get("type"),
+                "Public Key Exponent": v.get("publicKey", {}).get("exponent"),
+                "Public Key Modulus": v.get("publicKey", {}).get("modulus"),
+                "Version": v.get("version"),
+                "Number Of Member Gateways": v.get("numberOfMemberGateways", 0),
+                "Load Balancing Setting": v.get("loadBalancingSetting"),
+                "Allow Cloud Connection Refresh": v.get("allowCloudConnectionRefresh"),
+                "Allow Custom Connectors": v.get("allowCustomConnectors"),
+            }
+
+            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+
+    int_cols = ["Number Of Member Gateways"]
+    bool_cols = ["Allow Cloud Connection Refresh", "Allow Custom Connectors"]
+    df[bool_cols] = df[bool_cols].astype(bool)
+    df[int_cols] = df[int_cols].astype(int)
+
+    return df
+
+
+def _resolve_gateway_id(gateway: str | UUID) -> UUID:
+
+    dfG = list_gateways()
+    if _is_valid_uuid(gateway):
+        dfG_filt = dfG[dfG["Gateway Id"] == gateway]
+    else:
+        dfG_filt = dfG[dfG["Gateway Name"] == gateway]
+
+    if len(dfG_filt) == 0:
+        raise ValueError(f"{icons.red_dot} The '{gateway}' does not exist.")
+
+    return dfG_filt["Gateway Id"].iloc[0]
+
+
+def delete_gateway(gateway: str | UUID):
+    """
+    Deletes a gateway.
+
+    This is a wrapper function for the following API: `Gateways - Delete Gateway <https://learn.microsoft.com/rest/api/fabric/core/gateways/delete-gateway>`_.
+
+    Parameters
+    ----------
+    gateway : str | UUID
+        The gateway name or ID.
+    """
+
+    gateway_id = _resolve_gateway_id(gateway)
+    client = fabric.FabricRestClient()
+    response = client.delete(f"/v1/gateways/{gateway_id}")
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    print(f"{icons.green_dot} The '{gateway}' gateway has been deleted.")
+
+
+def list_gateway_role_assigments(gateway: str | UUID) -> pd.DataFrame:
+    """
+    Returns a list of gateway role assignments.
+
+    This is a wrapper function for the following API: `Gateways - List Gateway Role Assignments <https://learn.microsoft.com/rest/api/fabric/core/gateways/list-gateway-role-assignments>`_.
+
+    Parameters
+    ----------
+    gateway : str | UUID
+        The gateway name or ID.
+
+    Returns
+    -------
+    pandas.DataFrame
+        A pandas dataframe showing a list of gateway role assignments.
+    """
+
+    gateway_id = _resolve_gateway_id(gateway)
+    client = fabric.FabricRestClient()
+    response = client.get(f"/v1/gateways/{gateway_id}/roleAssignments")
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    df = pd.DataFrame(columns=[])
+
+    responses = pagination(client, response)
+
+    for r in responses:
+        for v in r.get("value", []):
+            new_data = {
+                "Gateway Role Assignment Id": v.get("id"),
+                "Principal Id": v.get("principal", {}).get("id"),
+                "Principal Type": v.get("principal", {}).get("type"),
+                "Role": v.get("role"),
+            }
+
+            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+
+    return df
+
+
+def delete_gateway_role_assignment(gateway: str | UUID, role_assignement_id: UUID):
+    """
+    Delete the specified role assignment for the gateway.
+
+    This is a wrapper function for the following API: `Gateways - Delete Gateway Role Assignment <https://learn.microsoft.com/rest/api/fabric/core/gateways/delete-gateway-role-assignment>`_.
+
+    Parameters
+    ----------
+    gateway : str | UUID
+        The gateway name or ID.
+    role_assignement_id : UUID
+        The role assignment ID.
+    """
+
+    gateway_id = _resolve_gateway_id(gateway)
+    client = fabric.FabricRestClient()
+    response = client.delete(
+        f"/v1/gateways/{gateway_id}/roleAssignments/{role_assignement_id}"
+    )
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    print(
+        f"{icons.green_dot} The '{role_assignement_id}' role assignment for the '{gateway}' gateway has been deleted."
+    )
+
+
+def _resolve_gateway_member_id(gateway: str | UUID, gateway_member: str | UUID) -> UUID:
+
+    gateway_id = _resolve_gateway_id(gateway)
+    dfM = list_gateway_members(gateway=gateway_id)
+
+    if _is_valid_uuid(gateway_member):
+        dfM_filt = dfM[dfM["Member Id"] == gateway_member]
+    else:
+        dfM_filt = dfM[dfM["Member Name"] == gateway_member]
+    if len(dfM_filt) == 0:
+        raise ValueError(
+            f"{icons.red_dot} The '{gateway_member}' gateway member does not exist within the '{gateway}' gateway."
+        )
+
+    return dfM_filt["Member Id"].iloc[0]
+
+
+def delete_gateway_member(gateway: str | UUID, gateway_member: str | UUID):
+    """
+    Delete gateway member of an on-premises gateway.
+
+    This is a wrapper function for the following API: `Gateways - Delete Gateway Member <https://learn.microsoft.com/rest/api/fabric/core/gateways/delete-gateway-member>`_.
+
+    Parameters
+    ----------
+    gateway : str | UUID
+        The gateway name or ID.
+    gateway_member : str | UUID
+        The gateway member name or ID.
+    """
+
+    gateway_id = _resolve_gateway_id(gateway)
+    member_id = _resolve_gateway_member_id(
+        gateway=gateway_id, gateway_member=gateway_member
+    )
+
+    client = fabric.FabricRestClient()
+    response = client.delete(f"/v1/gateways/{gateway_id}/members/{member_id}")
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    print(
+        f"{icons.green_dot} The '{member_id}' member for the '{gateway}' gateway has been deleted."
+    )
+
+
+def list_gateway_members(gateway: str | UUID) -> pd.DataFrame:
+    """
+    Lists gateway members of an on-premises gateway.
+
+    This is a wrapper function for the following API: `Gateways - List Gateway Members <https://learn.microsoft.com/rest/api/fabric/core/gateways/list-gateway-members>`_.
+
+    Parameters
+    ----------
+    gateway : str | UUID
+        The gateway name or ID.
+
+    Returns
+    -------
+    pandas.DataFrame
+        A pandas dataframe showing a list of gateway members of an on-premises gateway.
+    """
+
+    gateway_id = _resolve_gateway_id(gateway)
+    client = fabric.FabricRestClient()
+    response = client.get(f"/v1/gateways/{gateway_id}/members")
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    df = pd.DataFrame(
+        columns=[
+            "Member Id",
+            "Member Name",
+            "Public Key Exponent",
+            "Public Key Modulus",
+            "Version",
+            "Enabled",
+        ]
+    )
+
+    for v in response.json().get("value", []):
+        new_data = {
+            "Member Id": v.get("id"),
+            "Member Name": v.get("displayName"),
+            "Public Key Exponent": v.get("publicKey", {}).get("exponent"),
+            "Public Key Modulus": v.get("publicKey", {}).get("modulus"),
+            "Version": v.get("version"),
+            "Enabled": v.get("enabled"),
+        }
+
+        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+
+    bool_cols = ["Enabled"]
+    df[bool_cols] = df[bool_cols].astype(bool)
+
+    return df
+
+
+def create_vnet_gateway(
+    name: str,
+    capacity: str | UUID,
+    inactivity_minutes_before_sleep: int,
+    number_of_member_gateways: int,
+    subscription_id: str,
+    resource_group: str,
+    virtual_network: str,
+    subnet: str,
+):
+    """
+    Creates a virtual network gateway.
+
+    This is a wrapper function for the following API: `Gateways - Create Gateway <https://learn.microsoft.com/rest/api/fabric/core/gateways/create-gateway>`_.
+
+    Parameters
+    ----------
+    name : str
+        The gateway name.
+    capacity : str | 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.
+    number_of_member_gateways: int
+        The number of member gateways. A number between 1 and 7.
+    subscription_id : str
+        The subscription ID.
+    resource_group : str
+        The name of the resource group.
+    virtual_network : str
+        The name of the virtual network.
+    subnet : str
+        The name of the subnet.
+    """
+
+    client = fabric.FabricRestClient()
+
+    capacity_id = resolve_capacity_id(capacity)
+    payload = {
+        "type": "VirtualNetwork",
+        "displayName": name,
+        "capacityId": capacity_id,
+        "virtualNetworkAzureResource": {
+            "subscriptionId": subscription_id,
+            "resourceGroupName": resource_group,
+            "virtualNetworkName": virtual_network,
+            "subnetName": subnet,
+        },
+        "inactivityMinutesBeforeSleep": inactivity_minutes_before_sleep,
+        "numberOfMemberGateways": number_of_member_gateways,
+    }
+    response = client.post("/v1/gateways", json=payload)
+
+    if response.status_code != 201:
+        raise FabricHTTPException(response)
+
+    print(
+        f"{icons.green_dot} The '{name}' gateway was created within the '{capacity}' capacity."
+    )
+
+
+def update_on_premises_gateway(
+    gateway: str,
+    allow_cloud_connection_refresh: Optional[bool] = None,
+    allow_custom_connectors: Optional[bool] = None,
+    load_balancing_setting: Optional[str] = None,
+):
+    """
+    Updates an on-premises gateway.
+
+    This is a wrapper function for the following API: `Gateways - Update Gateway <https://learn.microsoft.com/rest/api/fabric/core/gateways/update-gateway>`_.
+
+    Parameters
+    ----------
+    gateway : str | 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.
+    allow_custom_connectors : bool, default=None
+        Whether to allow custom connectors to be used with this on-premises gateway. True - Allow, False - Do not allow.
+    load_balancing_setting : str, default=None
+        The `load balancing setting <https://learn.microsoft.com/rest/api/fabric/core/gateways/update-gateway?tabs=HTTP#loadbalancingsetting>`_ of the on-premises gateway.
+    """
+
+    gateway_id = _resolve_gateway_id(gateway)
+
+    payload = {}
+
+    if allow_cloud_connection_refresh is not None:
+        payload["allowCloudConnectionRefresh"] = allow_cloud_connection_refresh
+    if allow_custom_connectors is not None:
+        payload["allowCustomConnectors"] = allow_custom_connectors
+    if load_balancing_setting is not None:
+        payload["loadBalancingSetting"] = load_balancing_setting
+
+    if not payload:
+        raise ValueError(
+            f"{icons.yellow_dot} The '{gateway}' gateway has not been update as no valid settings were provided."
+        )
+
+    payload["type"] = "OnPremises"
+
+    client = fabric.FabricRestClient()
+    response = client.patch(f"/v1/gateways/{gateway_id}", json=payload)
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    print(f"{icons.green_dot} The '{gateway}' has been updated accordingly.")
+
+
+def update_vnet_gateway(
+    gateway: str,
+    capacity: str | UUID,
+    inactivity_minutes_before_sleep: Optional[int] = None,
+    number_of_member_gateways: Optional[int] = None,
+):
+    """
+    Updates a virtual network gateway.
+
+    This is a wrapper function for the following API: `Gateways - Update Gateway <https://learn.microsoft.com/rest/api/fabric/core/gateways/update-gateway>`_.
+
+    Parameters
+    ----------
+    gateway : str | UUID
+        The gateway name or ID.
+    capacity: str | 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.
+    number_of_member_gateways : int, default=None
+        The number of member gateways. A number between 1 and 7.
+    """
+
+    gateway_id = _resolve_gateway_id(gateway)
+
+    payload = {}
+
+    if capacity is not None:
+        capacity_id = resolve_capacity_id(capacity)
+        payload["capacityId"] = capacity_id
+    if inactivity_minutes_before_sleep is not None:
+        payload["inactivityMinutesBeforeSleep"] = inactivity_minutes_before_sleep
+    if number_of_member_gateways is not None:
+        payload["numberOfMemberGateways"] = number_of_member_gateways
+
+    if not payload:
+        raise ValueError(
+            f"{icons.yellow_dot} The '{gateway}' gateway has not been update as no valid settings were provided."
+        )
+
+    payload["type"] = "VirtualNetwork"
+
+    client = fabric.FabricRestClient()
+    response = client.patch(f"/v1/gateways/{gateway_id}", json=payload)
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    print(f"{icons.green_dot} The '{gateway}' has been updated accordingly.")

sempy_labs/_generate_semantic_model.py

@@ -54,40 +54,60 @@ def create_blank_semantic_model(
             f"{icons.red_dot} Compatiblity level must be at least {min_compat}."
         )
 
-    tmsl = f"""
-    {{
-        "createOrReplace": {{
-        "object": {{
-            "database": '{dataset}'
-        }},
-        "database": {{
-            "name": '{dataset}',
-            "compatibilityLevel": {compatibility_level},
-            "model": {{
-                "cultures": [
-                    {{
-                        "name": "en-US",
-                        "linguisticMetadata": {{
-                            "content": {{
-                                "Version": "1.0.0",
-                                "Language": "en-US"
-                            }},
-                            "contentType": "json"
+    # If the model does not exist
+    if len(dfD_filt) == 0:
+        tmsl = f"""
+        {{
+            "createOrReplace": {{
+            "object": {{
+                "database": '{dataset}'
+            }},
+            "database": {{
+                "name": '{dataset}',
+                "compatibilityLevel": {compatibility_level},
+                "model": {{
+                    "cultures": [
+                        {{
+                            "name": "en-US",
+                            "linguisticMetadata": {{
+                                "content": {{
+                                    "Version": "1.0.0",
+                                    "Language": "en-US"
+                                }},
+                                "contentType": "json"
+                            }}
                         }}
+                    ],
+                    "collation": "Latin1_General_100_BIN2_UTF8",
+                    "dataAccessOptions": {{
+                        "legacyRedirects": true,
+                        "returnErrorValuesAsNull": true,
+                    }},
+                    "defaultPowerBIDataSourceVersion": "powerBI_V3",
+                    "sourceQueryCulture": "en-US",
                     }}
-                ],
-                "collation": "Latin1_General_100_BIN2_UTF8",
-                "dataAccessOptions": {{
-                    "legacyRedirects": true,
-                    "returnErrorValuesAsNull": true,
-                }},
-                "defaultPowerBIDataSourceVersion": "powerBI_V3",
-                "sourceQueryCulture": "en-US",
                 }}
             }}
         }}
-    }}
-    """
+        """
+    else:
+        tmsl = f"""
+        {{
+            "createOrReplace": {{
+            "object": {{
+                "database": '{dataset}'
+            }},
+            "database": {{
+                "name": '{dataset}',
+                "compatibilityLevel": {compatibility_level},
+                "model": {{
+                "culture": "en-US",
+                "defaultPowerBIDataSourceVersion": "powerBI_V3"
+                }}
+            }}
+            }}
+        }}
+        """
 
     fabric.execute_tmsl(script=tmsl, workspace=workspace)
 
@@ -221,7 +241,7 @@ def update_semantic_model_from_bim(
         json=request_body,
     )
 
-    lro(client, response, status_codes=[200, 202])
+    lro(client, response, status_codes=[200, 202], return_status_code=True)
 
     print(
         f"{icons.green_dot} The '{dataset}' semantic model has been updated within the '{workspace}' workspace."
@@ -288,6 +308,7 @@ def deploy_semantic_model(
             dataset=target_dataset,
             bim_file=bim,
             workspace=target_workspace,
+            overwrite=overwrite,
         )
     # Update the semantic model if the model exists
     else:

sempy_labs/_git.py

@@ -217,7 +217,7 @@ def get_git_connection(workspace: Optional[str] = None) -> pd.DataFrame:
     return df
 
 
-def initialize_git_connection(workspace: Optional[str] = None):
+def initialize_git_connection(workspace: Optional[str] = None) -> str:
     """
     Initializes a connection for a workspace that is connected to Git.
 
@@ -229,6 +229,11 @@ def initialize_git_connection(workspace: Optional[str] = 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.
+
+    Returns
+    -------
+    str
+        Remote full SHA commit hash.
     """
 
     workspace, workspace_id = resolve_workspace_name_and_id(workspace)
@@ -245,6 +250,8 @@ def initialize_git_connection(workspace: Optional[str] = None):
         f"{icons.green_dot} The '{workspace}' workspace git connection has been initialized."
     )
 
+    return response.json().get("remoteCommitHash")
+
 
 def commit_to_git(
     comment: str, item_ids: str | List[str] = None, workspace: Optional[str] = None
@@ -324,14 +331,15 @@ def update_from_git(
 
     Parameters
     ----------
-    workspace_head : str
-        Full SHA hash that the workspace is synced to. This value may be null only after Initialize Connection.
-        In other cases, the system will validate that the given value is aligned with the head known to the system.
-    remove_commit_hash : str
+    remote_commit_hash : str
         Remote full SHA commit hash.
     confilict_resolution_policy : str
         The `conflict resolution policy <https://learn.microsoft.com/rest/api/fabric/core/git/update-from-git?tabs=HTTP#conflictresolutionpolicy>`_.
+    workspace_head : str
+        Full SHA hash that the workspace is synced to. This value may be null only after Initialize Connection.
+        In other cases, the system will validate that the given value is aligned with the head known to the system.
     allow_override : bool, default=False
+        User consent to override incoming items during the update from Git process. When incoming items are present and the allow override items is not specified or is provided as false, the update operation will not start. Default value is false.
     workspace : str, default=None
         The Fabric workspace name.
         Defaults to None which resolves to the workspace of the attached lakehouse

sempy_labs/_helper_functions.py

@@ -17,6 +17,17 @@ import numpy as np
 from IPython.display import display, HTML
 
 
+def _build_url(url: str, params: dict) -> str:
+    """
+    Build the url with a list of parameters
+    """
+    url_parts = list(urllib.parse.urlparse(url))
+    url_parts[4] = urllib.parse.urlencode(params)
+    url = urllib.parse.urlunparse(url_parts)
+
+    return url
+
+
 def create_abfss_path(
     lakehouse_id: UUID, lakehouse_workspace_id: UUID, delta_table_name: str
 ) -> str:
@@ -1219,7 +1230,7 @@ def _show_chart(spec, title):
                         <div id="vis"></div>
                     </td>
                 </tr>
-            </table>    
+            </table>
             <script type="text/javascript">
                 var spec = {spec};
                 var opt = {{"renderer": "canvas", "actions": false}};
@@ -1264,8 +1275,8 @@ def _process_and_display_chart(df, title, widget):
                 }
             },
             "x": {
-                "field": "Start", 
-                "type": "quantitative", 
+                "field": "Start",
+                "type": "quantitative",
                 "title": "milliseconds",
                 "axis": {
                     "titleFontSize": 20

sempy_labs/_list_functions.py

@@ -1371,7 +1371,7 @@ def list_report_semantic_model_objects(
     Shows a list of semantic model objects (i.e. columns, measures, hierarchies) used in all reports which feed data from
     a given semantic model.
 
-    Requirement: Reports must be in the PBIR format.
+    Note: As with all functions which rely on the ReportWrapper, this function requires the report(s) to be in the 'PBIR' format.
 
     Parameters
     ----------

sempy_labs/_model_auto_build.py

@@ -2,7 +2,7 @@ import sempy.fabric as fabric
 import pandas as pd
 from sempy_labs.tom import connect_semantic_model
 from sempy_labs._generate_semantic_model import create_blank_semantic_model
-from sempy_labs.directlake._get_shared_expression import get_shared_expression
+from sempy_labs.directlake._generate_shared_expression import generate_shared_expression
 from typing import Optional
 from sempy._utils._log import log
 
@@ -58,7 +58,9 @@ def model_auto_build(
     ) as tom:
 
         # DL Only
-        expr = get_shared_expression(lakehouse=lakehouse, workspace=lakehouse_workspace)
+        expr = generate_shared_expression(
+            item_name=lakehouse, item_type="Lakehouse", workspace=lakehouse_workspace
+        )
         tom.add_expression(name="DatbaseQuery", expression=expr)
 
         for sheet in sheets:

sempy_labs/_model_bpa.py

@@ -400,23 +400,10 @@ def run_model_bpa(
         dfExport["Timestamp"] = now
         dfExport["RunId"] = runId
         dfExport["Configured By"] = configured_by
-
         dfExport["RunId"] = dfExport["RunId"].astype("int")
 
-        colName = "Capacity Name"
-        dfExport.insert(0, colName, dfExport.pop(colName))
-        colName = "Capacity Id"
-        dfExport.insert(1, colName, dfExport.pop(colName))
-        colName = "Workspace Name"
-        dfExport.insert(2, colName, dfExport.pop(colName))
-        colName = "Workspace Id"
-        dfExport.insert(3, colName, dfExport.pop(colName))
-        colName = "Dataset Name"
-        dfExport.insert(4, colName, dfExport.pop(colName))
-        colName = "Configured By"
-        dfExport.insert(5, colName, dfExport.pop(colName))
-
-        dfExport.columns = dfExport.columns.str.replace(" ", "_")
+        dfExport = dfExport[list(icons.bpa_schema.keys())]
+        dfExport["RunId"] = dfExport["RunId"].astype("int")
         schema = {
             key.replace(" ", "_"): value for key, value in icons.bpa_schema.items()
         }