add tags to all events

Author: oliverb123

example.py

@@ -104,13 +104,18 @@
 print(posthog.get_remote_config_payload("encrypted_payload_flag_key"))
 
 
-# You can add tags to a context, and these are automatically added to exceptions captured within that context.
+# You can add tags to a context, and these are automatically added to exceptions and other events captured
+# within that context.
+
 # You can enter a new context using a with statement. Any exceptions thrown in the context will be captured,
-# and tagged with the context tags. By default, it inherits tags from the parent context.
+# and tagged with the context tags. Other events captured will also be tagged with the context tags. By default,
+# the new context inherits tags from the parent context.
 with posthog.new_context():
-    posthog.tag("user_id", 123)
     posthog.tag("transaction_id", "abc123")
     posthog.tag("some_arbitrary_value", {"tags": "can be dicts"})
+
+    # This event will be captured with the tags set above
+    posthog.capture("order_processed")
     # This exception will be captured with the tags set above
     raise Exception("Order processing failed")
 

posthog/client.py

@@ -467,6 +467,11 @@ def capture(
         require("properties", properties, dict)
         require("event", event, string_types)
 
+        # Grab current context tags, if any exist
+        context_tags = get_tags()
+        if context_tags:
+            properties.update(context_tags)
+
         msg = {
             "properties": properties,
             "timestamp": timestamp,
@@ -668,11 +673,6 @@ def capture_exception(
                 self.log.debug("Exception already captured, skipping")
                 return
 
-            # Grab current context tags, if any exist
-            context_tags = get_tags()
-            if context_tags:
-                properties.update(context_tags)
-
             # if there's no distinct_id, we'll generate one and set personless mode
             # via $process_person_profile = false
             if distinct_id is None:

posthog/scopes.py

@@ -14,25 +14,29 @@ def new_context(fresh=False):
     # TODO - we could extend this context idea to also apply to other event types eventually,
     # but right now it only applies to exceptions...
     """
-    Create a new context scope that will be active for the duration of the with block.
-    Any tags set within this scope will be isolated to this context. Any exceptions raised
-    within the context will be captured and tagged with the context tags.
-
-    Args:
-        fresh: Whether to start with a fresh context (default: False).
-               If False, inherits tags from parent context.
-               If True, starts with no tags.
-
-    Examples:
-        # Inherit parent context tags
-        with posthog.new_context():
-            posthog.tag("user_id", "123")
-            raise ValueError("Something went wrong")
-
-        # Start with fresh context (no inherited tags)
-        with posthog.new_context(fresh=True):
-            posthog.tag("user_id", "123")
-            raise ValueError("Something went wrong")
+     Create a new context scope that will be active for the duration of the with block.
+     Any tags set within this scope will be isolated to this context. Any exceptions raised
+    or events captured within the context will be tagged with the context tags.
+
+     Args:
+         fresh: Whether to start with a fresh context (default: False).
+                If False, inherits tags from parent context.
+                If True, starts with no tags.
+
+     Examples:
+         # Inherit parent context tags
+         with posthog.new_context():
+             posthog.tag("request_id", "123")
+             # Both this event and the exception will be tagged with the context tags
+             posthog.capture("event_name", {"property": "value"})
+             raise ValueError("Something went wrong")
+
+         # Start with fresh context (no inherited tags)
+         with posthog.new_context(fresh=True):
+             posthog.tag("request_id", "123")
+             # Both this event and the exception will be tagged with the context tags
+             posthog.capture("event_name", {"property": "value"})
+             raise ValueError("Something went wrong")
 
     """
     import posthog
@@ -86,9 +90,8 @@ def clear_tags() -> None:
 
 def scoped(fresh=False):
     """
-    Decorator that creates a new context for the function, wraps the function in a
-    try/except block, and if an exception occurs, captures it with the current context
-    tags before re-raising it.
+    Decorator that creates a new context for the function. Simply wraps
+    the function in a with posthog.new_context(): block.
 
     Args:
         fresh: Whether to start with a fresh context (default: False)
@@ -98,8 +101,12 @@ def scoped(fresh=False):
         def process_payment(payment_id):
             posthog.tag("payment_id", payment_id)
             posthog.tag("payment_method", "credit_card")
+
+            # This event will be captured with tags
+            posthog.capture("payment_started")
             # If this raises an exception, it will be captured with tags
             # and then re-raised
+            some_risky_function()
     """
 
     def decorator(func: F) -> F: