Merge pull request #777 from getsentry/fingerprint-soft-timeouts

Add fingerprint to soft timeouts in Celery

Author: dcramer

raven/contrib/celery/__init__.py

@@ -9,6 +9,7 @@
 
 import logging
 
+from celery.exceptions import SoftTimeLimitExceeded
 from celery.signals import after_setup_logger, task_failure
 from raven.handlers.logging import SentryHandler
 
@@ -24,15 +25,21 @@ def filter(self, record):
 
 
 def register_signal(client):
-    def process_failure_signal(sender, task_id, args, kwargs, **kw):
+    def process_failure_signal(sender, task_id, args, kwargs, einfo, **kw):
         # This signal is fired inside the stack so let raven do its magic
+        if isinstance(einfo.exception, SoftTimeLimitExceeded):
+            fingerprint = ['celery', 'SoftTimeLimitExceeded', sender]
+        else:
+            fingerprint = None
         client.captureException(
             extra={
                 'task_id': task_id,
                 'task': sender,
                 'args': args,
                 'kwargs': kwargs,
-            })
+            },
+            fingerprint=fingerprint,
+        )
 
     task_failure.connect(process_failure_signal, weak=False)