azure-connectors 0.1.0.dev1__py3-none-any.whl

azure/connectors/__init__.py

@@ -0,0 +1,39 @@
+# Copyright (c) Microsoft Corporation. All rights reserved.
+
+"""
+Azure Logic Apps Connector SDK for Python.
+
+This package provides infrastructure for calling Azure Logic Apps connectors
+from Python applications, including authentication, HTTP clients, and strongly-typed
+generated connector clients.
+"""
+
+from .sdk.client_base import ConnectorClientBase
+from .sdk.options import ConnectorClientOptions
+from .sdk.authentication import TokenProvider, ManagedIdentityTokenProvider, ConnectionStringTokenProvider, AzureIdentityTokenProvider
+from .sdk.exceptions import ConnectorException
+from .sdk.trigger_payload import TriggerCallbackPayload, TriggerCallbackBody
+from .kusto import KustoClient
+from .office365 import Office365Client
+from .sharepointonline import SharepointonlineClient
+from .teams import TeamsClient
+from .msgraph import MsgraphgroupsanduserClient
+
+__version__ = "0.1.0"
+
+__all__ = [
+    "ConnectorClientBase",
+    "ConnectorClientOptions",
+    "TokenProvider",
+    "ManagedIdentityTokenProvider",
+    "ConnectionStringTokenProvider",
+    "AzureIdentityTokenProvider",
+    "ConnectorException",
+    "TriggerCallbackPayload",
+    "TriggerCallbackBody",
+    "KustoClient",
+    "Office365Client",
+    "SharepointonlineClient",
+    "TeamsClient",
+    "MsgraphgroupsanduserClient",
+]

azure/connectors/kusto.py

@@ -0,0 +1,323 @@
+# Copyright (c) Microsoft Corporation. All rights reserved.
+
+# KustoClient.py - Auto-generated DirectClient SDK
+# Do not edit this file directly.
+
+from dataclasses import dataclass, field
+from typing import Optional, List, Any, Dict
+from datetime import datetime
+from urllib.parse import quote
+
+from azure.connectors.sdk import (
+    ConnectorClientBase,
+    ConnectorClientOptions,
+    TokenProvider,
+    ManagedIdentityTokenProvider,
+    ConnectorException,
+)
+
+
+# Type Definitions
+
+@dataclass
+class Table:
+    """Response for Run KQL query"""
+
+    pass
+
+@dataclass
+class VisualizeResults:
+    """Response for Run KQL query and render a chart"""
+
+    pass
+
+@dataclass
+class AsyncCommandResult:
+    """Response for Run async control command"""
+
+    pass
+
+@dataclass
+class MCPQueryResponse:
+    """Response for Kusto Query MCP Server"""
+
+    pass
+
+@dataclass
+class ObjectEntity:
+    """Definition: Object"""
+
+    pass
+
+@dataclass
+class Row:
+    """Definition: Row"""
+
+    pass
+
+@dataclass
+class QueryAndVisualizeSchema:
+    """Definition: QueryAndVisualizeSchema"""
+
+    cluster: Optional[ClusterName] = None
+    db: Optional[DatabaseName] = None
+    csl: Optional[Query] = None
+    chart_type: Optional[ChartType] = None
+
+@dataclass
+class CommandAndVisualizeSchema:
+    """Definition: CommandAndVisualizeSchema"""
+
+    cluster: Optional[ClusterName] = None
+    db: Optional[DatabaseName] = None
+    csl: Optional[str] = None
+    """Specify the control command you would like to run"""
+    chart_type: Optional[ChartType] = None
+
+@dataclass
+class QueryAndListSchema:
+    """Definition: QueryAndListSchema"""
+
+    cluster: Optional[ClusterName] = None
+    db: Optional[DatabaseName] = None
+    csl: Optional[Query] = None
+
+@dataclass
+class ControlCommandAndListSchema:
+    """Definition: ControlCommandAndListSchema"""
+
+    cluster: Optional[ClusterName] = None
+    db: Optional[DatabaseName] = None
+    csl: Optional[str] = None
+    """Specify the show control command you would like to run"""
+
+@dataclass
+class ClusterName:
+    """Definition: ClusterName"""
+
+    pass
+
+@dataclass
+class DatabaseName:
+    """Definition: DatabaseName"""
+
+    pass
+
+@dataclass
+class Query:
+    """Definition: Query"""
+
+    pass
+
+@dataclass
+class ChartType:
+    """Definition: ChartType"""
+
+    pass
+
+@dataclass
+class MCPQueryRequest:
+    """Definition: MCPQueryRequest"""
+
+    jsonrpc: Optional[str] = None
+    id: Optional[str] = None
+    method: Optional[str] = None
+    params: Optional[Dict[str, Any]] = None
+    result: Optional[Dict[str, Any]] = None
+    error: Optional[Dict[str, Any]] = None
+    callback_endpoint: Optional[str] = None
+
+
+# Client Class
+
+class KustoClient(ConnectorClientBase):
+    """Typed client for kusto connector."""
+
+    def __init__(
+        self,
+        connection_runtime_url: str,
+        token_provider: Optional[TokenProvider] = None,
+        options: Optional[ConnectorClientOptions] = None,
+    ):
+        """
+        Initialize a KustoClient.
+
+        Args:
+            connection_runtime_url: The connection runtime URL from Azure Portal.
+            token_provider: Optional token provider. Defaults to ManagedIdentityTokenProvider.
+            options: Optional connector client options.
+        """
+        if not connection_runtime_url:
+            raise ValueError("connection_runtime_url cannot be None or empty")
+
+        if token_provider is None:
+            token_provider = ManagedIdentityTokenProvider()
+
+        super().__init__(token_provider, options)
+        self._connection_runtime_url = connection_runtime_url.rstrip('/')
+
+    @property
+    def connector_name(self) -> str:
+        return "kusto"
+
+    async def list_kusto_results_async(
+        self,
+        input: QueryAndListSchema,
+    ):
+        """
+        Run KQL query
+
+        Runs the KQL query and returns the result as a set of rows which can be iterated over in the following connectors e.g TableName | take 10.
+        """
+        path = f"{self._connection_runtime_url}/ListKustoResults/false"
+
+        response = await self.http_client.send_async("POST", path, body=input)
+
+        if not (200 <= response.status < 300):
+            raise ConnectorException(
+                f"POST {path}",
+                response.status,
+                response.text,
+            )
+
+        if not response.text:
+            return None
+
+        import json
+        return json.loads(response.text)
+
+    async def list_kusto_show_command_results_async(
+        self,
+        input: ControlCommandAndListSchema,
+    ):
+        """
+        Run show control command
+
+        Runs the show control command and returns the result as a set of rows which can be iterated over in the following connectors e.g .show table TableName policy caching.
+        """
+        path = f"{self._connection_runtime_url}/ListKustoShowCommandResults"
+
+        response = await self.http_client.send_async("POST", path, body=input)
+
+        if not (200 <= response.status < 300):
+            raise ConnectorException(
+                f"POST {path}",
+                response.status,
+                response.text,
+            )
+
+        if not response.text:
+            return None
+
+        import json
+        return json.loads(response.text)
+
+    async def run_kusto_query_and_visualize_results_async(
+        self,
+        input: QueryAndVisualizeSchema,
+    ):
+        """
+        Run KQL query and render a chart
+
+        Runs the KQL query and returns result as a chart of your choice e.g TableName | where Timestamp > ago(1h) | project timestamp, value.
+        """
+        path = f"{self._connection_runtime_url}/RunKustoAndVisualizeResults/false"
+
+        response = await self.http_client.send_async("POST", path, body=input)
+
+        if not (200 <= response.status < 300):
+            raise ConnectorException(
+                f"POST {path}",
+                response.status,
+                response.text,
+            )
+
+        if not response.text:
+            return None
+
+        import json
+        return json.loads(response.text)
+
+    async def run_kusto_command_and_visualize_results_async(
+        self,
+        input: CommandAndVisualizeSchema,
+    ):
+        """
+        Run control command and render a chart
+
+        Runs the control command and returns the result as a chart of your choice e.g .clear table TableName data.
+        """
+        path = f"{self._connection_runtime_url}/RunKustoAndVisualizeResults/true"
+
+        response = await self.http_client.send_async("POST", path, body=input)
+
+        if not (200 <= response.status < 300):
+            raise ConnectorException(
+                f"POST {path}",
+                response.status,
+                response.text,
+            )
+
+        if not response.text:
+            return None
+
+        import json
+        return json.loads(response.text)
+
+    async def run_async_control_command_and_wait_async(
+        self,
+        input: ControlCommandAndListSchema,
+    ):
+        """
+        Run async control command
+
+        Runs control command in async mode and returns its ID, state and status on completion. Command can run for maximum 1 hour. The 'async' keyword is mandatory e.g .set-or-append async TargetTable <| SourceTable.
+        """
+        path = f"{self._connection_runtime_url}/RunAsyncControlCommandAndWait"
+
+        response = await self.http_client.send_async("POST", path, body=input)
+
+        if not (200 <= response.status < 300):
+            raise ConnectorException(
+                f"POST {path}",
+                response.status,
+                response.text,
+            )
+
+        if not response.text:
+            return None
+
+        import json
+        return json.loads(response.text)
+
+    async def mcp_kusto_query_management_async(
+        self,
+        input: MCPQueryRequest,
+        session_id: Optional[str] = None,
+    ):
+        """
+        Kusto Query MCP Server
+
+        This MCP server runs Kusto queries and manages the results.
+        """
+        path = f"{self._connection_runtime_url}/mcp/KustoQueryManagement"
+        query_params = []
+        if session_id is not None:
+            query_params.append(f"sessionId={quote(str(session_id).lower() if isinstance(session_id, bool) else str(session_id))}")
+        if query_params:
+            path += '?' + '&'.join(query_params)
+
+        response = await self.http_client.send_async("POST", path, body=input)
+
+        if not (200 <= response.status < 300):
+            raise ConnectorException(
+                f"POST {path}",
+                response.status,
+                response.text,
+            )
+
+        if not response.text:
+            return None
+
+        import json
+        return json.loads(response.text)