Fix BufferingSMTPHandler used as a context

Since a1cb1581, BufferingSMTPHandler was used directly as a context to
send an unhandled exception termination report email, but that class
doesn't support such use...

1 files changed, 11 insertions, 0 deletions

diff --git a/calm/buffering_smtp_handler.py b/calm/buffering_smtp_handler.py
index b7dfee7..2cd1b63 100644
--- a/calm/buffering_smtp_handler.py
+++ b/calm/buffering_smtp_handler.py
@@ -95,3 +95,14 @@ class BufferingSMTPHandler(logging.handlers.BufferingHandler):
         # the capacity we pass to BufferingHandler is irrelevant since we
         # override shouldFlush so it never indicates we have reached capacity
         return False
+
+    def __enter__(self):
+        logging.getLogger().addHandler(self)
+        return self
+
+    def __exit__(self, exception_type, exception_value, traceback):
+        self.close()
+        logging.getLogger().removeHandler(self)
+
+        # process any exception in the with-block normally
+        return False