Api for setting arbitrary contexts in spans.

Author: antonpirker

sentry_sdk/tracing.py

@@ -93,6 +93,7 @@ class Span(object):
         "timestamp",
         "_tags",
         "_data",
+        "_contexts",
         "_span_recorder",
         "hub",
         "_context_manager_state",
@@ -140,6 +141,7 @@ def __init__(
         self.hub = hub
         self._tags = {}  # type: Dict[str, str]
         self._data = {}  # type: Dict[str, Any]
+        self._contexts = {}  # type: Dict[str, Any]
         self._containing_transaction = containing_transaction
         self.start_timestamp = datetime.utcnow()
         try:
@@ -468,6 +470,10 @@ def set_http_status(self, http_status):
         else:
             self.set_status("unknown_error")
 
+    def set_context(self, key, value):
+        # type: (str, Any) -> None
+        self._contexts[key] = value
+
     def is_success(self):
         # type: () -> bool
         return self.status == "ok"
@@ -685,11 +691,15 @@ def finish(self, hub=None):
         # to be garbage collected
         self._span_recorder = None
 
+        contexts = {}
+        contexts.update(self._contexts)
+        contexts.update({"trace": self.get_trace_context()})
+
         event = {
             "type": "transaction",
             "transaction": self.name,
             "transaction_info": {"source": self.source},
-            "contexts": {"trace": self.get_trace_context()},
+            "contexts": contexts,
             "tags": self._tags,
             "timestamp": self.timestamp,
             "start_timestamp": self.start_timestamp,