set context if None

Author: qeternity

elasticapm/instrumentation/packages/asyncio/aioredis.py

@@ -56,6 +56,8 @@ def call(self, module, method, wrapped, instance, args, kwargs):
  ) as span:
  if span.context is not None:
  span.context["destination"] = _get_destination_info(instance)
+ else:
+ span.context = {"destination": _get_destination_info(instance)}
 
  return wrapped(*args, **kwargs)
 
@@ -73,6 +75,8 @@ def call(self, module, method, wrapped, instance, args, kwargs):
  ) as span:
  if span.context is not None:
  span.context["destination"] = _get_destination_info(instance)
+ else:
+ span.context = {"destination": _get_destination_info(instance)}
 
  return wrapped(*args, **kwargs)