Fix CeleryInteration when sentry is used (#95)

CHANGES.md

@@ -3,7 +3,7 @@
 ## 0.17.2 (unreleased)
 ----------------------
 
-- Nothing changed yet.
+- Fixed context forwarding to celery tasks in case Sentry is used.
 
 
 ## 0.17.1 (2024-10-01)

clean_python/celery/config.py

@@ -1,12 +1,17 @@
+import logging
+
 from celery import Celery
 from celery import current_app
+from celery.signals import worker_process_init
 
 from clean_python import Json
 from clean_python import ValueObject
 from clean_python.celery import BaseTask
 
 __all__ = ["CeleryConfig"]
 
+logger = logging.getLogger(__name__)
+
 
 class CeleryConfig(ValueObject):
     timezone: str = "Europe/Amsterdam"
@@ -29,3 +34,21 @@ def apply(self, strict_typing: bool = True) -> Celery:
         app.strict_typing = strict_typing
         app.config_from_object(self)
         return app
+
+
+@worker_process_init.connect
+def worker_init(**kwargs):
+    # Fix Sentry configuration (if inplace)
+    try:
+        import sentry_sdk
+        from sentry_sdk.integrations.celery import CeleryIntegration
+
+        integration = sentry_sdk.get_client().get_integration(CeleryIntegration)
+        if integration is not None and integration.propagate_traces:
+            integration.propagate_traces = False
+            logger.warning(
+                "Automatically disabled Sentry's trace propagation. "
+                "Set CeleryIntegration(propagate_traces=False) to disable this warning."
+            )
+    except ImportError:
+        pass