Event grid track 2

.gitignore

@@ -96,4 +96,9 @@ sdk/storage/azure-storage-file-datalake/tests/settings_real.py
 sdk/cosmos/azure-cosmos/test/test_config.py
 
 # temp path to to run regression test
-.tmp_code_path/
+.tmp_code_path/
+
+# api view
+.history
+.env
+dev.env

sdk/eventgrid/azure-eventgrid/azure/eventgrid/__init__.py

@@ -1,18 +1,17 @@
 # coding=utf-8
 # --------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
-# Licensed under the MIT License. See License.txt in the project root for
-# license information.
-#
+# Licensed under the MIT License. See License.txt in the project root for license information.
 # Code generated by Microsoft (R) AutoRest Code Generator.
-# Changes may cause incorrect behavior and will be lost if the code is
-# regenerated.
+# Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
-from .event_grid_client import EventGridClient
+from ._publisher_client import EventGridPublisherClient
+from ._consumer import EventGridConsumer
+from ._helpers import generate_shared_access_signature
+from ._models import CloudEvent, CustomEvent, EventGridEvent, DeserializedEvent
+from .shared_access_signature_credential import EventGridSharedAccessSignatureCredential
 from .version import VERSION
 
-__all__ = ['EventGridClient']
-
-__version__ = VERSION
-
+__all__ = ['EventGridPublisherClient', 'EventGridConsumer', 'CloudEvent', 'CustomEvent', 'DeserializedEvent', 'EventGridEvent', 'generate_shared_access_signature', 'EventGridSharedAccessSignatureCredential']
+__version__ = VERSION

sdk/eventgrid/azure-eventgrid/azure/eventgrid/_constants.py

@@ -0,0 +1,9 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------------------------
+
+EVENTGRID_KEY_HEADER = 'aeg-sas-key'
+EVENTGRID_TOKEN_HEADER = 'aeg-sas-token'
+DEFAULT_API_VERSION = "2018-01-01"
+SAFE_ENCODE = '~()*!.\''

sdk/eventgrid/azure-eventgrid/azure/eventgrid/_consumer.py

@@ -0,0 +1,48 @@
+# coding=utf-8
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is regenerated.
+# --------------------------------------------------------------------------
+
+from typing import TYPE_CHECKING
+
+from azure.core import PipelineClient
+from azure.servicebus import ReceivedMessage
+from msrest import Deserializer, Serializer
+import json
+
+if TYPE_CHECKING:
+    # pylint: disable=unused-import,ungrouped-imports
+    from typing import Any
+
+from ._models import DeserializedEvent
+from ._helpers import generate_shared_access_signature
+from . import _constants as constants
+
+class EventGridConsumer:
+    """
+    A consumer responsible for deserializing event handler messages into a list of event type objects
+    specified in the EventGridEvents/CloudEvents.
+    """
+
+    def __init__(self, **kwargs):
+        # type: (Any) -> None
+        pass
+
+    def deserialize_events(self, events, **kwargs):
+        # type: (azure.eventhub.EventData, azure.functions.EventGridEvent, azure.servicebus.message.ReceivedMessage, azure.functions.HttpRequest, azure.storage.queue.QueueMessage) -> List[models.DeserializedEvent]
+        """A message of a list of events in CloudEvent/EventGridEvent format from an event handler will be parsed and returned as a list of
+        EventContainer objects.
+        :param events: The event handler message to be deserialized.
+        :type events: azure.eventhub.EventData or azure.functions.EventGridEvent or  azure.servicebus.Message or azure.functions.HttpRequest or azure.storage.queue.QueueMessage
+        :rtype: List[models.DeserializedEvent]
+
+        :raise: :class:`ValueError`, when events are not of CloudEvent or EventGridEvent format.
+        """
+        if isinstance(events, ReceivedMessage):
+            dict_event = json.loads(str(events))
+            return [DeserializedEvent(dict_event)]
+
+        return None

sdk/eventgrid/azure-eventgrid/azure/eventgrid/_generated/__init__.py

@@ -0,0 +1,5 @@
+# coding=utf-8
+# ------------------------------------
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+# ------------------------------------

sdk/eventgrid/azure-eventgrid/azure/eventgrid/_generated/event_grid_publisher_client/__init__.py

@@ -0,0 +1,16 @@
+# coding=utf-8
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is regenerated.
+# --------------------------------------------------------------------------
+
+from ._event_grid_publisher_client import EventGridPublisherClient
+__all__ = ['EventGridPublisherClient']
+
+try:
+    from ._patch import patch_sdk
+    patch_sdk()
+except ImportError:
+    pass

sdk/eventgrid/azure-eventgrid/azure/eventgrid/_generated/event_grid_publisher_client/_configuration.py

@@ -0,0 +1,51 @@
+# coding=utf-8
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is regenerated.
+# --------------------------------------------------------------------------
+
+from typing import TYPE_CHECKING
+
+from azure.core.configuration import Configuration
+from azure.core.pipeline import policies
+
+if TYPE_CHECKING:
+    # pylint: disable=unused-import,ungrouped-imports
+    from typing import Any
+
+VERSION = "unknown"
+
+class EventGridPublisherClientConfiguration(Configuration):
+    """Configuration for EventGridPublisherClient.
+
+    Note that all parameters used to create this instance are saved as instance
+    attributes.
+
+    """
+
+    def __init__(
+        self,
+        **kwargs  # type: Any
+    ):
+        # type: (...) -> None
+        super(EventGridPublisherClientConfiguration, self).__init__(**kwargs)
+
+        self.api_version = "2018-01-01"
+        kwargs.setdefault('sdk_moniker', 'eventgridpublisherclient/{}'.format(VERSION))
+        self._configure(**kwargs)
+
+    def _configure(
+        self,
+        **kwargs  # type: Any
+    ):
+        # type: (...) -> None
+        self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs)
+        self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs)
+        self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs)
+        self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs)
+        self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs)
+        self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs)
+        self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs)
+        self.authentication_policy = kwargs.get('authentication_policy')

sdk/eventgrid/azure-eventgrid/azure/eventgrid/_generated/event_grid_publisher_client/_event_grid_publisher_client.py

@@ -0,0 +1,54 @@
+# coding=utf-8
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is regenerated.
+# --------------------------------------------------------------------------
+
+from typing import TYPE_CHECKING
+
+from azure.core import PipelineClient
+from msrest import Deserializer, Serializer
+
+if TYPE_CHECKING:
+    # pylint: disable=unused-import,ungrouped-imports
+    from typing import Any
+
+from ._configuration import EventGridPublisherClientConfiguration
+from .operations import EventGridPublisherClientOperationsMixin
+from . import models
+
+
+class EventGridPublisherClient(EventGridPublisherClientOperationsMixin):
+    """EventGrid Python Publisher Client.
+
+    :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present.
+    """
+
+    def __init__(
+        self,
+        **kwargs  # type: Any
+    ):
+        # type: (...) -> None
+        base_url = 'https://{topicHostname}'
+        self._config = EventGridPublisherClientConfiguration(**kwargs)
+        self._client = PipelineClient(base_url=base_url, config=self._config, **kwargs)
+
+        client_models = {k: v for k, v in models.__dict__.items() if isinstance(v, type)}
+        self._serialize = Serializer(client_models)
+        self._deserialize = Deserializer(client_models)
+
+
+    def close(self):
+        # type: () -> None
+        self._client.close()
+
+    def __enter__(self):
+        # type: () -> EventGridPublisherClient
+        self._client.__enter__()
+        return self
+
+    def __exit__(self, *exc_details):
+        # type: (Any) -> None
+        self._client.__exit__(*exc_details)

sdk/eventgrid/azure-eventgrid/azure/eventgrid/_generated/event_grid_publisher_client/aio/__init__.py

@@ -0,0 +1,10 @@
+# coding=utf-8
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is regenerated.
+# --------------------------------------------------------------------------
+
+from ._event_grid_publisher_client_async import EventGridPublisherClient
+__all__ = ['EventGridPublisherClient']

sdk/eventgrid/azure-eventgrid/azure/eventgrid/_generated/event_grid_publisher_client/aio/_configuration_async.py

@@ -0,0 +1,45 @@
+# coding=utf-8
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is regenerated.
+# --------------------------------------------------------------------------
+
+from typing import Any
+
+from azure.core.configuration import Configuration
+from azure.core.pipeline import policies
+
+VERSION = "unknown"
+
+class EventGridPublisherClientConfiguration(Configuration):
+    """Configuration for EventGridPublisherClient.
+
+    Note that all parameters used to create this instance are saved as instance
+    attributes.
+
+    """
+
+    def __init__(
+        self,
+        **kwargs: Any
+    ) -> None:
+        super(EventGridPublisherClientConfiguration, self).__init__(**kwargs)
+
+        self.api_version = "2018-01-01"
+        kwargs.setdefault('sdk_moniker', 'eventgridpublisherclient/{}'.format(VERSION))
+        self._configure(**kwargs)
+
+    def _configure(
+        self,
+        **kwargs: Any
+    ) -> None:
+        self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs)
+        self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs)
+        self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs)
+        self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs)
+        self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs)
+        self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs)
+        self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs)
+        self.authentication_policy = kwargs.get('authentication_policy')

sdk/eventgrid/azure-eventgrid/azure/eventgrid/_generated/event_grid_publisher_client/aio/_event_grid_publisher_client_async.py

@@ -0,0 +1,46 @@
+# coding=utf-8
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is regenerated.
+# --------------------------------------------------------------------------
+
+from typing import Any
+
+from azure.core import AsyncPipelineClient
+from msrest import Deserializer, Serializer
+
+from ._configuration_async import EventGridPublisherClientConfiguration
+from .operations_async import EventGridPublisherClientOperationsMixin
+from .. import models
+
+
+class EventGridPublisherClient(EventGridPublisherClientOperationsMixin):
+    """EventGrid Python Publisher Client.
+
+    :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present.
+    """
+
+    def __init__(
+        self,
+        **kwargs: Any
+    ) -> None:
+        base_url = 'https://{topicHostname}'
+        self._config = EventGridPublisherClientConfiguration(**kwargs)
+        self._client = AsyncPipelineClient(base_url=base_url, config=self._config, **kwargs)
+
+        client_models = {k: v for k, v in models.__dict__.items() if isinstance(v, type)}
+        self._serialize = Serializer(client_models)
+        self._deserialize = Deserializer(client_models)
+
+
+    async def close(self) -> None:
+        await self._client.close()
+
+    async def __aenter__(self) -> "EventGridPublisherClient":
+        await self._client.__aenter__()
+        return self
+
+    async def __aexit__(self, *exc_details) -> None:
+        await self._client.__aexit__(*exc_details)

sdk/eventgrid/azure-eventgrid/azure/eventgrid/_generated/event_grid_publisher_client/aio/operations_async/__init__.py

@@ -0,0 +1,13 @@
+# coding=utf-8
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is regenerated.
+# --------------------------------------------------------------------------
+
+from ._event_grid_publisher_client_operations_async import EventGridPublisherClientOperationsMixin
+
+__all__ = [
+    'EventGridPublisherClientOperationsMixin',
+]