semantic-link-labs 0.12.8__py3-none-any.whl

sempy_labs/_user_delegation_key.py

@@ -0,0 +1,44 @@
+from sempy_labs.lakehouse._blobs import _request_blob_api
+from sempy_labs._helper_functions import (
+    _xml_to_dict,
+)
+from datetime import datetime, timedelta, timezone
+import xml.etree.ElementTree as ET
+from sempy._utils._log import log
+
+
+@log
+def get_user_delegation_key():
+    """
+    Gets a key that can be used to sign a user delegation SAS (shared access signature). A user delegation SAS grants access to Azure Blob Storage resources by using Microsoft Entra credentials.
+
+    This is a wrapper function for the following API: `Get User Delegation Key <https://learn.microsoft.com/rest/api/storageservices/get-user-delegation-key>`_.
+
+    Returns
+    -------
+    str
+        The user delegation key value.
+    """
+
+    utc_now = datetime.now(timezone.utc)
+    start_time = utc_now + timedelta(minutes=2)
+    expiry_time = start_time + timedelta(minutes=60)
+    start_str = start_time.strftime("%Y-%m-%dT%H:%M:%SZ")
+    expiry_str = expiry_time.strftime("%Y-%m-%dT%H:%M:%SZ")
+
+    payload = f"""<?xml version="1.0" encoding="utf-8"?>
+    <KeyInfo>
+        <Start>{start_str}</Start>
+        <Expiry>{expiry_str}</Expiry>
+    </KeyInfo>"""
+
+    response = _request_blob_api(
+        request="?restype=service&comp=userdelegationkey",
+        method="post",
+        payload=payload,
+    )
+
+    root = ET.fromstring(response.content)
+    response_json = _xml_to_dict(root)
+
+    return response_json.get("UserDelegationKey", {}).get("Value", None)

sempy_labs/_utils.py

@@ -0,0 +1,79 @@
+item_types = {
+    "Dashboard": ["Dashboard", "dashboards"],
+    "DataPipeline": ["Data Pipeline", "dataPipelines", "pipeline-content.json"],
+    "Datamart": ["Datamart", "datamarts"],
+    "Environment": ["Environment", "environments"],
+    "Eventhouse": ["Eventhouse", "eventhouses", "EventhouseProperties.json"],
+    "Eventstream": ["Eventstream", "eventstreams", "eventstream.json"],
+    "GraphQLApi": ["GraphQL Api", "GraphQLApis"],
+    "KQLDashboard": ["KQL Dashboard", "kqlDashboards", "RealTimeDashboard.json"],
+    "KQLDatabase": [
+        "KQL Database",
+        "kqlDatabases",
+    ],  # "DatabaseProperties.json", "DatabaseSchema.kql"
+    "KQLQueryset": ["KQL Queryset", "kqlQuerysets", "RealTimeQueryset.json"],
+    "Lakehouse": ["Lakehouse", "lakehouses"],
+    "MLExperiment": ["ML Experiment", "mlExperiments"],
+    "MLModel": ["ML Model", "mlModels"],
+    "MirroredDatabase": [
+        "Mirrored Database",
+        "mirroredDatabases",
+        "mirroredDatabase.json",
+    ],
+    "MirroredWarehouse": ["Mirrored Warehouse", "mirroredWarehouses"],
+    "MountedDataFactory": [
+        "Mounted Data Factory",
+        "mountedDataFactories",
+        "mountedDataFactory-content.json",
+    ],
+    "Notebook": ["Notebook", "notebooks"],
+    "PaginatedReport": ["Paginated Report", "paginatedReports"],
+    "Reflex": ["Reflex", "reflexes", "ReflexEntities.json"],
+    "Report": ["Report", "reports", "report.json"],
+    "SQLDatabase": ["SQL Database", "sqlDatabases"],
+    "SQLEndpoint": ["SQL Endpoint", "sqlEndpoints"],
+    "SemanticModel": ["Semantic Model", "semanticModels", "model.bim"],
+    "SparkJobDefinition": [
+        "Spark Job Definition",
+        "sparkJobDefinitions",
+        "SparkJobDefinitionV1.json",
+    ],
+    "Warehouse": ["Warehouse", "warehouses"],
+}
+
+
+items = {
+    "CopyJob": "copyJobs",
+    "Dataflow": "dataflows",
+    "Eventhouse": "eventhouses",
+    "GraphQLApi": "GraphQLApis",
+    "Report": "reports",
+    "SemanticModel": "semanticModels",
+    # "Environment": "environments",
+    "KQLDatabase": "kqlDatabases",
+    "KQLDashboard": "kqlDashboards",
+    "KQLQueryset": "kqlQuerysets",
+    "DataPipeline": "dataPipelines",
+    "Notebook": "notebooks",
+    "SparkJobDefinition": "sparkJobDefinitions",
+    "Eventstream": "eventstreams",
+    "MirroredWarehouse": "mirroredWarehouses",
+    "MirroredDatabase": "mirroredDatabases",
+    "MountedDataFactory": "mountedDataFactories",
+    "VariableLibrary": "variableLibraries",
+    "ApacheAirFlowJob": "ApacheAirflowJobs",
+    "WarehouseSnapshot": "warehousesnapshots",
+    "DigitalTwinBuilder": "digitaltwinbuilders",
+    "DigitalTwinBuilderFlow": "DigitalTwinBuilderFlows",
+    "MirroredAzureDatabricksCatalog": "mirroredAzureDatabricksCatalogs",
+    "Map": "Maps",
+    "AnomalyDetector": "anomalydetectors",
+    "UserDataFunction": "UserDataFunctions",
+    "SQLDatabase": "sqlDatabases",
+    "Reflex": "reflexes",
+    "GraphModel": "GraphModels",
+    "GraphQuerySet": "GraphQuerySets",
+    "OperationsAgent": "OperationsAgents",
+    "SnowflakeDatabase": "snowflakeDatabases",
+    "CosmosDBDatabase": "cosmosDbDatabases",
+}