From c60abe1b484b00e0fe6e4e62ebab8e189f03395d Mon Sep 17 00:00:00 2001 From: Josh Durgin Date: Wed, 21 Sep 2011 15:44:16 -0700 Subject: [PATCH] rbd.py: make Image a context manager Signed-off-by: Josh Durgin --- src/pybind/rbd.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/pybind/rbd.py b/src/pybind/rbd.py index b2228291d8252..4aeb58b2ce8fb 100644 --- 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 -- 2.39.5