SimpleQueue and SimpleBuffer can now be used as contexts

1 files changed, 6 insertions, 0 deletions

diff --git a/kombu/simple.py b/kombu/simple.py
index da54b708..f473aaa3 100644
--- a/kombu/simple.py
+++ b/kombu/simple.py
@@ -21,6 +21,12 @@ from kombu import messaging
 class SimpleBase(object):
     _consuming = False
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, *exc_info):
+        self.close()
+
     def __init__(self, channel, producer, consumer, no_ack=False,
             channel_autoclose=False):
         self.channel = channel