From 32d59587a112957b62aca35f9f8ea05fff6e9fab Mon Sep 17 00:00:00 2001 From: Rob Browning Date: Sat, 31 Mar 2018 15:30:02 -0500 Subject: [PATCH] Make PackWriter a "with" context manager Signed-off-by: Rob Browning --- lib/bup/git.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/lib/bup/git.py b/lib/bup/git.py index db70448..35bf838 100644 --- a/lib/bup/git.py +++ b/lib/bup/git.py @@ -637,6 +637,12 @@ class PackWriter: def __del__(self): self.close() + def __enter__(self): + return self + + def __exit__(self, type, value, traceback): + self.close() + def _open(self): if not self.file: objdir = dir = os.path.join(self.repo_dir, 'objects') -- 2.39.2