rbd.py: make Image a context manager

Signed-off-by: Josh Durgin <josh.durgin@dreamhost.com>

diff --git a/src/pybind/rbd.py b/src/pybind/rbd.py
index b2228291d825285c1c30a889612f5b42f3830ce6..4aeb58b2ce8fb0fabf34d53a6469d21116702960 100644 (file)
--- a/src/pybind/rbd.py
+++ b/src/pybind/rbd.py
@@ -136,6 +136,13 @@ class Image(object):
         if ret != 0:
             raise make_ex(ret, 'error opening image %s at snapshot %s' % (name, snapshot))
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, type_, value, traceback):
+        self.close()
+        return False
+
     def close(self):
         if not self.closed:
             self.closed = True