feat: Add queue.submit span when a Django task is enqueued

Author: sentrivana

sentry_sdk/consts.py

@@ -936,6 +936,7 @@ class OP:
     QUEUE_SUBMIT_RAY = "queue.submit.ray"
     QUEUE_TASK_RAY = "queue.task.ray"
     QUEUE_TASK_DRAMATIQ = "queue.task.dramatiq"
+    QUEUE_SUBMIT_DJANGO = "queue.submit.django"
     SUBPROCESS = "subprocess"
     SUBPROCESS_WAIT = "subprocess.wait"
     SUBPROCESS_COMMUNICATE = "subprocess.communicate"

sentry_sdk/integrations/django/__init__.py

@@ -62,6 +62,7 @@
 )
 from sentry_sdk.integrations.django.middleware import patch_django_middlewares
 from sentry_sdk.integrations.django.signals_handlers import patch_signals
+from sentry_sdk.integrations.django.tasks import patch_tasks
 from sentry_sdk.integrations.django.views import patch_views
 
 if DJANGO_VERSION[:2] > (1, 8):
@@ -271,6 +272,7 @@ def _django_queryset_repr(value, hint):
         patch_views()
         patch_templates()
         patch_signals()
+        patch_tasks()
         add_template_context_repr_sequence()
 
         if patch_caching is not None:

sentry_sdk/integrations/django/tasks.py

@@ -0,0 +1,38 @@
+from functools import wraps
+
+import sentry_sdk
+from sentry_sdk.consts import OP
+
+try:
+    # django.tasks were added in Django 6.0
+    from django.tasks.base import Task
+except ImportError:
+    Task = None
+
+
+def patch_tasks():
+    # type: () -> None
+    if Task is None:
+        return
+
+    old_task_enqueue = Task.enqueue
+
+    @wraps(old_task_enqueue)
+    def _sentry_enqueue(self, *args, **kwargs):
+        # type: (Any, Any) -> Any
+        from sentry_sdk.integrations.django import DjangoIntegration
+
+        integration = sentry_sdk.get_client().get_integration(DjangoIntegration)
+        if integration is None:
+            return old_task_enqueue(*args, **kwargs)
+
+        name = (
+            getattr(self.func, "__name__", repr(self.func)) or "<unknown Django task>"
+        )
+
+        with sentry_sdk.start_span(
+            op=OP.QUEUE_SUBMIT_DJANGO, name=name, origin=DjangoIntegration.origin
+        ):
+            return old_task_enqueue(*args, **kwargs)
+
+    Task.enqueue = _sentry_enqueue