fix: Ignore Celery exceptions used for control flow (getsentry#355)

untitaker · web-flow · commit f98255aa53d7 · 2019-04-29T14:39:12.000+02:00
* fix: Ignore Celery exceptions used for control flow

* fix: Linters
diff --git a/sentry_sdk/integrations/celery.py b/sentry_sdk/integrations/celery.py
@@ -2,7 +2,12 @@
 
 import sys
 
-from celery.exceptions import SoftTimeLimitExceeded, Retry  # type: ignore
+from celery.exceptions import (  # type: ignore
+    SoftTimeLimitExceeded,
+    Retry,
+    Ignore,
+    Reject,
+)
 
 from sentry_sdk.hub import Hub
 from sentry_sdk.utils import capture_internal_exceptions, event_from_exception
@@ -12,6 +17,9 @@
 from sentry_sdk.integrations.logging import ignore_logger
 
 
+CELERY_CONTROL_FLOW_EXCEPTIONS = (Retry, Ignore, Reject)
+
+
 class CeleryIntegration(Integration):
     identifier = "celery"
 
@@ -137,7 +145,7 @@ def _capture_exception(task, exc_info):
 
     if hub.get_integration(CeleryIntegration) is None:
         return
-    if isinstance(exc_info[1], Retry):
+    if isinstance(exc_info[1], CELERY_CONTROL_FLOW_EXCEPTIONS):
         return
     if hasattr(task, "throws") and isinstance(exc_info[1], task.throws):
         return
