[EventHubs] geodr preview (#35644)

* update changelog readme setup.py

* pass georep prop to desired capabilities

Author: swathipil

sdk/eventhub/azure-eventhub/CHANGELOG.md

@@ -1,5 +1,11 @@
 # Release History
 
+## 5.13.0b1 (2024-05-16)
+
+### Features Added
+
+- Added support for georeplication and disaster recovery-enabled Event Hubs.
+
 ## 5.12.0 (2024-05-16)
 
 ### Features Added

sdk/eventhub/azure-eventhub/README.md

@@ -38,7 +38,7 @@ There, you can also find detailed instructions for using the Azure CLI, Azure Po
 Install the Azure Event Hubs client library for Python with pip:
 
 ```
-$ pip install azure-eventhub
+$ pip install azure-eventhub==5.13.0b1
 ```
 
 ### Authenticate the client

sdk/eventhub/azure-eventhub/azure/eventhub/_constants.py

@@ -33,6 +33,7 @@
 EPOCH_SYMBOL = b"com.microsoft:epoch"
 TIMEOUT_SYMBOL = b"com.microsoft:timeout"
 RECEIVER_RUNTIME_METRIC_SYMBOL = b"com.microsoft:enable-receiver-runtime-metric"
+GEOREPLICATION_SYMBOL = b"com.microsoft:georeplication"
 
 MAX_MESSAGE_LENGTH_BYTES = 1024 * 1024
 MAX_USER_AGENT_LENGTH = 512

sdk/eventhub/azure-eventhub/azure/eventhub/_consumer.py

@@ -17,6 +17,7 @@
     EPOCH_SYMBOL,
     TIMEOUT_SYMBOL,
     RECEIVER_RUNTIME_METRIC_SYMBOL,
+    GEOREPLICATION_SYMBOL,
 )
 
 if TYPE_CHECKING:
@@ -148,9 +149,9 @@ def _create_handler(self, auth: Union[uamqp_JWTTokenAuth, JWTTokenAuth]) -> None
             event_position_selector(self._offset, self._offset_inclusive),
         )
         desired_capabilities = (
-            [RECEIVER_RUNTIME_METRIC_SYMBOL]
+            [RECEIVER_RUNTIME_METRIC_SYMBOL, GEOREPLICATION_SYMBOL]
             if self._track_last_enqueued_event_properties
-            else None
+            else [GEOREPLICATION_SYMBOL]
         )
 
         self._handler = self._amqp_transport.create_receive_client(

sdk/eventhub/azure-eventhub/azure/eventhub/_producer.py

@@ -29,7 +29,7 @@
     is_tracing_enabled,
     TraceAttributes
 )
-from ._constants import TIMEOUT_SYMBOL
+from ._constants import TIMEOUT_SYMBOL, GEOREPLICATION_SYMBOL
 from .amqp import AmqpAnnotatedMessage
 
 _LOGGER = logging.getLogger(__name__)
@@ -136,6 +136,8 @@ def __init__(
     def _create_handler(
         self, auth: Union[uamqp_JWTTokenAuth, JWTTokenAuth]
     ) -> None:
+
+        desired_capabilities = [GEOREPLICATION_SYMBOL]
         self._handler = self._amqp_transport.create_send_client(
             config=self._client._config,  # pylint:disable=protected-access
             target=self._target,
@@ -146,6 +148,7 @@ def _create_handler(
             keep_alive_interval=self._keep_alive,
             client_name=self._name,
             link_properties=self._link_properties, # type: ignore
+            desired_capabilities=desired_capabilities,
             properties=create_properties(
                 self._client._config.user_agent,  # pylint: disable=protected-access
                 amqp_transport=self._amqp_transport,

sdk/eventhub/azure-eventhub/azure/eventhub/_version.py

@@ -3,4 +3,4 @@
 # Licensed under the MIT License.
 # ------------------------------------
 
-VERSION = "5.11.8"
+VERSION = "5.13.0b1"

sdk/eventhub/azure-eventhub/azure/eventhub/aio/_consumer_async.py

@@ -14,7 +14,12 @@
 from ._async_utils import get_dict_with_loop_if_needed
 from .._common import EventData
 from .._utils import create_properties, event_position_selector
-from .._constants import EPOCH_SYMBOL, TIMEOUT_SYMBOL, RECEIVER_RUNTIME_METRIC_SYMBOL
+from .._constants import (
+    EPOCH_SYMBOL,
+    TIMEOUT_SYMBOL,
+    RECEIVER_RUNTIME_METRIC_SYMBOL,
+    GEOREPLICATION_SYMBOL,
+)
 
 if TYPE_CHECKING:
     try:
@@ -145,9 +150,9 @@ def _create_handler(
             event_position_selector(self._offset, self._offset_inclusive),
         )
         desired_capabilities = (
-            [RECEIVER_RUNTIME_METRIC_SYMBOL]
+            [RECEIVER_RUNTIME_METRIC_SYMBOL, GEOREPLICATION_SYMBOL]
             if self._track_last_enqueued_event_properties
-            else None
+            else [GEOREPLICATION_SYMBOL]
         )
 
         self._handler = self._amqp_transport.create_receive_client(

sdk/eventhub/azure-eventhub/azure/eventhub/aio/_producer_async.py

@@ -9,6 +9,7 @@
 from typing import Iterable, Union, Optional, Any, AnyStr, List, TYPE_CHECKING, cast
 
 from .._common import EventData, EventDataBatch
+from .._constants import GEOREPLICATION_SYMBOL
 from .._producer import _set_partition_key
 from .._utils import (
     create_properties,
@@ -119,6 +120,7 @@ def __init__(self, client: EventHubProducerClient, target: str, **kwargs: Any) -
     def _create_handler(
         self, auth: Union["uamqp_JWTTokenAsync", JWTTokenAuthAsync]
     ) -> None:
+        desired_capabilities = [GEOREPLICATION_SYMBOL]
         self._handler = self._amqp_transport.create_send_client(
             config=self._client._config,  # pylint:disable=protected-access
             target=self._target,
@@ -129,6 +131,7 @@ def _create_handler(
             keep_alive_interval=self._keep_alive,
             client_name=self._name,
             link_properties=self._link_properties,
+            desired_capabilities=desired_capabilities,
             properties=create_properties(
                 self._client._config.user_agent,  # pylint: disable=protected-access
                 amqp_transport=self._amqp_transport,

sdk/eventhub/azure-eventhub/samples/README.md

@@ -101,7 +101,7 @@ There, you can also find detailed instructions for using the Azure CLI, Azure Po
 
 1. Install the Azure Event Hubs client library for Python with [pip](https://pypi.org/project/pip/):
 ```bash
-pip install azure-eventhub
+pip install azure-eventhub==5.13.0b1
 ```
 
 To run samples that utilize the Azure Active Directory for authentication, please install the `azure-identity` library:

sdk/eventhub/azure-eventhub/setup.py

@@ -56,7 +56,7 @@
     url='https://github.com/Azure/azure-sdk-for-python/tree/main/sdk/eventhub/azure-eventhub',
     keywords="azure, azure sdk",
     classifiers=[
-        "Development Status :: 5 - Production/Stable",
+        "Development Status :: 4 - Beta",
         'Programming Language :: Python',
         'Programming Language :: Python :: 3 :: Only',
         'Programming Language :: Python :: 3',