Merge pull request #319 from RickyCook/rollbar-ignore-some-exceptions

Rollbar ignore some exceptions

Author: RickyCook

dockci/api/exceptions.py

@@ -5,6 +5,7 @@
 class BaseActionExceptionMixin(HTTPException):
     """ An HTTP exception for when an action can't be performed """
     response = None
+    no_rollbar = True
 
     def __init__(self, action=None):
         super(BaseActionExceptionMixin, self).__init__()
@@ -39,6 +40,7 @@ class WrappedException(HTTPException):
     Wraps an exception in HTTPException so that it can have a status code
     """
     response = None
+    no_rollbar = True
 
     def __init__(self, ex):
         super(WrappedException, self).__init__()

dockci/server.py

@@ -122,6 +122,14 @@ def app_init():
     app_init_workers()
 
 
+def wrapped_report_exception(app, exception):
+    """ Wrapper for ``report_exception`` to ignore some exceptions """
+    if getattr(exception, 'no_rollbar', False):
+        return
+
+    return rollbar.contrib.flask.report_exception(app, exception)
+
+
 def app_init_rollbar():
     """ Initialize Rollbar for error/exception reporting """
     try:
@@ -138,9 +146,7 @@ def app_init_rollbar():
         allow_logging_basic_config=False,
     )
 
-    flask.got_request_exception.connect(
-        rollbar.contrib.flask.report_exception, APP,
-    )
+    flask.got_request_exception.connect(wrapped_report_exception, APP)
 
 
 def app_init_workers():