fix: pass before_send through module-level setup

Author: sipa-echo-zaoa

posthog/__init__.py

@@ -58,6 +58,7 @@
 from posthog.types import (
     FeatureFlag,
     FlagsAndPayloads,
+    BeforeSendCallback
 )
 from posthog.types import (
     FeatureFlagResult as FeatureFlagResult,
@@ -250,6 +251,7 @@ def get_tags() -> Dict[str, Any]:
 project_root = None  # type: Optional[str]
 # Used for our AI observability feature to not capture any prompt or output just usage + metadata
 privacy_mode = False  # type: bool
+before_send = None  # type: Optional[BeforeSendCallback]
 # Whether to enable feature flag polling for local evaluation by default. Defaults to True.
 # We recommend setting this to False if you are only using the personalApiKey for evaluating remote config payloads via `get_remote_config_payload` and not using local evaluation.
 enable_local_evaluation = True  # type: bool
@@ -875,6 +877,7 @@ def setup() -> Client:
             # or deprecate this proxy option fully (it's already in the process of deprecation, no new clients should be using this method since like 5-6 months)
             enable_exception_autocapture=enable_exception_autocapture,
             log_captured_exceptions=log_captured_exceptions,
+            before_send=before_send,
             enable_local_evaluation=enable_local_evaluation,
             flag_definition_cache_provider=flag_definition_cache_provider,
             capture_exception_code_variables=capture_exception_code_variables,

posthog/test/test_before_send.py

@@ -1,6 +1,8 @@
+import importlib
 import unittest
 
 import mock
+import posthog
 
 from posthog.client import Client
 from posthog.test.test_utils import FAKE_TEST_API_KEY
@@ -216,3 +218,33 @@ def scrub_pii(event):
             self.assertEqual(enqueued_msg["properties"]["email"], "***@example.com")
             self.assertNotIn("credit_card", enqueued_msg["properties"])
             self.assertEqual(enqueued_msg["properties"]["form_name"], "contact")
+
+
+class TestModuleLevelBeforeSend(unittest.TestCase):
+    def setUp(self):
+        importlib.reload(posthog)
+
+    def tearDown(self):
+        if posthog.default_client:
+            posthog.shutdown()
+        importlib.reload(posthog)
+
+    def test_before_send_callback_used_during_module_level_setup(self):
+        def my_before_send(event):
+            event["properties"]["module_level_before_send"] = True
+            return event
+
+        with mock.patch("posthog.client.batch_post") as mock_post:
+            posthog.api_key = FAKE_TEST_API_KEY
+            posthog.before_send = my_before_send
+            posthog.sync_mode = True
+
+            msg_uuid = posthog.capture("test_event", distinct_id="user1")
+
+            self.assertIsNotNone(msg_uuid)
+            self.assertIs(posthog.default_client.before_send, my_before_send)
+
+            mock_post.assert_called_once()
+            batch_data = mock_post.call_args[1]["batch"]
+            enqueued_msg = batch_data[0]
+            self.assertTrue(enqueued_msg["properties"]["module_level_before_send"])