semantic-link-labs 0.7.4__py3-none-any.whl → 0.8.0__py3-none-any.whl

sempy_labs/_sql.py

@@ -6,7 +6,7 @@ import struct
 import uuid
 from itertools import chain, repeat
 from sempy.fabric.exceptions import FabricHTTPException
-from sempy_labs._helper_functions import resolve_warehouse_id
+from sempy_labs._helper_functions import resolve_warehouse_id, resolve_lakehouse_id
 
 
 def bytes2mswin_bstr(value: bytes) -> bytes:
@@ -28,30 +28,48 @@ def bytes2mswin_bstr(value: bytes) -> bytes:
     return struct.pack("<i", len(encoded_bytes)) + encoded_bytes
 
 
-class ConnectWarehouse:
+class ConnectBase:
     def __init__(
         self,
-        warehouse: str,
+        name: str,
         workspace: Optional[Union[str, uuid.UUID]] = None,
         timeout: Optional[int] = None,
+        endpoint_type: str = "warehouse",
     ):
         from sempy.fabric._token_provider import SynapseTokenProvider
         import pyodbc
 
         workspace = fabric.resolve_workspace_name(workspace)
         workspace_id = fabric.resolve_workspace_id(workspace)
-        warehouse_id = resolve_warehouse_id(warehouse=warehouse, workspace=workspace)
 
-        # get the TDS endpoint
+        # Resolve the appropriate ID (warehouse or lakehouse)
+        if endpoint_type == "warehouse":
+            resource_id = resolve_warehouse_id(warehouse=name, workspace=workspace)
+        else:
+            resource_id = resolve_lakehouse_id(lakehouse=name, workspace=workspace)
+
+        # Get the TDS endpoint
         client = fabric.FabricRestClient()
-        response = client.get(f"v1/workspaces/{workspace_id}/warehouses/{warehouse_id}")
+        response = client.get(
+            f"v1/workspaces/{workspace_id}/{endpoint_type}s/{resource_id}"
+        )
         if response.status_code != 200:
             raise FabricHTTPException(response)
-        tds_endpoint = response.json().get("properties", {}).get("connectionString")
 
+        if endpoint_type == "warehouse":
+            tds_endpoint = response.json().get("properties", {}).get("connectionString")
+        else:
+            tds_endpoint = (
+                response.json()
+                .get("properties", {})
+                .get("sqlEndpointProperties", {})
+                .get("connectionString")
+            )
+
+        # Set up the connection string
         access_token = SynapseTokenProvider()()
         tokenstruct = bytes2mswin_bstr(access_token.encode())
-        conn_str = f"DRIVER={{ODBC Driver 18 for SQL Server}};SERVER={tds_endpoint};DATABASE={warehouse};Encrypt=Yes;"
+        conn_str = f"DRIVER={{ODBC Driver 18 for SQL Server}};SERVER={tds_endpoint};DATABASE={name};Encrypt=Yes;"
 
         if timeout is not None:
             conn_str += f"Connect Timeout={timeout};"
@@ -63,7 +81,7 @@ class ConnectWarehouse:
         self, sql: Union[str, List[str]]
     ) -> Union[List[pd.DataFrame], pd.DataFrame, None]:
         """
-        Runs a SQL or T-SQL query (or multiple queries) against a Fabric Warehouse.
+        Runs a SQL or T-SQL query (or multiple queries) against a Fabric Warehouse/Lakehouse.
 
         Parameters
         ----------
@@ -76,10 +94,10 @@ class ConnectWarehouse:
             A list of pandas DataFrames if multiple SQL queries return results,
             a single DataFrame if one query is executed and returns results, or None.
         """
+
         cursor = None
-        results = []  # To store results from multiple queries if needed
+        results = []
 
-        # If the input is a single string, convert it to a list for consistency
         if isinstance(sql, str):
             sql = [sql]
 
@@ -101,10 +119,7 @@ class ConnectWarehouse:
                     results.append(result)
 
             # Return results if any queries returned a result set
-            if results:
-                return results if len(results) > 1 else results[0]
-            else:
-                return None
+            return results if len(results) > 1 else (results[0] if results else None)
 
         finally:
             if cursor:
@@ -118,3 +133,33 @@ class ConnectWarehouse:
 
     def close(self):
         self.connection.close()
+
+
+class ConnectWarehouse(ConnectBase):
+    def __init__(
+        self,
+        warehouse: str,
+        workspace: Optional[Union[str, uuid.UUID]] = None,
+        timeout: Optional[int] = None,
+    ):
+        super().__init__(
+            name=warehouse,
+            workspace=workspace,
+            timeout=timeout,
+            endpoint_type="warehouse",
+        )
+
+
+class ConnectLakehouse(ConnectBase):
+    def __init__(
+        self,
+        lakehouse: str,
+        workspace: Optional[Union[str, uuid.UUID]] = None,
+        timeout: Optional[int] = None,
+    ):
+        super().__init__(
+            name=lakehouse,
+            workspace=workspace,
+            timeout=timeout,
+            endpoint_type="lakehouse",
+        )

sempy_labs/_vertipaq.py

@@ -551,13 +551,13 @@ def vertipaq_analyzer(
             df.columns = df.columns.str.replace(" ", "_")
 
             schema = {
-                "Capacity_Name": "string",
-                "Capacity_Id": "string",
-                "Workspace_Name": "string",
-                "Workspace_Id": "string",
-                "Dataset_Name": "string",
-                "Dataset_Id": "string",
-                "Configured_By": "string",
+                "Capacity_Name": data_type_string,
+                "Capacity_Id": data_type_string,
+                "Workspace_Name": data_type_string,
+                "Workspace_Id": data_type_string,
+                "Dataset_Name": data_type_string,
+                "Dataset_Id": data_type_string,
+                "Configured_By": data_type_string,
             }
 
             schema.update(
@@ -566,6 +566,8 @@ def vertipaq_analyzer(
                     for key, value in vertipaq_map[key_name].items()
                 }
             )
+            schema["RunId"] = data_type_long
+            schema["Timestamp"] = data_type_timestamp
 
             delta_table_name = f"VertipaqAnalyzer_{obj}".lower()
             save_as_delta_table(

sempy_labs/admin/__init__.py

@@ -0,0 +1,53 @@
+from sempy_labs.admin._basic_functions import (
+    assign_workspaces_to_capacity,
+    list_capacities,
+    list_tenant_settings,
+    list_capacities_delegated_tenant_settings,
+    unassign_workspaces_from_capacity,
+    list_external_data_shares,
+    revoke_external_data_share,
+    list_workspaces,
+    list_datasets,
+    list_item_access_details,
+    list_access_entities,
+    list_workspace_access_details,
+    list_items,
+)
+from sempy_labs.admin._domains import (
+    list_domains,
+    list_domain_workspaces,
+    assign_domain_workspaces,
+    assign_domain_workspaces_by_capacities,
+    create_domain,
+    update_domain,
+    delete_domain,
+    resolve_domain_id,
+    unassign_domain_workspaces,
+    unassign_all_domain_workspaces,
+)
+
+__all__ = [
+    "list_items",
+    "list_workspace_access_details",
+    "list_access_entities",
+    "list_item_access_details",
+    "list_datasets",
+    "list_workspaces",
+    "assign_workspaces_to_capacity",
+    "list_capacities",
+    "list_tenant_settings",
+    "list_domains",
+    "list_domain_workspaces",
+    "assign_domain_workspaces",
+    "assign_domain_workspaces_by_capacities",
+    "create_domain",
+    "update_domain",
+    "delete_domain",
+    "resolve_domain_id",
+    "unassign_domain_workspaces",
+    "unassign_all_domain_workspaces",
+    "list_capacities_delegated_tenant_settings",
+    "unassign_workspaces_from_capacity",
+    "list_external_data_shares",
+    "revoke_external_data_share",
+]