inherit tags by default

Author: oliverb123

example.py

@@ -106,7 +106,7 @@
 
 # You can add tags to a context, and these are automatically added to exceptions 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
+# and tagged with the context tags. By default, it inherits tags from the parent context.
 with posthog.new_context():
     posthog.tag("user_id", 123)
     posthog.tag("transaction_id", "abc123")
@@ -115,12 +115,28 @@
     raise Exception("Order processing failed")
 
 
-# You can also use the `@posthog.scoped` decorator to enter a new context.
-@posthog.scoped
+# Use fresh=True to start with a clean context (no inherited tags)
+with posthog.new_context(fresh=True):
+    posthog.tag("session_id", "xyz789")
+    # Only session_id tag will be present, no inherited tags
+    raise Exception("Session handling failed")
+
+
+# You can also use the `@posthog.scoped()` decorator to enter a new context.
+# By default, it inherits tags from the parent context
+@posthog.scoped()
 def process_order(order_id):
     posthog.tag("order_id", order_id)
     # Exception will be captured and tagged automatically
     raise Exception("Order processing failed")
 
 
+# Use fresh=True to start with a clean context (no inherited tags)
+@posthog.scoped(fresh=True)
+def process_payment(payment_id):
+    posthog.tag("payment_id", payment_id)
+    # Only payment_id tag will be present, no inherited tags
+    raise Exception("Payment processing failed")
+
+
 posthog.shutdown()

posthog/scopes.py

@@ -10,25 +10,36 @@ def _get_current_context() -> Dict[str, Any]:
 
 
 @contextmanager
-def new_context():
+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.
 
-    Example:
+    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")
-            # The exception will be captured and tagged with the context tags
+            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")
 
     """
     import posthog
 
+    current_tags = _get_current_context().copy()
     current_stack = _context_stack.get()
-    new_stack = current_stack + [{}]
+    new_stack = current_stack + [{}] if fresh else current_stack + [current_tags]
     token = _context_stack.set(new_stack)
 
     try:
@@ -73,28 +84,32 @@ def clear_tags() -> None:
 F = TypeVar("F", bound=Callable[..., Any])
 
 
-def scoped(func: F) -> F:
+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.
 
     Args:
-        func: The function to wrap
+        fresh: Whether to start with a fresh context (default: False)
 
     Example:
-        @posthog.scoped
+        @posthog.scoped()
         def process_payment(payment_id):
             posthog.tag("payment_id", payment_id)
             posthog.tag("payment_method", "credit_card")
             # If this raises an exception, it will be captured with tags
             # and then re-raised
     """
-    from functools import wraps
 
-    @wraps(func)
-    def wrapper(*args, **kwargs):
-        with new_context():
-            return func(*args, **kwargs)
+    def decorator(func: F) -> F:
+        from functools import wraps
+
+        @wraps(func)
+        def wrapper(*args, **kwargs):
+            with new_context(fresh=fresh):
+                return func(*args, **kwargs)
+
+        return cast(F, wrapper)
 
-    return cast(F, wrapper)
+    return decorator

posthog/test/test_scopes.py

@@ -28,7 +28,7 @@ def test_new_context_isolation(self):
         # Set tag in outer context
         tag("outer", "value")
 
-        with new_context():
+        with new_context(fresh=True):
             # Inner context should start empty
             self.assertEqual(get_tags(), {})
 
@@ -39,6 +39,10 @@ def test_new_context_isolation(self):
             # Outer tag should not be visible
             self.assertNotIn("outer", get_tags())
 
+        with new_context(fresh=False):
+            # Inner context should start empty
+            self.assertEqual(get_tags(), {"outer": "value"})
+
         # After exiting context, inner tag should be gone
         self.assertNotIn("inner", get_tags())
 
@@ -48,10 +52,10 @@ def test_new_context_isolation(self):
     def test_nested_contexts(self):
         tag("level1", "value1")
 
-        with new_context():
+        with new_context(fresh=True):
             tag("level2", "value2")
 
-            with new_context():
+            with new_context(fresh=True):
                 tag("level3", "value3")
                 self.assertEqual(get_tags(), {"level3": "value3"})
 
@@ -63,7 +67,7 @@ def test_nested_contexts(self):
 
     @patch("posthog.capture_exception")
     def test_scoped_decorator_success(self, mock_capture):
-        @scoped
+        @scoped()
         def successful_function(x, y):
             tag("x", x)
             tag("y", y)
@@ -91,7 +95,7 @@ def check_context_on_capture(exception, **kwargs):
 
         mock_capture.side_effect = check_context_on_capture
 
-        @scoped
+        @scoped()
         def failing_function():
             tag("important_context", "value")
             raise test_exception