filter logging level for Flask sentry middleware

1 files changed, 6 insertions, 2 deletions

diff --git a/raven/contrib/flask/__init__.py b/raven/contrib/flask/__init__.py
index 00c43bb..d6cd5bd 100644
--- a/raven/contrib/flask/__init__.py
+++ b/raven/contrib/flask/__init__.py
@@ -67,11 +67,12 @@ class Sentry(object):
     >>> sentry.captureMessage('hello, world!')
     """
     def __init__(self, app=None, client=None, client_cls=Client, dsn=None,
-                 logging=False):
+                 logging=False, level=None):
         self.dsn = dsn
         self.logging = logging
         self.client_cls = client_cls
         self.client = client
+        self.level = level
 
         if app:
             self.init_app(app)
@@ -98,7 +99,10 @@ class Sentry(object):
             self.client = make_client(self.client_cls, app, self.dsn)
 
         if self.logging:
-            setup_logging(SentryHandler(self.client))
+            if self.level is not None:
+                setup_logging(SentryHandler(self.client, level=self.level))
+            else: 
+                setup_logging(SentryHandler(self.client))
 
         got_request_exception.connect(self.handle_exception, sender=app)