Assign Sentry logger to a global variable and add an access function

metabrainz · Apr 27, 2017 · 87f2156 · 87f2156
1 parent 40d3e8e
commit 87f2156
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/brainzutils/flask/loggers.py b/brainzutils/flask/loggers.py
@@ -4,6 +4,8 @@
 import raven.contrib.flask
 import raven.transport.threaded_requests
 
+sentry_client = None
+
 
 def add_file_handler(app, filename, max_bytes=512 * 1024, backup_count=100):
     """Adds file logging."""
@@ -52,9 +54,15 @@ def add_sentry(app, dsn, level=logging.WARNING, **options):
     """
     app.config["SENTRY_TRANSPORT"] = raven.transport.threaded_requests.ThreadedRequestsHTTPTransport
     app.config["SENTRY_CONFIG"] = options
-    raven.contrib.flask.Sentry(
+    global sentry_client
+    sentry_client = raven.contrib.flask.Sentry(
         app=app,
         dsn=dsn,
         level=level,
         logging=True,
     )
+
+
+def get_sentry_client():
+    global sentry_client
+    return sentry_client