Add option to patch stdlib logging to send events

Closes #19
Closes #20

Author: Rémy HUBSCHER

CHANGELOG.md

@@ -1,5 +1,9 @@
 # CHANGELOG
 
+## 0.6.0 (2019-07-11)
+
+- Add option to patch stdlib logging to send events (thanks, @Natim!)
+
 ## 0.5.0 (2019-01-10)
 
 - Use new-style, aiohttp 2.3+ middleware definition,

README.rst

@@ -43,6 +43,26 @@ With this, you can specify environment details, filter out specific exceptions,
         ),
     )
 
+If you are using the standard library's ``logging`` module,
+we have a convenient parameter to patch it for you,
+to have logger calls send events to Sentry automatically:
+
+.. warning::
+    This modifies your logging configuration globally
+    when you instantiate the middleware.
+    Even if you don't end up using the middleware instance for a request,
+    all your logs will be sent to Sentry.
+
+.. code-block:: python
+
+    import logging
+    from aiohttp import web
+    from aiohttp_sentry import SentryMiddleware
+
+    app = web.Application(
+        middlewares=[SentryMiddleware(patch_logging=True, sentry_log_level=logging.WARNING)],
+    )
+
 .. _aiohttp: http://aiohttp.readthedocs.io/en/stable/
 .. _Sentry: http://sentry.io/
 

aiohttp_sentry/__init__.py

@@ -1,14 +1,26 @@
+import asyncio
+import logging
 import sys
 
-import asyncio
 from aiohttp import web
+
 import raven
 import raven_aiohttp
+from raven.conf import setup_logging
+from raven.handlers.logging import SentryHandler
 
 
 @web.middleware
 class SentryMiddleware:
-    def __init__(self, sentry_kwargs=None, *, install_excepthook=True, loop=None):
+    def __init__(
+        self,
+        sentry_kwargs=None,
+        *,
+        install_excepthook=True,
+        loop=None,
+        patch_logging=False,
+        sentry_log_level=logging.ERROR
+    ):
         if sentry_kwargs is None:
             sentry_kwargs = {}
 
@@ -21,6 +33,12 @@ def __init__(self, sentry_kwargs=None, *, install_excepthook=True, loop=None):
         }
         self.client = raven.Client(**sentry_kwargs)
 
+        # Setup Sentry logger - https://docs.sentry.io/clients/python/integrations/#logging
+        if patch_logging:
+            handler = SentryHandler(client=self.client)
+            handler.setLevel(sentry_log_level)
+            setup_logging(handler)
+
         if install_excepthook:
             self.update_excepthook(loop)
 
@@ -49,7 +67,7 @@ def aiohttp_transport_excepthook(*exc_info):
                 event_loop.run_until_complete(transport.close())
             # the idea of using the original excepthook
             # was taken from raven-python repository:
-            # https://github.com/getsentry/raven-python/blob/f6d79c3bcc25e804b6259fa9c4a6e030f9033bb2/raven/base.py#L280
+            # https://github.com/getsentry/raven-python/blob/f6d79c3b/raven/base.py#L280
             original_excepthook(*exc_info)
 
         sys.excepthook = aiohttp_transport_excepthook

setup.py

@@ -12,7 +12,7 @@
 
 setup(
     name="aiohttp-sentry",
-    version="0.5.0",
+    version="0.6.0",
     url="https://github.com/underyx/aiohttp-sentry",
     author="Bence Nagy",
     author_email="[email protected]",