exception_level_filters: support string for filtered class name

rollbar/__init__.py

@@ -403,7 +403,16 @@ def prev_page(self):
 
 
 def _filtered_level(exception):
-    for cls, level in SETTINGS['exception_level_filters']:
+    for i, item in enumerate(SETTINGS['exception_level_filters']):
+        cls, level = item
+        if isinstance(cls, str):
+            # Lazily resolve class name
+            parts = cls.split('.')
+            module = ".".join(parts[:-1])
+            if module not in sys.modules or not hasattr(sys.modules[module], parts[-1]):
+                continue
+            cls = getattr(sys.modules[module], parts[-1])
+            SETTINGS['exception_level_filters'][i] = (cls, level)
         if isinstance(exception, cls):
             return level
 

rollbar/test/test_rollbar.py

@@ -1,5 +1,7 @@
 import copy
+import exceptions
 import mock
+import unittest
 import urllib
 
 import rollbar
@@ -114,6 +116,43 @@ def _raise():
         self.assertNotIn('args', payload['data']['body']['trace']['frames'][-1])
         self.assertNotIn('kwargs', payload['data']['body']['trace']['frames'][-1])
 
+    @mock.patch('rollbar.send_payload')
+    def test_exception_filters(self, send_payload):
+
+        rollbar.SETTINGS['exception_level_filters'] = [
+            (exceptions.OSError, 'ignored'),
+            ('exceptions.NotImplementedError', 'ignored'),
+            ('bogus.DoesntExist', 'ignored'),
+        ]
+
+        def _raise_exception():
+            try:
+                raise Exception('foo')
+            except:
+                rollbar.report_exc_info()
+
+        def _raise_os_error():
+            try:
+                raise OSError('bar')
+            except:
+                rollbar.report_exc_info()
+
+        def _raise_not_implemented_error():
+            try:
+                raise NotImplementedError('bar')
+            except:
+                rollbar.report_exc_info()
+
+        _raise_exception()
+        self.assertTrue(send_payload.called)
+
+        _raise_os_error()
+        self.assertEqual(1, send_payload.call_count)
+
+        _raise_not_implemented_error()
+        self.assertEqual(1, send_payload.call_count)
+
+
     @mock.patch('rollbar.send_payload')
     def test_report_messsage(self, send_payload):
         rollbar.report_message('foo')
@@ -714,3 +753,6 @@ def step2():
 def called_with(arg1):
     arg1 = 'changed'
     step1()
+
+if __name__ == '__main__':
+    unittest.main()