Fix possible AttributeError

Azure · Sep 13, 2019 · 3ced2ce · 3ced2ce
1 parent c3b6db9
commit 3ced2ce
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sdk/eventhub/azure-eventhubs/azure/eventhub/aio/consumer_async.py b/sdk/eventhub/azure-eventhubs/azure/eventhub/aio/consumer_async.py
@@ -186,7 +186,7 @@ async def _receive(self, timeout_time=None, max_batch_size=None, **kwargs):
             # Tracing
             current_span = get_parent_span()
             if current_span and event_data.application_properties:
-                traceparent = event_data.application_properties.get(b"Diagnostic-Id", None).decode('ascii')
+                traceparent = event_data.application_properties.get(b"Diagnostic-Id", "").decode('ascii')
                 if traceparent:
                     current_span.link(traceparent)
 

diff --git a/sdk/eventhub/azure-eventhubs/azure/eventhub/consumer.py b/sdk/eventhub/azure-eventhubs/azure/eventhub/consumer.py
@@ -180,7 +180,7 @@ def _receive(self, timeout_time=None, max_batch_size=None, **kwargs):
             # Tracing
             current_span = get_parent_span()
             if current_span and event_data.application_properties:
-                traceparent = event_data.application_properties.get(b"Diagnostic-Id", None).decode('ascii')
+                traceparent = event_data.application_properties.get(b"Diagnostic-Id", "").decode('ascii')
                 if traceparent:
                     current_span.link(traceparent)