azure-connectors 0.1.0__tar.gz

azure_connectors-0.1.0/PKG-INFO

@@ -0,0 +1,41 @@
+Metadata-Version: 2.4
+Name: azure-connectors
+Version: 0.1.0
+Summary: Azure Logic Apps Connector SDK for Python
+Author-email: "Azure Logic Apps team at Microsoft Corp." <azurelogicapps@microsoft.com>
+License: MIT
+Project-URL: Documentation, https://github.com/Azure/connectors-python-sdk
+Project-URL: Repository, https://github.com/Azure/connectors-python-sdk
+Project-URL: Changelog, https://github.com/Azure/connectors-python-sdk/blob/main/CHANGELOG.md
+Keywords: azure,connectors,logic-apps,python,serverless
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Operating System :: Microsoft :: Windows
+Classifier: Operating System :: POSIX
+Classifier: Operating System :: MacOS :: MacOS X
+Classifier: Environment :: Web Environment
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Intended Audience :: Developers
+Classifier: Typing :: Typed
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: aiohttp
+Requires-Dist: azure-identity[aio]
+Provides-Extra: dev
+Requires-Dist: flake8<8,>=6; extra == "dev"
+Requires-Dist: mypy; extra == "dev"
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: pytest-asyncio; extra == "dev"
+Requires-Dist: httpx; extra == "dev"
+Requires-Dist: requests==2.*; extra == "dev"
+Requires-Dist: coverage; extra == "dev"
+Requires-Dist: pytest-sugar; extra == "dev"
+Requires-Dist: pytest-cov; extra == "dev"
+Requires-Dist: pytest-xdist; extra == "dev"
+Requires-Dist: pytest-randomly; extra == "dev"
+Requires-Dist: pytest-instafail; extra == "dev"
+Requires-Dist: pytest-rerunfailures; extra == "dev"

azure_connectors-0.1.0/azure/connectors/__init__.py

@@ -0,0 +1,64 @@
+# 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
+
+# Generated connector clients — some may have import errors from incomplete
+# code generation (e.g., undefined type references). Wrap each import so
+# a single broken connector doesn't prevent the SDK core from loading.
+# See: https://github.com/Azure/connectors-python-sdk/issues/13
+try:
+    from .kusto import KustoClient
+except (ImportError, NameError):
+    KustoClient = None  # type: ignore[assignment,misc]
+try:
+    from .office365 import Office365Client
+except (ImportError, NameError):
+    Office365Client = None  # type: ignore[assignment,misc]
+try:
+    from .sharepointonline import SharepointonlineClient
+except (ImportError, NameError):
+    SharepointonlineClient = None  # type: ignore[assignment,misc]
+try:
+    from .teams import TeamsClient
+except (ImportError, NameError):
+    TeamsClient = None  # type: ignore[assignment,misc]
+try:
+    from .msgraph import MsgraphgroupsanduserClient
+except (ImportError, NameError):
+    MsgraphgroupsanduserClient = None  # type: ignore[assignment,misc]
+
+__version__ = "0.1.0"
+
+__all__ = [
+    "ConnectorClientBase",
+    "ConnectorClientOptions",
+    "TokenProvider",
+    "ManagedIdentityTokenProvider",
+    "ConnectionStringTokenProvider",
+    "AzureIdentityTokenProvider",
+    "ConnectorException",
+    "TriggerCallbackPayload",
+    "TriggerCallbackBody",
+    "KustoClient",
+    "Office365Client",
+    "SharepointonlineClient",
+    "TeamsClient",
+    "MsgraphgroupsanduserClient",
+]

azure_connectors-0.1.0/azure/connectors/kusto.py

@@ -0,0 +1,325 @@
+# Copyright (c) Microsoft Corporation. All rights reserved.
+
+# KustoClient.py - Auto-generated DirectClient SDK
+# Do not edit this file directly.
+
+from __future__ import annotations
+
+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)